From 65019345827c9c4b4966ba7ad16cfc3d44bae706 Mon Sep 17 00:00:00 2001 From: Bel LaPointe <153096461+breel-render@users.noreply.github.com> Date: Sat, 14 Dec 2024 19:32:21 -0700 Subject: [PATCH] stub --- cmd/server/main.go | 126 ++++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 125 insertions(+), 1 deletion(-) diff --git a/cmd/server/main.go b/cmd/server/main.go index 57bc283..3bb5cc7 100644 --- a/cmd/server/main.go +++ b/cmd/server/main.go @@ -2,9 +2,19 @@ package main import ( "context" + "flag" + "fmt" "io" + "log" + "net/http" + "os" "os/signal" + "strings" "syscall" + "time" + + "github.com/coder/websocket" + "golang.org/x/time/rate" ) func main() { @@ -17,5 +27,119 @@ func main() { } func run(ctx context.Context) error { - return io.EOF + fs := flag.NewFlagSet(os.Args[0], flag.ContinueOnError) + port := fs.Int("p", 8080, "port") + if err := fs.Parse(os.Args[1:]); err != nil { + return err + } + + S := &S{ + ctx: ctx, + limiter: rate.NewLimiter(10, 10), + } + + s := &http.Server{ + Addr: fmt.Sprintf(":%d", *port), + Handler: S, + } + + go func() { + <-ctx.Done() + ctx, can := context.WithTimeout(context.Background(), time.Second) + defer can() + s.Shutdown(ctx) + }() + + log.Println("listening on", *port) + if err := s.ListenAndServe(); err != nil && ctx.Err() == nil { + return err + } + log.Println("shut down") + return nil +} + +type S struct { + ctx context.Context + limiter *rate.Limiter +} + +func (s *S) ServeHTTP(w http.ResponseWriter, r *http.Request) { + if err := s.serveHTTP(w, r); err != nil { + log.Println(r.URL.Path, "//", err.Error(), r.Header) + http.Error(w, err.Error(), http.StatusInternalServerError) + } +} + +func (s *S) serveHTTP(w http.ResponseWriter, r *http.Request) error { + if isV1(r) || isWS(r) { + return s.serveAPI(w, r) + } + return s.serveStatic(w, r) +} + +func isV1(r *http.Request) bool { + return strings.HasPrefix(r.URL.Path, "/v1/") +} + +func isWS(r *http.Request) bool { + return r.URL.Path == "/ws" || strings.HasPrefix(r.URL.Path, "/ws/") +} + +func (s *S) serveStatic(w http.ResponseWriter, r *http.Request) error { + return fmt.Errorf("not impl static") +} + +func (s *S) serveAPI(w http.ResponseWriter, r *http.Request) error { + if err := s.injectContext(w, r); err == io.EOF { + http.Redirect(w, r, "/", http.StatusSeeOther) + return nil + } else if err != nil { + return err + } + + if isWS(r) { + return s.serveWS(w, r) + } else if isV1(r) { + return s.serveV1(w, r) + } + + http.NotFound(w, r) + return nil +} + +type Session struct { + ID string +} + +func (s *S) injectContext(w http.ResponseWriter, r *http.Request) error { + id, err := r.Cookie("uuid") + if err != nil || id.Value == "" { + return io.EOF + } + ctx := r.Context() + ctx = context.WithValue(ctx, "session", Session{ + ID: id.Value, + }) + *r = *r.WithContext(ctx) + return nil +} + +func (s *S) serveWS(httpw http.ResponseWriter, httpr *http.Request) error { + ctx := httpr.Context() + + c, err := websocket.Accept(httpw, httpr, nil) + if err != nil { + return err + } + defer c.CloseNow() + + if err := c.Write(ctx, 1, []byte("hello world")); err != nil { + return err + } + + return fmt.Errorf("not impl") +} + +func (s *S) serveV1(w http.ResponseWriter, r *http.Request) error { + return fmt.Errorf("not impl: v1") }