More basic stuff

1 files changed, 57 insertions, 5 deletions

diff --git a/main.go b/main.go
index 240f245..b3a3019 100644
--- a/main.go
+++ b/main.go
@@ -3,11 +3,21 @@ package main
 import (
 	"database/sql"
 	"log"
+	"net"
+	"net/http"
 	"os"
+	"path/filepath"
+	"time"
 
 	_ "github.com/mattn/go-sqlite3"
 )
 
+// TODO: Perhaps maintain the DB, and the gallery directory, as globals.
+
+func openDB(gd string) (*sql.DB, error) {
+	return sql.Open("sqlite3", filepath.Join(gd, "gallery.db"))
+}
+
 // init GD - initialize a "gallery directory" that contains gallery.sqlite,
 // images, thumbs.
 func cmdInit(args []string) {
@@ -16,19 +26,61 @@ func cmdInit(args []string) {
 	}
 
 	gd := args[0]
-
-	db, err := sql.Open("sqlite3", gd+"/foo.db")
+	db, err := openDB(gd)
 	if err != nil {
 		log.Fatalln(err)
 	}
 
-	// TODO: Figure out a go generate to create a Go file to create the db.
-	_ = db
+	if _, err = db.Query(initializeSQL); err != nil {
+		log.Fatalln(err)
+	}
+	if err := os.MkdirAll(filepath.Join(gd, "images"), 0777); err != nil {
+		log.Fatalln(err)
+	}
+	if err := os.MkdirAll(filepath.Join(gd, "thumbs"), 0777); err != nil {
+		log.Fatalln(err)
+	}
 }
 
 // run GD ADDRESS - run a web UI against GD on ADDRESS
 func cmdRun(args []string) {
-	// TODO: Use "public" from CWD for files.
+	if len(args) != 2 {
+		log.Fatalln("usage: GD ADDRESS")
+	}
+
+	gd := args[0]
+	db, err := openDB(gd)
+	if err != nil {
+		log.Fatalln(err)
+	}
+
+	_ = db
+	address := args[1]
+
+	http.Handle("/", http.FileServer(http.Dir("public")))
+	// TODO: These subdirectories should be indirect
+	// (skip the hash subpath, don't require the .webp suffix).
+	http.Handle("/images",
+		http.FileServer(http.Dir(filepath.Join(gd, "images"))))
+	http.Handle("/thumbs",
+		http.FileServer(http.Dir(filepath.Join(gd, "thumbs"))))
+
+	host, port, err := net.SplitHostPort(address)
+	if err != nil {
+		log.Println(err)
+	} else if host == "" {
+		log.Println("http://" + net.JoinHostPort("localhost", port))
+	} else {
+		log.Println("http://" + address)
+	}
+
+	s := &http.Server{
+		Addr:           address,
+		ReadTimeout:    60 * time.Second,
+		WriteTimeout:   60 * time.Second,
+		MaxHeaderBytes: 32 << 10,
+	}
+	log.Fatalln(s.ListenAndServe())
 }
 
 // import GD ROOT... - add files to the "entry" table