diff options
Diffstat (limited to 'roseh.moe.go')
| -rw-r--r-- | roseh.moe.go | 370 |
1 files changed, 3 insertions, 367 deletions
diff --git a/roseh.moe.go b/roseh.moe.go index d9a948b..206679b 100644 --- a/roseh.moe.go +++ b/roseh.moe.go @@ -2,7 +2,6 @@ package main import ( "archive/tar" - "archive/zip" "cmp" "compress/gzip" "crypto/hkdf" @@ -11,7 +10,6 @@ import ( "crypto/sha1" "crypto/sha256" "crypto/subtle" - "database/sql" "embed" "encoding/base64" "encoding/binary" @@ -20,8 +18,6 @@ import ( "flag" "fmt" "html/template" - "image" - "image/jpeg" "io" "io/fs" "log" @@ -30,16 +26,13 @@ import ( "net/http/httputil" "net/url" "os" - "path/filepath" "slices" "strconv" "strings" "sync" "time" - _ "github.com/mattn/go-sqlite3" "github.com/skip2/go-qrcode" - "golang.org/x/image/draw" "roseh.moe/pkg/oae2" "roseh.moe/pkg/roseh.moe/internal/pwhash" "roseh.moe/pkg/wordlist" @@ -55,8 +48,6 @@ var ( holeTempDir = flag.String("hole-temp-dir", "hole", "directory to store temporary files for the wormhole") codeDir = flag.String("code-dir", "code", "directory to serve git repos") nezuko = flag.String("nezuko", "nezuko", "nezuko partition") - martDir = flag.String("mart-dir", "/mnt/nezuko/downloads/[metart.com] Photosets - 1999 to 2023", "met art data directory") - martDB = flag.String("mart-db", "/mnt/nezuko/files/met-art.sqlite", "met art index") serverStartTime = time.Now() ) @@ -914,297 +905,6 @@ func (h *nezukoHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { h.srv.ServeHTTP(w, r) } -var ( - //go:embed templates/mart/index.html.template - metArtIndexString string - metArtIndexTemplate = template.Must(outlinedTemplate(metArtIndexString)) -) - -type metArtIndexTemplateArgsGirl struct { - Name string - URL string - CoverID int64 -} - -type metArtIndexTemplateArgs struct { - Girls []metArtIndexTemplateArgsGirl -} - -type metArtIndexHandler struct { - s *stmts -} - -func (h *metArtIndexHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { - if !stupidCache(w, r) { - return - } - rows, err := h.s.listGirls.Query() - if err != nil { - log.Printf("Warning: list girls: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - var girls []metArtIndexTemplateArgsGirl - for rows.Next() { - var girl metArtIndexTemplateArgsGirl - if err := rows.Scan(&girl.Name, &girl.CoverID); err != nil { - log.Printf("Warning: list girls: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - girl.URL = strings.ReplaceAll(girl.Name, " ", "-") - girls = append(girls, girl) - } - if err := rows.Err(); err != nil { - log.Printf("Warning: list girls: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - if err := metArtIndexTemplate.Execute(w, metArtIndexTemplateArgs{girls}); err != nil { - log.Printf("Warning: mart index: %s", err) - } -} - -type metArtPhotoHandler struct { - s *stmts -} - -func (h *metArtPhotoHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { - if !stupidCache(w, r) { - return - } - id, err := strconv.ParseInt(strings.TrimSuffix(r.PathValue("photo"), ".jpeg"), 10, 64) - if err != nil { - notFound(w, r) - return - } - var path, filename string - if err := h.s.findPhoto.QueryRow(id).Scan(&path, &filename); err != nil { - if err == sql.ErrNoRows { - notFound(w, r) - return - } - log.Printf("Warning: find photo: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - archive, err := zip.OpenReader(filepath.Join(*martDir, path)) - if err != nil { - log.Printf("Warning: open archive: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - defer archive.Close() - f, err := archive.Open(filename) - if err != nil { - log.Printf("Warning: open archive file: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - defer f.Close() - w.Header().Set("Content-Type", "image/jpeg") - if stat, err := f.Stat(); err == nil { - w.Header().Set("Content-Length", strconv.FormatInt(stat.Size(), 10)) - } - if _, err := io.Copy(w, f); err != nil { - log.Printf("Warning: met art image: %s", err) - } -} - -var ( - //go:embed templates/mart/girl.html.template - metArtGirlString string - metArtGirlTemplate = template.Must(outlinedTemplate(metArtGirlString)) -) - -type metArtGirlTemplateArgsSet struct { - URL string - CoverID int64 - Name string -} - -type metArtGirlTemplateArgs struct { - Name string - URL string - Sets []metArtGirlTemplateArgsSet -} - -type metArtGirlHandler struct { - s *stmts -} - -func (h *metArtGirlHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { - if !stupidCache(w, r) { - return - } - name := strings.ReplaceAll(r.PathValue("girl"), "-", " ") - rows, err := h.s.listSetsForGirl.Query(name) - if err != nil { - log.Printf("Warning: list sets: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - var sets []metArtGirlTemplateArgsSet - for rows.Next() { - var id, coverID int64 - var name string - if err := rows.Scan(&id, &coverID, &name); err != nil { - log.Printf("Warning: list sets: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - sets = append(sets, metArtGirlTemplateArgsSet{ - Name: name, - URL: fmt.Sprintf("%d-%s", id, strings.ReplaceAll(name, " ", "-")), - CoverID: coverID, - }) - } - if err := rows.Err(); err != nil { - log.Printf("Warning: list sets: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - if len(sets) == 0 { - notFound(w, r) - return - } - if err := metArtGirlTemplate.Execute(w, metArtGirlTemplateArgs{Name: name, URL: r.PathValue("girl"), Sets: sets}); err != nil { - log.Printf("Warning: mart girl: %s", err) - } -} - -type metArtGirlPreviewHandler struct { - s *stmts -} - -func (h *metArtGirlPreviewHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { - if !stupidCache(w, r) { - return - } - id, err := strconv.ParseInt(strings.TrimSuffix(r.PathValue("photo"), ".jpeg"), 10, 64) - if err != nil { - notFound(w, r) - return - } - var path, filename string - if err := h.s.findPhoto.QueryRow(id).Scan(&path, &filename); err != nil { - if err == sql.ErrNoRows { - notFound(w, r) - return - } - log.Printf("Warning: find photo: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - archive, err := zip.OpenReader(filepath.Join(*martDir, path)) - if err != nil { - log.Printf("Warning: open archive: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - defer archive.Close() - f, err := archive.Open(filename) - if err != nil { - log.Printf("Warning: open archive file: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - defer f.Close() - fullSize, _, err := image.Decode(f) - if err != nil { - log.Printf("Warning: open preview photo: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - resized := image.NewRGBA(image.Rect(0, 0, 250, fullSize.Bounds().Dy()*250/fullSize.Bounds().Dx())) - draw.ApproxBiLinear.Scale(resized, resized.Bounds(), fullSize, fullSize.Bounds(), draw.Src, nil) - w.Header().Set("Content-Type", "image/jpeg") - if err := jpeg.Encode(w, resized, &jpeg.Options{Quality: 50}); err != nil { - log.Printf("Warning: met art preview image: %s", err) - } -} - -var ( - //go:embed templates/mart/set.html.template - metArtSetString string - metArtSetTemplate = template.Must(outlinedTemplate(metArtSetString)) -) - -type metArtSetTemplateArgs struct { - Name string - Photos []int64 -} - -type metArtSetHandler struct { - s *stmts -} - -func (h *metArtSetHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { - if !stupidCache(w, r) { - return - } - setID, _, _ := strings.Cut(r.PathValue("set"), "-") - id, err := strconv.ParseInt(setID, 10, 64) - if err != nil { - notFound(w, r) - return - } - wg := new(sync.WaitGroup) - var setName string - var setErr error - wg.Go(func() { - setErr = h.s.setName.QueryRow(id).Scan(&setName) - }) - var photos []int64 - var photosErr error - wg.Go(func() { - rows, err := h.s.listPhotosForSet.Query(id) - if err != nil { - photosErr = err - return - } - for rows.Next() { - var photoID int64 - if err := rows.Scan(&photoID); err != nil { - photosErr = err - return - } - photos = append(photos, photoID) - } - if err := rows.Err(); err != nil { - photosErr = err - return - } - }) - wg.Wait() - if err := setErr; err != nil { - log.Printf("Error: find set name: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - if err := photosErr; err != nil { - log.Printf("Warning: list set photos: %s", err) - http.Error(w, "internal error", http.StatusInternalServerError) - return - } - if len(photos) == 0 { - notFound(w, r) - return - } - if err := metArtSetTemplate.Execute(w, metArtSetTemplateArgs{Name: setName, Photos: photos}); err != nil { - log.Printf("Warning: mart set: %s", err) - } -} - -func stupidCache(w http.ResponseWriter, r *http.Request) bool { - w.Header().Set("Last-Modified", "Mon, 01 Jan 1990 00:00:00 GMT") - if _, ok := r.Header["If-Modified-Since"]; ok { - w.WriteHeader(http.StatusNotModified) - return false - } - return true -} - type jellyfinReverseProxy struct { proxy httputil.ReverseProxy } @@ -1217,56 +917,6 @@ func (p *jellyfinReverseProxy) ServeHTTP(w http.ResponseWriter, r *http.Request) p.proxy.ServeHTTP(w, r) } -type stmts struct { - listGirls *sql.Stmt - findPhoto *sql.Stmt - girlName *sql.Stmt - listSetsForGirl *sql.Stmt - setName *sql.Stmt - listPhotosForSet *sql.Stmt -} - -func prepareStatements(db *sql.DB) (*stmts, error) { - s := new(stmts) - var err error - s.listGirls, err = db.Prepare(` - SELECT - girls.name, - ( - SELECT cover_photo_id - FROM sets - WHERE sets.girl_id = girls.girl_id - ORDER BY date ASC, sets.name ASC LIMIT 1 - ) - FROM girls - ORDER BY substr(girls.name, instr(girls.name, ' ')+1), girls.name - `) - if err != nil { - return nil, err - } - s.findPhoto, err = db.Prepare("SELECT path, filename FROM photos JOIN sets USING (set_id) WHERE photo_id = ?") - if err != nil { - return nil, err - } - s.girlName, err = db.Prepare("SELECT name FROM girls WHERE girl_id = ?") - if err != nil { - return nil, err - } - s.listSetsForGirl, err = db.Prepare("SELECT set_id, cover_photo_id, sets.name FROM girls JOIN sets USING (girl_id) WHERE girls.name = ? ORDER BY date ASC, sets.name ASC") - if err != nil { - return nil, err - } - s.setName, err = db.Prepare("SELECT name FROM sets WHERE set_id = ?") - if err != nil { - return nil, err - } - s.listPhotosForSet, err = db.Prepare("SELECT photo_id FROM sets JOIN photos USING (set_id) WHERE set_id = ? AND idx IS NOT NULL ORDER BY idx ASC") - if err != nil { - return nil, err - } - return s, nil -} - func main() { flag.Parse() @@ -1274,15 +924,6 @@ func main() { log.Fatal(err) } - db, err := sql.Open("sqlite3", *martDB) - if err != nil { - log.Fatal(err) - } - stmts, err := prepareStatements(db) - if err != nil { - log.Fatal(err) - } - go func() { for range time.NewTicker(2 * time.Hour).C { if err := cleanHole(); err != nil { @@ -1293,7 +934,7 @@ func main() { mux := http.NewServeMux() mux.HandleFunc("GET /{$}", index) - mux.HandleFunc("GET /games/{$}", gamesIndex) + mux.HandleFunc("GET /games/", gamesIndex) mux.HandleFunc("GET /games/pong", pong) mux.HandleFunc("GET /games/rms", stallmanShooter) mux.HandleFunc("GET /qr.png", qrHandler) @@ -1304,17 +945,12 @@ func main() { mux.HandleFunc("POST /login", login) mux.HandleFunc("GET /notepad", notepad) mux.HandleFunc("POST /notepad", autosave) - mux.HandleFunc("GET /cmd/{$}", cmdIndex) + mux.HandleFunc("GET /cmd/", cmdIndex) mux.HandleFunc("GET /cmd/{cmd}", cmd) - mux.HandleFunc("GET /pkg/{$}", pkgIndex) + mux.HandleFunc("GET /pkg/", pkgIndex) mux.HandleFunc("GET /pkg/{pkg}", pkg) mux.Handle("GET /code/", http.StripPrefix("/code/", http.FileServer(http.Dir(*codeDir)))) mux.Handle("GET /nezuko/", &nezukoHandler{http.StripPrefix("/nezuko/", http.FileServer(http.Dir(*nezuko)))}) - mux.Handle("GET /mart/{$}", &metArtIndexHandler{stmts}) - mux.Handle("GET /mart/{girl}/{$}", &metArtGirlHandler{stmts}) - mux.Handle("GET /mart/{girl}/{set}", &metArtSetHandler{stmts}) - mux.Handle("GET /mart/photos/{photo}", &metArtPhotoHandler{stmts}) - mux.Handle("GET /mart/photos/preview/{photo}", &metArtGirlPreviewHandler{stmts}) mux.HandleFunc("GET /static/", static) mux.HandleFunc("GET /favicon.ico", favicon) apiMux := http.NewServeMux() |