refactor: move into package
This commit is contained in:
parent
79114e8b8e
commit
f9c064e948
GPG Key ID:
8B3B0604F164E04F
|
|
@ -23,7 +23,7 @@ type info struct{}
|
||||||
func (info) RegisterHTTP(mux *http.ServeMux) {
|
func (info) RegisterHTTP(mux *http.ServeMux) {
|
||||||
mux.HandleFunc("/app-info", func(w http.ResponseWriter, r *http.Request) {
|
mux.HandleFunc("/app-info", func(w http.ResponseWriter, r *http.Request) {
|
||||||
name, version := service.AppName()
|
name, version := service.AppName()
|
||||||
fmt.Fprint(w, name, version)
|
fmt.Fprint(w, name, '@', version)
|
||||||
})
|
})
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
103
app.paste.go
103
app.paste.go
|
|
@ -2,19 +2,12 @@ package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"crypto/rand"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"log"
|
|
||||||
"net/http"
|
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
|
||||||
|
|
||||||
"go.sour.is/pkg/env"
|
"go.sour.is/pkg/env"
|
||||||
"go.sour.is/pkg/lg"
|
"go.sour.is/pkg/lg"
|
||||||
"go.sour.is/pkg/service"
|
"go.sour.is/pkg/service"
|
||||||
|
|
||||||
"go.sour.is/paste/v2/assets"
|
|
||||||
"go.sour.is/paste/v2/paste"
|
"go.sour.is/paste/v2/paste"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -28,104 +21,12 @@ var _ = apps.Register(50, func(ctx context.Context, svc *service.Harness) error
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
p, err := paste.New(store)
|
srv, err := paste.NewService(store, randBytes)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.Add(&pasteSRV{
|
svc.Add(srv)
|
||||||
ui: http.FileServer(http.FS(assets.GetUI())),
|
|
||||||
paste: p,
|
|
||||||
randBytes: int(randBytes),
|
|
||||||
})
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
})
|
})
|
||||||
|
|
||||||
type pasteSRV struct {
|
|
||||||
ui http.Handler
|
|
||||||
paste *paste.Paste
|
|
||||||
randBytes int
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *pasteSRV) RegisterHTTP(mux *http.ServeMux) {
|
|
||||||
mux.Handle("/ui/", lg.Htrace(http.StripPrefix("/ui/", a.ui), "paste-assets"))
|
|
||||||
|
|
||||||
mux.Handle("/api", http.StripPrefix("/api", a))
|
|
||||||
mux.Handle("/api/", http.StripPrefix("/api/", a))
|
|
||||||
|
|
||||||
mux.Handle("/paste", http.StripPrefix("/paste", a))
|
|
||||||
mux.Handle("/paste/", http.StripPrefix("/paste/", a))
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *pasteSRV) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
||||||
switch r.Method {
|
|
||||||
case http.MethodGet:
|
|
||||||
switch {
|
|
||||||
case strings.HasPrefix(r.URL.Path, "rng"):
|
|
||||||
p.getRandom(w)
|
|
||||||
|
|
||||||
case strings.HasPrefix(r.URL.Path, "get"), r.URL.Path != "":
|
|
||||||
p.getPaste(w, strings.TrimPrefix(r.URL.Path, "get/"))
|
|
||||||
|
|
||||||
default:
|
|
||||||
http.Error(w, http.StatusText(http.StatusNotFound), http.StatusNotFound)
|
|
||||||
}
|
|
||||||
|
|
||||||
case http.MethodPost:
|
|
||||||
switch {
|
|
||||||
case r.URL.Path == "":
|
|
||||||
p.postPaste(w, r)
|
|
||||||
|
|
||||||
default:
|
|
||||||
http.Error(w, http.StatusText(http.StatusNotFound), http.StatusNotFound)
|
|
||||||
}
|
|
||||||
|
|
||||||
default:
|
|
||||||
http.Error(w, http.StatusText(http.StatusMethodNotAllowed), http.StatusMethodNotAllowed)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *pasteSRV) getRandom(w http.ResponseWriter) {
|
|
||||||
log.Println("get random")
|
|
||||||
s := make([]byte, p.randBytes)
|
|
||||||
_, _ = rand.Read(s)
|
|
||||||
|
|
||||||
w.Header().Set("content-type", "application/octet-stream")
|
|
||||||
w.WriteHeader(http.StatusOK)
|
|
||||||
_, _ = w.Write(s)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *pasteSRV) getPaste(w http.ResponseWriter, id string) {
|
|
||||||
log.Println("get paste", id)
|
|
||||||
w.Header().Set("content-type", "application/octet-stream")
|
|
||||||
|
|
||||||
err := p.paste.Get(w, id)
|
|
||||||
if err != nil {
|
|
||||||
switch {
|
|
||||||
case errors.Is(err, paste.ErrGone):
|
|
||||||
w.WriteHeader(http.StatusGone)
|
|
||||||
case errors.Is(err, paste.ErrNotFound):
|
|
||||||
w.WriteHeader(http.StatusNotFound)
|
|
||||||
case errors.Is(err, paste.ErrReadingContent):
|
|
||||||
w.WriteHeader(http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Fprintf(w, "ERR %s", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *pasteSRV) postPaste(w http.ResponseWriter, r *http.Request) {
|
|
||||||
w.Header().Set("content-type", "application/octet-stream")
|
|
||||||
|
|
||||||
id, err := p.paste.Set(r.Body)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Fprintf(w, "ERR %s", err)
|
|
||||||
return
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Println("post paste", id)
|
|
||||||
fmt.Fprint(w, "OK ", id)
|
|
||||||
}
|
|
||||||
|
|
|
||||||
19
paste/public/index.html
Normal file
19
paste/public/index.html
Normal file
|
|
@ -0,0 +1,19 @@
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<meta name="viewport" content="width=device-width,initial-scale=1,shrink-to-fit=no">
|
||||||
|
<meta name="theme-color" content="#000000">
|
||||||
|
|
||||||
|
<link rel="manifest" href="/ui/manifest.json">
|
||||||
|
<link rel="shortcut icon" href="/ui/favicon.ico">
|
||||||
|
|
||||||
|
<title>DN42 Paste</title>
|
||||||
|
|
||||||
|
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap@4.0.0/dist/css/bootstrap.min.css" integrity="sha384-Gn5384xqQ1aoWXA+058RXPxPg6fy4IWvTNh0E263XmFcJlSAwiGgFAW/dAiS6JXm" crossorigin="anonymous">
|
||||||
|
<link href="paste.css" rel="stylesheet">
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<noscript>You need to enable JavaScript to run this app.</noscript>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
0
paste/public/paste.css
Normal file
0
paste/public/paste.css
Normal file
114
paste/service.go
Normal file
114
paste/service.go
Normal file
|
|
@ -0,0 +1,114 @@
|
||||||
|
package paste
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/rand"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"go.sour.is/paste/v2/assets"
|
||||||
|
"go.sour.is/pkg/lg"
|
||||||
|
)
|
||||||
|
|
||||||
|
type service struct {
|
||||||
|
ui http.Handler
|
||||||
|
paste *Paste
|
||||||
|
randBytes int
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewService(store string, randBytes int64) (*service, error) {
|
||||||
|
p, err := New(store)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &service{
|
||||||
|
ui: http.FileServer(http.FS(assets.GetUI())),
|
||||||
|
paste: p,
|
||||||
|
randBytes: int(randBytes),
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *service) RegisterHTTP(mux *http.ServeMux) {
|
||||||
|
mux.Handle("/ui/", lg.Htrace(http.StripPrefix("/ui/", a.ui), "paste-assets"))
|
||||||
|
|
||||||
|
mux.Handle("/api", http.StripPrefix("/api", a))
|
||||||
|
mux.Handle("/api/", http.StripPrefix("/api/", a))
|
||||||
|
|
||||||
|
mux.Handle("/paste", http.StripPrefix("/paste", a))
|
||||||
|
mux.Handle("/paste/", http.StripPrefix("/paste/", a))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *service) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||||
|
switch r.Method {
|
||||||
|
case http.MethodGet:
|
||||||
|
switch {
|
||||||
|
case strings.HasPrefix(r.URL.Path, "rng"):
|
||||||
|
p.getRandom(w)
|
||||||
|
|
||||||
|
case strings.HasPrefix(r.URL.Path, "get"), r.URL.Path != "":
|
||||||
|
p.getPaste(w, strings.TrimPrefix(r.URL.Path, "get/"))
|
||||||
|
|
||||||
|
default:
|
||||||
|
http.Error(w, http.StatusText(http.StatusNotFound), http.StatusNotFound)
|
||||||
|
}
|
||||||
|
|
||||||
|
case http.MethodPost:
|
||||||
|
switch {
|
||||||
|
case r.URL.Path == "":
|
||||||
|
p.postPaste(w, r)
|
||||||
|
|
||||||
|
default:
|
||||||
|
http.Error(w, http.StatusText(http.StatusNotFound), http.StatusNotFound)
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
http.Error(w, http.StatusText(http.StatusMethodNotAllowed), http.StatusMethodNotAllowed)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *service) getRandom(w http.ResponseWriter) {
|
||||||
|
log.Println("get random")
|
||||||
|
s := make([]byte, p.randBytes)
|
||||||
|
_, _ = rand.Read(s)
|
||||||
|
|
||||||
|
w.Header().Set("content-type", "application/octet-stream")
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
_, _ = w.Write(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *service) getPaste(w http.ResponseWriter, id string) {
|
||||||
|
log.Println("get paste", id)
|
||||||
|
w.Header().Set("content-type", "application/octet-stream")
|
||||||
|
|
||||||
|
err := p.paste.Get(w, id)
|
||||||
|
if err != nil {
|
||||||
|
switch {
|
||||||
|
case errors.Is(err, ErrGone):
|
||||||
|
w.WriteHeader(http.StatusGone)
|
||||||
|
case errors.Is(err, ErrNotFound):
|
||||||
|
w.WriteHeader(http.StatusNotFound)
|
||||||
|
case errors.Is(err, ErrReadingContent):
|
||||||
|
w.WriteHeader(http.StatusInternalServerError)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Fprintf(w, "ERR %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *service) postPaste(w http.ResponseWriter, r *http.Request) {
|
||||||
|
w.Header().Set("content-type", "application/octet-stream")
|
||||||
|
|
||||||
|
id, err := p.paste.Set(r.Body)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Fprintf(w, "ERR %s", err)
|
||||||
|
return
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println("post paste", id)
|
||||||
|
fmt.Fprint(w, "OK ", id)
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user