From 95c8722dd862e03e3a8bf3e8ca080aab9d8c8e95 Mon Sep 17 00:00:00 2001 From: Přemysl Eric Janouch Date: Sun, 10 Dec 2023 07:46:47 +0100 Subject: Rearrange code --- main.go | 214 ++++++++++++++++++++++++++++++++++------------------------------ 1 file changed, 115 insertions(+), 99 deletions(-) diff --git a/main.go b/main.go index a57f3af..1ef03ab 100644 --- a/main.go +++ b/main.go @@ -73,62 +73,63 @@ func dbCollectStrings(query string) ([]string, error) { return result, nil } -type directoryManager struct { - cache map[string]int64 // Unix-style paths to directory.id -} - -func (dm *directoryManager) uncachedIDForPath( - tx *sql.Tx, path []string) (int64, error) { - var parent sql.NullInt64 - for _, name := range path { - if err := tx.QueryRow( - `SELECT id FROM directory WHERE name = ? AND parent IS ?`, - name, parent).Scan(&parent); err == nil { - continue - } else if !errors.Is(err, sql.ErrNoRows) { - return 0, err - } +// --- Initialization ---------------------------------------------------------- - if result, err := tx.Exec( - `INSERT INTO directory(name, parent) VALUES (?, ?)`, - name, parent); err != nil { - return 0, err - } else if id, err := result.LastInsertId(); err != nil { - return 0, err - } else { - parent = sql.NullInt64{Int64: id, Valid: true} - } +// cmdInit initializes a "gallery directory" that contains gallery.sqlite, +// images, thumbs. +func cmdInit(args []string) error { + if len(args) != 1 { + return errors.New("usage: GD") } - return parent.Int64, nil -} -func (dm *directoryManager) IDForDirectoryPath( - tx *sql.Tx, path string) (int64, error) { - // Relative paths could be handled differently, - // but right now, they're assumed to start at the root. - path = filepath.ToSlash(filepath.Clean(path)) - list := strings.Split(path, "/") - if len(list) > 1 && list[0] == "" { - list = list[1:] + if err := openDB(args[0]); err != nil { + return err } - if len(list) == 0 { - return 0, nil + if _, err := db.Exec(initializeSQL); err != nil { + return err } - if dm.cache == nil { - dm.cache = make(map[string]int64) - } else if id, ok := dm.cache[path]; ok { - return id, nil + // XXX: There's technically no reason to keep images as symlinks, + // we might just keep absolute paths in the database as well. + if err := os.MkdirAll( + filepath.Join(galleryDirectory, "images"), 0755); err != nil { + return err } - - id, err := dm.uncachedIDForPath(tx, list) - if err != nil { - return 0, err + if err := os.MkdirAll( + filepath.Join(galleryDirectory, "thumbs"), 0755); err != nil { + return err } - dm.cache[path] = id - return id, nil + return nil } +// --- Web --------------------------------------------------------------------- + +var hashRE = regexp.MustCompile(`^/.*?/([0-9a-f]{40})$`) +var staticHandler http.Handler + +var page = template.Must(template.New("/").Parse(` + Gallery + + + + +

{{ .Name }}

+ + + {{ range .Entries }} + + + + {{ end }} + + +`)) + // XXX: This is preliminary. type entry struct { Parent int64 @@ -197,59 +198,6 @@ func dbCollectDirectory(id int64) (directory, error) { return d, rows2.Err() } -// cmdInit initializes a "gallery directory" that contains gallery.sqlite, -// images, thumbs. -func cmdInit(args []string) error { - if len(args) != 1 { - return errors.New("usage: GD") - } - - if err := openDB(args[0]); err != nil { - return err - } - if _, err := db.Exec(initializeSQL); err != nil { - return err - } - - // XXX: There's technically no reason to keep images as symlinks, - // we might just keep absolute paths in the database as well. - if err := os.MkdirAll( - filepath.Join(galleryDirectory, "images"), 0755); err != nil { - return err - } - if err := os.MkdirAll( - filepath.Join(galleryDirectory, "thumbs"), 0755); err != nil { - return err - } - return nil -} - -var hashRE = regexp.MustCompile(`^/.*?/([0-9a-f]{40})$`) -var staticHandler http.Handler - -var page = template.Must(template.New("/").Parse(` - Gallery - - - - -

{{ .Name }}

- - - {{ range .Entries }} - - - - {{ end }} - - -`)) - func handleRequest(w http.ResponseWriter, r *http.Request) { if r.URL.Path != "/" { staticHandler.ServeHTTP(w, r) @@ -321,8 +269,68 @@ func cmdRun(args []string) error { return s.ListenAndServe() } +// --- Import ------------------------------------------------------------------ + +type directoryManager struct { + cache map[string]int64 // Unix-style paths to directory.id +} + +func (dm *directoryManager) uncachedIDForPath( + tx *sql.Tx, path []string) (int64, error) { + var parent sql.NullInt64 + for _, name := range path { + if err := tx.QueryRow( + `SELECT id FROM directory WHERE name = ? AND parent IS ?`, + name, parent).Scan(&parent); err == nil { + continue + } else if !errors.Is(err, sql.ErrNoRows) { + return 0, err + } + + if result, err := tx.Exec( + `INSERT INTO directory(name, parent) VALUES (?, ?)`, + name, parent); err != nil { + return 0, err + } else if id, err := result.LastInsertId(); err != nil { + return 0, err + } else { + parent = sql.NullInt64{Int64: id, Valid: true} + } + } + return parent.Int64, nil +} + +func (dm *directoryManager) IDForDirectoryPath( + tx *sql.Tx, path string) (int64, error) { + // Relative paths could be handled differently, + // but right now, they're assumed to start at the root. + path = filepath.ToSlash(filepath.Clean(path)) + list := strings.Split(path, "/") + if len(list) > 1 && list[0] == "" { + list = list[1:] + } + if len(list) == 0 { + return 0, nil + } + + if dm.cache == nil { + dm.cache = make(map[string]int64) + } else if id, ok := dm.cache[path]; ok { + return id, nil + } + + id, err := dm.uncachedIDForPath(tx, list) + if err != nil { + return 0, err + } + dm.cache[path] = id + return id, nil +} + func isImage(path string) (bool, error) { cmd := exec.Command("xdg-mime", "query", "filetype", path) + + // XXX: Early returns may leak resources. stdout, err := cmd.StdoutPipe() if err != nil { return false, err @@ -481,6 +489,8 @@ func cmdSync(args []string) error { return nil } +// --- Check ------------------------------------------------------------------- + // cmdCheck checks if all files tracked in the DB are accessible. func cmdCheck(args []string) error { if len(args) != 1 { @@ -495,6 +505,8 @@ func cmdCheck(args []string) error { return nil } +// --- Thumbnailing ------------------------------------------------------------ + func makeThumbnail(pathImage, pathThumb string) (int, int, error) { thumbDirname, _ := filepath.Split(pathThumb) if err := os.MkdirAll(thumbDirname, 0755); err != nil { @@ -597,6 +609,8 @@ func cmdThumbnail(args []string) error { return nil } +// --- Perceptual hash --------------------------------------------------------- + func makeDhash(hasher, pathThumb string) (uint64, error) { cmd := exec.Command(hasher, pathThumb) @@ -658,6 +672,8 @@ func cmdDhash(args []string) error { return nil } +// --- Main -------------------------------------------------------------------- + var commands = map[string]struct { handler func(args []string) error }{ -- cgit v1.2.3-70-g09d2