From f3d05a65ffc6862c2f655c9699b718c9a4286aaa Mon Sep 17 00:00:00 2001 From: sanine-a Date: Tue, 2 May 2023 15:39:16 -0500 Subject: add configuration of host address and login page --- conf/main.go | 67 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ db/db.go | 6 ++++++ phlox/main.go | 24 ++++++++++++++------- 3 files changed, 90 insertions(+), 7 deletions(-) diff --git a/conf/main.go b/conf/main.go index 182d97f..d0785f8 100644 --- a/conf/main.go +++ b/conf/main.go @@ -71,6 +71,11 @@ func main() { 9: SetPath(), 10: SetAddress(), 11: ListEndpoints(), + + 12: SetHost(), + 13: GetHost(), + 14: SetLogin(), + 15: GetLogin(), } for val, cmd := range cmds { @@ -116,3 +121,65 @@ func SchemaVersion() *Command { }, } } + + +func SetHost() *Command { + return &Command{ + Flag: "set-host-address", + Usage: "set the address to serve the reverse proxy on", + Invoke: func() { + OpenDb() + defer p.Close() + address := ReadLine("Address: ") + err := p.SetHostAddress(address) + if err != nil { log.Fatal(err) } + fmt.Printf("set host address to %v\n", address) + }, + } +} + + +func GetHost() *Command { + return &Command{ + Flag: "get-host-address", + Usage: "get the address to serve the reverse proxy on", + Invoke: func() { + OpenDb() + defer p.Close() + address, err := p.GetHostAddress() + if err != nil { log.Fatal(err) } + fmt.Printf("host address is %v\n", address) + }, + } +} + + +func SetLogin() *Command { + return &Command{ + Flag: "set-login-page", + Usage: "set the file to serve as the login page", + Invoke: func() { + OpenDb() + defer p.Close() + path := ReadLine("File Path (blank for internal default): ") + err := p.SetLoginPage(path) + if err != nil { log.Fatal(err) } + fmt.Printf("set login page to %v\n", path) + }, + } +} + + +func GetLogin() *Command { + return &Command{ + Flag: "get-login-page", + Usage: "get the file to serve the reverse proxy on", + Invoke: func() { + OpenDb() + defer p.Close() + path, err := p.GetLoginPage() + if err != nil { log.Fatal(err) } + fmt.Printf("login page is at %v\n", path) + }, + } +} diff --git a/db/db.go b/db/db.go index 3210846..d9b4578 100644 --- a/db/db.go +++ b/db/db.go @@ -150,6 +150,12 @@ func (p *Phlox) Open(filename string) error { } +func (p *Phlox) Close() error { + err := p.db.Close() + return err +} + + func (p *Phlox) GetSchemaVersion() (int, error) { row := p.db.QueryRow("select max(version) from schema;") if row.Err() != nil { diff --git a/phlox/main.go b/phlox/main.go index 3516374..ee9416c 100644 --- a/phlox/main.go +++ b/phlox/main.go @@ -4,14 +4,30 @@ import ( "fmt" "io" "strings" + "flag" "net" "net/http" "net/url" log "github.com/sirupsen/logrus" + db "sanine.net/git/phlox/db" ) +var p *db.Plhox + + func main() { + var dbfile string + flag.StringVar(&dbfile, "db", "/etc/phlox/phlox.conf", "path to the configuration db") + flag.Parse() + + p := &db.Phlox{} + err := p.Open(dbfile) + if err != nil { + log.Fatal(err) + } + defer p.Close() + const addr = "localhost:3333" log.Info("configuring reverse proxy...") @@ -22,12 +38,6 @@ func main() { } -type Endpoint struct { - Path string - Origin *url.URL -} - - func configureEndpoint(path, address string) { log.Infof("proxying endpoint %v to %v", path, address) origin, err := url.Parse(address) @@ -68,7 +78,7 @@ func proxyRequest(w http.ResponseWriter, req *http.Request, end Endpoint) *http // misc request cleanups req.URL.Scheme = end.Origin.Scheme req.RequestURI = "" - + // make request response, err := http.DefaultClient.Do(req) if err != nil { -- cgit v1.2.1