Most of the server logic

Closes #6.
Closes #5.
Closes #4.

Author: foxcpp

config.go

@@ -25,6 +25,8 @@ type Config struct {
 	DB DBConfig
 	DownloadAuth AuthConfig
 	UploadAuth AuthConfig
+	StorageDir string
+	HTTPSUpstream bool
 }
 
 var Default Config

db.go

@@ -1,11 +1,144 @@
 package filedrop
 
-import "database/sql"
+import (
+	"database/sql"
+	"strings"
+	"time"
+)
 
 type db struct {
 	*sql.DB
+
+	addFile *sql.Stmt
+	remFile *sql.Stmt
+	getFile *sql.Stmt
+
+	addUse *sql.Stmt
+	shouldDelete *sql.Stmt
 }
 
 func openDB(driver, dsn string) (*db, error) {
-	return &db{}, nil
+	if driver == "sqlite3" {
+		// We apply some tricks for SQLite to avoid "database is locked" errors.
+
+		if !strings.HasPrefix(dsn, "file:") {
+			dsn = "file:" + dsn
+		}
+		if !strings.Contains(dsn, "?") {
+			dsn = dsn + "?"
+		}
+		dsn = dsn + "cache=shared&_journal=WAL&_busy_timeout=5000"
+	}
+
+	db := new(db)
+	var err error
+	db.DB, err = sql.Open(driver, dsn)
+	if err != nil {
+		return nil, err
+	}
+
+
+	if driver == "sqlite3" {
+		// Also some optimizations for SQLite to make it FAA-A-A-AST.
+		db.Exec(`PRAGMA foreign_keys = ON`)
+		db.Exec(`PRAGMA auto_vacuum = INCREMENTAL`)
+		db.Exec(`PRAGMA journal_mode = WAL`)
+		db.Exec(`PRAGMA defer_foreign_keys = ON`)
+		db.Exec(`PRAGMA synchronous = NORMAL`)
+		db.Exec(`PRAGMA temp_store = MEMORY`)
+		db.Exec(`PRAGMA cache_size = 5000`)
+	}
+
+	if err := db.initSchema(); err != nil {
+		panic(err)
+	}
+	if err := db.initStmts(); err != nil {
+		panic(err)
+	}
+	return db, nil
 }
+
+func (db *db) initSchema() error {
+	_, err := db.Exec(`CREATE TABLE IF NOT EXISTS filedrop (
+		uuid TEXT PRIMARY KEY NOT NULL,
+		uses INTEGER NOT NULL DEFAULT 0,
+		maxUses INTEGER,
+		storeUntil INTEGER
+	)`)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (db *db) initStmts() error {
+	var err error
+	db.addFile, err = db.Prepare(`INSERT INTO filedrop(uuid, maxUses, storeUntil) VALUES (?, ?, ?)`)
+	if err != nil {
+		return err
+	}
+	db.remFile, err = db.Prepare(`DELETE FROM filedrop WHERE uuid = ?`)
+	if err != nil {
+		return err
+	}
+	db.getFile, err = db.Prepare(`SELECT uses, maxUses, storeUntil FROM filedrop WHERE uuid = ?`)
+	if err != nil {
+		return err
+	}
+	db.shouldDelete, err = db.Prepare(`SELECT exists(SELECT uuid FROM filedrop WHERE uuid = ? AND (storeUntil < ? OR maxUses == uses))`)
+	if err != nil {
+		return err
+	}
+	db.addUse, err = db.Prepare(`UPDATE filedrop SET uses = (SELECT uses+1 FROM filedrop WHERE uuid = ?) WHERE uuid = ?`)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (db *db) AddFile(tx *sql.Tx, uuid string, maxUses uint, storeUntil time.Time) error {
+	maxUsesN := sql.NullInt64{Int64: int64(maxUses), Valid: maxUses != 0}
+	storeUntilN := sql.NullInt64{Int64: storeUntil.Unix(), Valid: !storeUntil.IsZero()}
+
+	if tx != nil {
+		_, err := tx.Stmt(db.addFile).Exec(uuid, maxUsesN, storeUntilN)
+		return err
+	} else {
+		_, err := db.addFile.Exec(uuid, maxUsesN, storeUntilN)
+		return err
+	}
+}
+
+func (db *db) RemoveFile(tx *sql.Tx, uuid string) error {
+	if tx != nil {
+		_, err := tx.Stmt(db.remFile).Exec(uuid)
+		return err
+	} else {
+		_, err := db.remFile.Exec(uuid)
+		return err
+	}
+}
+
+func (db *db) ShouldDelete(tx *sql.Tx, uuid string) bool {
+	var row *sql.Row
+	if tx != nil {
+		row = tx.Stmt(db.shouldDelete).QueryRow(uuid, time.Now().Unix())
+	} else {
+		row = db.shouldDelete.QueryRow(uuid, time.Now().Unix())
+	}
+	res := 0
+	if err := row.Scan(&res); err != nil {
+		return false
+	}
+	return res == 1
+}
+
+func (db *db) AddUse(tx *sql.Tx, uuid string) error {
+	if tx != nil {
+		_, err := tx.Stmt(db.addUse).Exec(uuid, uuid)
+		return err
+	} else {
+		_, err := db.addUse.Exec(uuid, uuid)
+		return err
+	}
+}

go.mod

@@ -1,3 +1,7 @@
 module github.com/foxcpp/filedrop
 
-require github.com/mattn/go-sqlite3 v1.9.0
+require (
+	github.com/mattn/go-sqlite3 v1.9.0
+	github.com/pkg/errors v0.8.0
+	github.com/satori/go.uuid v1.2.0
+)

go.sum

@@ -1,2 +1,6 @@
 github.com/mattn/go-sqlite3 v1.9.0 h1:pDRiWfl+++eC2FEFRy6jXmQlvp4Yh3z1MJKg4UeYM/4=
 github.com/mattn/go-sqlite3 v1.9.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc=
+github.com/pkg/errors v0.8.0 h1:WdK/asTD0HN+q6hsWO3/vpuAkAr+tw6aNJNDFFf0+qw=
+github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
+github.com/satori/go.uuid v1.2.0 h1:0uYX9dsZ2yD7q2RtLRtPSdGDWzjeM3TbMJP9utgA0ww=
+github.com/satori/go.uuid v1.2.0/go.mod h1:dA0hQrYB0VpLJoorglMZABFdXlWrHn1NEOzdhQKdks0=