main.go

package main

import (
	"github.com/elazarl/goproxy"
	"github.com/lightsing/makehttps/config"
	"github.com/lightsing/makehttps/proxy"
	"github.com/lightsing/makehttps/rules"
	log "github.com/sirupsen/logrus"
	"net/http"
	"time"
)

const name = "mhga"

var Version string

func main() {
	// Only log the warning severity or above.
	config := config.Init()
	for _, rule := range config.Rules {
		if err := rules.CheckRule(&rule); err == nil {
			config.AvailableRules = append(config.AvailableRules, rule.Path)
		} else {
			log.Errorf("Rule check fail by (%s)", err)
		}
	}
	/*if len(config.AvailableRules) == 0 {
		panic("no available rules")
	}*/
	start := time.Now()
	rules := rules.NewRuleSets()
	for _, path := range config.AvailableRules {
		rules.LoadRuleSets(path)
	}
	log.Infof("Load all rule in %s", time.Since(start))
	server := goproxy.NewProxyHttpServer()
	responseBuilder := proxy.NewResponseBuilder(Version)
	server.OnRequest().DoFunc(
		func(req *http.Request, ctx *goproxy.ProxyCtx) (*http.Request, *http.Response) {
			start := time.Now()
			if result, ok := rules.Apply(req.RequestURI); ok {
				result := *result
				log.Infof("[%s] Redirect %s to %s", time.Since(start), req.RequestURI, result)
				var code int
				if req.Method == "GET" {
					code = http.StatusMovedPermanently
				} else {
					code = http.StatusTemporaryRedirect
				}
				return req, responseBuilder.Gen(req, result, code)
			}
			log.Infof("[%s] Nothing to do with %s", time.Since(start), req.RequestURI)
			return req, nil
		})
	log.Fatal(http.ListenAndServe(config.Address, server))

}