package goxy import ( "crypto/tls" "encoding/gob" "log" "net/http" "net/http/httputil" "net/url" "os" "strings" ) type Server struct { Data string Route http.Server } func NewServer(fname string) (*Server, error) { r := make(Route) s := http.Server{TLSConfig: &tls.Config{GetCertificate: r.GetCertificate}} server := &Server{Route: r, Server: s, Data: fname} if err := server.Load(); err != nil { log.Println(err) } if err := server.Restore(); err != nil { return nil, err } Register(server) return server, nil } // Save routes to persistent file func (s Server) Save() error { if s.Data != "" { fd, err := os.Create(s.Data) if err != nil { return err } defer fd.Close() return gob.NewEncoder(fd).Encode(s.Route) } return nil } // Load routes from persistent file func (s *Server) Load() error { if s.Data != "" { fd, err := os.Open(s.Data) if err != nil { return err } defer fd.Close() return gob.NewDecoder(fd).Decode(&s.Route) } return nil } // Restore and update routes from in-memory state func (s *Server) Restore() error { mux := http.NewServeMux() for k, v := range s.Route { if v.Cert != nil && v.Key != nil { cert, err := tls.X509KeyPair(v.Cert, v.Key) if err != nil { return err } v.cert = &cert s.Route[k] = v } up, err := url.Parse(v.Upstream) if err != nil { return err } if !strings.Contains(v.ServerName, "/") { v.ServerName += "/" } switch up.Scheme { case "ws": mux.Handle(v.ServerName, httputil.NewSingleHostReverseProxy(up)) default: mux.Handle(v.ServerName, NewReverseProxy(up)) } } s.Server.Handler = mux return nil } func (s *Server) Start() <-chan error { errc := make(chan error, 3) go func() { errc <- s.ListenAndServe() }() go func() { errc <- s.ListenAndServeTLS("", "") }() go func() { errc <- http.ListenAndServe(":http-alt", nil) }() return errc }