feat(template): add Fileserver for download

Reviewed-on: #8

cmd/svg-templater/main.go

@@ -0,0 +1,23 @@
+package main
+
+import (
+	"log"
+
+	"tomatentum.net/svg-templater/internal/command"
+	"tomatentum.net/svg-templater/internal/database"
+	"tomatentum.net/svg-templater/pkg/format"
+)
+
+func main() {
+	if !format.CheckInkscape() {
+		panic("Inkscape not found")
+	}
+	if err := database.OpenSQLite(); err != nil {
+		log.Fatal("Failed opening DB:\n", err)
+		return
+	}
+	defer database.Close()
+	database.InitDB()
+	command.PrepareCommandLine()
+	command.HandleCommandline()
+}

go.mod

@@ -1,3 +1,16 @@
-module svg-templater
+module tomatentum.net/svg-templater
 
 go 1.24.9
+
+require (
+	github.com/dustin/go-humanize v1.0.1 // indirect
+	github.com/glebarez/go-sqlite v1.22.0 // indirect
+	github.com/google/uuid v1.5.0 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
+	golang.org/x/sys v0.15.0 // indirect
+	modernc.org/libc v1.37.6 // indirect
+	modernc.org/mathutil v1.6.0 // indirect
+	modernc.org/memory v1.7.2 // indirect
+	modernc.org/sqlite v1.28.0 // indirect
+)

go.sum

@@ -0,0 +1,21 @@
+github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
+github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
+github.com/glebarez/go-sqlite v1.22.0 h1:uAcMJhaA6r3LHMTFgP0SifzgXg46yJkgxqyuyec+ruQ=
+github.com/glebarez/go-sqlite v1.22.0/go.mod h1:PlBIdHe0+aUEFn+r2/uthrWq4FxbzugL0L8Li6yQJbc=
+github.com/google/uuid v1.5.0 h1:1p67kYwdtXjb0gL0BPiP1Av9wiZPo5A8z2cWkTZ+eyU=
+github.com/google/uuid v1.5.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
+github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.15.0 h1:h48lPFYpsTvQJZF4EKyI4aLHaev3CxivZmv7yZig9pc=
+golang.org/x/sys v0.15.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+modernc.org/libc v1.37.6 h1:orZH3c5wmhIQFTXF+Nt+eeauyd+ZIt2BX6ARe+kD+aw=
+modernc.org/libc v1.37.6/go.mod h1:YAXkAZ8ktnkCKaN9sw/UDeUVkGYJ/YquGO4FTi5nmHE=
+modernc.org/mathutil v1.6.0 h1:fRe9+AmYlaej+64JsEEhoWuAYBkOtQiMEU7n/XgfYi4=
+modernc.org/mathutil v1.6.0/go.mod h1:Ui5Q9q1TR2gFm0AQRqQUaBWFLAhQpCwNcuhBOSedWPo=
+modernc.org/memory v1.7.2 h1:Klh90S215mmH8c9gO98QxQFsY+W451E8AnzjoE2ee1E=
+modernc.org/memory v1.7.2/go.mod h1:NO4NVCQy0N7ln+T9ngWqOQfi7ley4vpwvARR+Hjw95E=
+modernc.org/sqlite v1.28.0 h1:Zx+LyDDmXczNnEQdvPuEfcFVA2ZPyaD7UCZDjef3BHQ=
+modernc.org/sqlite v1.28.0/go.mod h1:Qxpazz0zH8Z1xCFyi5GSL3FzbtZ3fvbjmywNogldEW0=

internal/command/commandline.go

@@ -0,0 +1,37 @@
+package command
+
+import (
+	"flag"
+
+	"tomatentum.net/svg-templater/internal/server"
+	"tomatentum.net/svg-templater/pkg/svg"
+)
+
+var (
+	help              bool
+	generateTokenFlag bool
+	deleteTokenFlag   bool
+	datapath          string
+)
+
+func PrepareCommandLine() {
+	flag.BoolVar(&help, "help", false, "Displays the help String")
+	flag.BoolVar(&generateTokenFlag, "tokengen", false, "Generate token with name")
+	flag.BoolVar(&deleteTokenFlag, "tokendel", false, "Delete token with name")
+	flag.StringVar(&datapath, "data", "/var/lib/svg-templater", "Override data directory")
+}
+
+func HandleCommandline() {
+	flag.Parse()
+	if generateTokenFlag {
+		GenerateTokenCommand()
+	} else if deleteTokenFlag {
+		DeleteTokenCommand()
+	} else if help {
+		flag.PrintDefaults()
+	} else {
+		svg.Storage = svg.NewFileStorage(datapath, "public")
+		server.PrepareHTTP()
+		server.Start()
+	}
+}

internal/command/tokencommands.go

@@ -0,0 +1,34 @@
+package command
+
+import (
+	"flag"
+	"log"
+
+	"tomatentum.net/svg-templater/pkg/auth"
+)
+
+func GenerateTokenCommand() {
+	name := flag.Arg(0)
+	if name == "" {
+		log.Fatal("No Token Name was provided!")
+	}
+
+	token, err := auth.CreateToken(name)
+	if err != nil {
+		log.Fatal("Failed to generate token:\n", err)
+	}
+	log.Println(token)
+}
+
+func DeleteTokenCommand() {
+	name := flag.Arg(0)
+	if name == "" {
+		log.Fatal("No Token Name was provided!")
+	}
+
+	err := auth.DeleteToken(name)
+	if err != nil {
+		log.Fatal("Failed to delete token:\n", err)
+	}
+	log.Println("Deleted token " + name)
+}

internal/database/database.go

@@ -0,0 +1,35 @@
+package database
+
+import (
+	"database/sql"
+	"log"
+
+	_ "github.com/glebarez/go-sqlite"
+)
+
+const FILENAME string = "storage.db"
+
+var database *sql.DB
+
+func OpenSQLite() error {
+	db, err := sql.Open("sqlite", FILENAME)
+	if err != nil {
+		return err
+	}
+	log.Println("Successfully connected to SQLite Database")
+	database = db
+	return nil
+}
+
+func InitDB() {
+	_, err := database.Exec(TOKENTABLECREATE)
+	_, err = database.Exec(SVGTABLECREATE)
+	if err != nil {
+		log.Fatal("Failed to init database:\n", err)
+	}
+	log.Println("Successfully initiated the Database!")
+}
+
+func Close() error {
+	return database.Close()
+}

internal/database/svgdb.go

@@ -0,0 +1,28 @@
+package database
+
+import (
+	"encoding/json"
+
+	"tomatentum.net/svg-templater/pkg/svg"
+)
+
+const SVGTABLECREATE string = `
+	CREATE TABLE IF NOT EXISTS svg (
+		name varchar(16) PRIMARY KEY NOT NULL,
+		templatekeys longtext NOT NULL
+	);`
+
+const INSERTSVGSQL string = "INSERT INTO svg VALUES (?, ?);"
+
+func InsertSVG(data *svg.TemplateData) error {
+	json, err := json.Marshal(data.TemplateKeys)
+	if err != nil {
+		return err
+	}
+
+	if _, err := database.Exec(INSERTSVGSQL, data.Id, string(json)); err != nil {
+		return err
+	}
+
+	return nil
+}

internal/database/tokendb.go

@@ -0,0 +1,55 @@
+package database
+
+import (
+	"maps"
+)
+
+const TOKENTABLECREATE string = `
+	CREATE TABLE IF NOT EXISTS tokens (
+		name varchar(32) PRIMARY KEY NOT NULL,
+		token varchar(32) NOT NULL
+	);`
+
+const INSERTTOKENSQL string = "INSERT INTO tokens VALUES (?, ?);"
+const DELETETOKENSQL string = "DELETE FROM tokens WHERE name = ?;"
+const QUERYTOKENSQL string = "SELECT * FROM tokens;"
+
+var tokenCache map[string]string = make(map[string]string)
+
+func InsertToken(name string, token string) error {
+	_, err := database.Exec(INSERTTOKENSQL, name, token)
+	return err
+}
+
+func DeleteToken(name string) error {
+	_, err := database.Exec(DELETETOKENSQL, name)
+	return err
+}
+
+func ValidateTokenCache(token string) (bool, error) {
+	for v := range maps.Values(tokenCache) {
+		if token == v {
+			return true, nil
+		}
+	}
+	return ValidateToken(token)
+}
+
+func ValidateToken(token string) (bool, error) {
+	result, err := database.Query(QUERYTOKENSQL)
+	if err != nil {
+		return false, err
+	}
+	defer result.Close()
+
+	var name, foundToken string
+	for result.Next() {
+		result.Scan(&name, &foundToken)
+		tokenCache[name] = foundToken
+		if token == foundToken {
+			return true, nil
+		}
+	}
+	err = result.Err()
+	return false, err
+}

internal/routes/upload.go

@@ -0,0 +1,49 @@
+package routes
+
+import (
+	"bytes"
+	"encoding/json"
+	"io"
+	"log"
+	"net/http"
+
+	"tomatentum.net/svg-templater/pkg/svg/actions"
+)
+
+func CreateSVG(writer http.ResponseWriter, r http.Request) {
+
+	contentType := r.Header.Get("Content-Type")
+	if contentType != "image/svg+xml" {
+		http.Error(writer, "Incorrect Content-Type. Needs image/svg+xml.", http.StatusUnsupportedMediaType)
+		return
+	}
+
+	readsvg, err := io.ReadAll(r.Body)
+	if err != nil {
+		http.Error(writer, err.Error(), http.StatusInternalServerError)
+		log.Println("Error while reading Body\n", err)
+		return
+	}
+
+	if ok, err := validateSVG(readsvg); err != nil || !ok {
+		http.Error(writer, err.Error(), http.StatusUnsupportedMediaType)
+		log.Println("Wrong Media Type was uploaded\n", err)
+		return
+	}
+
+	data, err := actions.Create(readsvg)
+
+	if err != nil {
+		http.Error(writer, err.Error(), http.StatusInternalServerError)
+		log.Println("Failed creating SVG template\n", err)
+		return
+	}
+
+	writer.Header().Add("Content-Type", "application/json")
+	json.NewEncoder(writer).Encode(data)
+	log.Println("Created SVG Template " + data.Id)
+}
+
+func validateSVG(svgbuf []byte) (bool, error) {
+	return bytes.Contains(svgbuf[:512], []byte("<svg")), nil
+}

internal/server/http.go

@@ -0,0 +1,47 @@
+package server
+
+import (
+	"fmt"
+	"log"
+	"net/http"
+	"path/filepath"
+
+	"tomatentum.net/svg-templater/internal/routes"
+	"tomatentum.net/svg-templater/pkg/auth"
+	"tomatentum.net/svg-templater/pkg/svg"
+)
+
+func PrepareHTTP() {
+	registerAuthorizedFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		fmt.Fprintln(w, "You are authorized!")
+	})
+	registerAuthorizedFunc("/svg/", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method != "POST" {
+			http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+			return
+		}
+		routes.CreateSVG(w, *r)
+	})
+
+	registerAuthorized("/public/", http.StripPrefix("/public/", http.FileServer(svg.Storage.GetPublicDir())))
+}
+
+func Start() {
+	log.Println("Starting http server on :3000")
+	if err := http.ListenAndServe(":3000", nil); err != nil {
+		panic(err)
+	}
+}
+
+func registerAuthorized(path string, handler http.Handler) {
+	http.HandleFunc(path, auth.AuthMiddleware(handler))
+	log.Println("Registered authorized handler for", path)
+}
+
+func registerAuthorizedFunc(path string, f func(w http.ResponseWriter, r *http.Request)) {
+	registerAuthorized(path, http.HandlerFunc(f))
+}
+
+func GetPublicPath(path string) string {
+	return filepath.Join("public", path)
+}

main.go

@@ -1,5 +0,0 @@
-package main
-
-func main() {
-	println("Hello World")
-}

pkg/auth/auth.go

@@ -0,0 +1,28 @@
+package auth
+
+import (
+	"crypto/rand"
+	"encoding/hex"
+	"log"
+
+	"tomatentum.net/svg-templater/internal/database"
+)
+
+func generateToken() string {
+	token := make([]byte, 32)
+	if _, err := rand.Read(token); err != nil {
+		log.Fatal("Failed to generate Token:\n", err)
+		return ""
+	}
+	return hex.EncodeToString(token)
+}
+
+func CreateToken(name string) (string, error) {
+	token := generateToken()
+	err := database.InsertToken(name, token)
+	return token, err
+}
+
+func DeleteToken(name string) error {
+	return database.DeleteToken(name)
+}

pkg/auth/authmiddleware.go

@@ -0,0 +1,40 @@
+package auth
+
+import (
+	"log"
+	"net/http"
+	"strings"
+
+	"tomatentum.net/svg-templater/internal/database"
+)
+
+func AuthMiddleware(next http.Handler) http.HandlerFunc {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		valid, err := validateAuthHeader(r)
+		if err != nil {
+			http.Error(w, "Internal Server Error", http.StatusInternalServerError)
+			log.Println("Database Query for token validation failed.\n", r)
+			return
+		}
+
+		if valid {
+			next.ServeHTTP(w, r)
+		} else {
+			http.Error(w, "Unauthorized", http.StatusUnauthorized)
+		}
+	})
+}
+
+// Accepts ("Authorization": "BEARER <TOKEN>"), returns database error if query failed
+func validateAuthHeader(r *http.Request) (bool, error) {
+	header := strings.Split(r.Header.Get("Authorization"), " ")
+	if len(header) < 2 {
+		return false, nil
+	}
+	token := header[1]
+	if len(strings.TrimSpace(token)) < 1 {
+		return false, nil
+	}
+
+	return database.ValidateTokenCache(token)
+}

pkg/format/formatconverter.go

@@ -0,0 +1,97 @@
+package format
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strconv"
+	"strings"
+
+	"tomatentum.net/svg-templater/pkg/svg"
+)
+
+type ConversionParameters struct {
+	Format        string
+	Width, Height int
+}
+
+func ConvertByte(svgblob []byte, param ConversionParameters) ([]byte, error) {
+	reader, err := ConvertReader(svgblob, param)
+	if err != nil {
+		return nil, err
+	}
+	defer reader.Close()
+
+	result, err := io.ReadAll(reader)
+	if err != nil {
+		return nil, err
+	}
+	return result, nil
+}
+
+func ConvertReader(svgblob []byte, param ConversionParameters) (io.ReadCloser, error) {
+	if param.Format == "svg" {
+		return io.NopCloser(bytes.NewReader(svgblob)), nil
+	}
+	file, err := svg.CreateTemp(bytes.NewReader(svgblob), "svg")
+	if err != nil {
+		return nil, err
+	}
+	defer os.Remove(file)
+
+	return runCommand(file, param)
+}
+
+func runCommand(input string, param ConversionParameters) (io.ReadCloser, error) {
+	var args []string
+
+	var outExt string
+	switch param.Format {
+	case "png":
+		args = append(args, "--export-type=png")
+		outExt = ".png"
+	case "jpg":
+		args = append(args, "--export-type=png") // inkscape doesn't export jpg directly
+		outExt = ".png"
+	case "pdf":
+		args = append(args, "--export-type=pdf")
+		outExt = ".pdf"
+	default:
+		return nil, fmt.Errorf("format not supported: %s", param.Format)
+	}
+
+	if param.Width > 0 {
+		args = append(args, "-w", strconv.Itoa(param.Width))
+	}
+	if param.Height > 0 {
+		args = append(args, "-h", strconv.Itoa(param.Height))
+	}
+
+	outFile := strings.TrimSuffix(input, filepath.Ext(input)) + outExt
+	args = append(args, "-o", outFile)
+	args = append(args, input)
+
+	cmd := exec.Command("/opt/homebrew/bin/inkscape", args...)
+	cmd.Dir = svg.TempDir
+
+	out, err := cmd.CombinedOutput()
+
+	if err != nil {
+		return nil, fmt.Errorf("inkscape failed: %w: %s", err, string(out))
+	}
+
+	file, err := os.Open(outFile)
+	if err != nil {
+		return nil, err
+	}
+
+	return file, nil
+}
+
+func CheckInkscape() bool {
+	_, err := exec.LookPath("inkscape")
+	return err == nil
+}

pkg/svg/actions/upload.go

@@ -0,0 +1,50 @@
+package actions
+
+import (
+	"bytes"
+	"crypto/rand"
+	"encoding/hex"
+	"log"
+	"regexp"
+
+	"tomatentum.net/svg-templater/internal/database"
+	"tomatentum.net/svg-templater/pkg/svg"
+)
+
+func Create(svgbuf []byte) (svg.TemplateData, error) {
+	data := svg.TemplateData{Id: generateId(), TemplateKeys: nil}
+	populateKeys(&data, svgbuf)
+
+	_, err := svg.Storage.Create(data.Id, bytes.NewReader(svgbuf))
+	if err != nil {
+		return svg.TemplateData{}, err
+	}
+
+	if err := database.InsertSVG(&data); err != nil {
+		return svg.TemplateData{}, err
+	}
+
+	return data, nil
+}
+
+func populateKeys(data *svg.TemplateData, svg []byte) {
+	regex := regexp.MustCompile(`\{\{\s*(.*?)\s*\}\}`)
+	result := regex.FindAllSubmatch(svg, -1)
+	templateKeys := make([]string, len(result))
+
+	for i, matches := range result {
+		varname := matches[1] // first capture group
+		templateKeys[i] = string(varname)
+	}
+	data.TemplateKeys = templateKeys
+	log.Println("Found keys:\n", templateKeys)
+}
+
+func generateId() string {
+	token := make([]byte, 16)
+	if _, err := rand.Read(token); err != nil {
+		log.Fatal("Failed to generate Token:\n", err)
+		return ""
+	}
+	return hex.EncodeToString(token)
+}

pkg/svg/storage.go

@@ -0,0 +1,99 @@
+package svg
+
+import (
+	"io"
+	"log"
+	"net/http"
+	"os"
+	"path/filepath"
+)
+
+type SvgStorage interface {
+	Create(id string, svg io.Reader) (string, error)
+	Get(id string) (io.ReadCloser, error)
+	CreatePublic(data io.Reader, filetype string) (string, error)
+	GetPublic(path string) (io.ReadCloser, error)
+	GetPublicDir() http.Dir
+}
+
+var _ SvgStorage = FileSvgStorage{}
+var TempDir string = ""
+
+type FileSvgStorage struct {
+	basepath, publicSubPath string
+}
+
+func NewFileStorage(path, publicSubPath string) *FileSvgStorage {
+	err := os.MkdirAll(path, 0755)
+	if err != nil {
+		panic(err)
+	}
+	log.Println("Initialized file storage handler")
+	return &FileSvgStorage{path, publicSubPath}
+}
+
+func (f FileSvgStorage) Create(id string, svg io.Reader) (string, error) {
+	file, err := os.Create(filepath.Join(f.basepath, id+".svg"))
+
+	if err != nil {
+		return "", err
+	}
+	defer file.Close()
+
+	_, err = io.Copy(file, svg)
+	if err != nil {
+		return "", err
+	}
+
+	log.Println("Created File: ", file.Name())
+	return file.Name(), nil
+}
+
+func (f FileSvgStorage) Get(id string) (io.ReadCloser, error) {
+	file, err := os.Open(filepath.Join(f.basepath, id+".svg"))
+	if err != nil {
+		return nil, err
+	}
+	return file, nil
+}
+
+func (f FileSvgStorage) CreatePublic(data io.Reader, filetype string) (string, error) {
+	path := filepath.Join(f.basepath, f.publicSubPath)
+	if err := os.Mkdir(path, 0755); err != nil {
+		return "", err
+	}
+
+	file, err := os.CreateTemp(path, "*."+filetype)
+	if err != nil {
+		return "", err
+	}
+	defer file.Close()
+
+	return file.Name(), nil
+}
+
+func (f FileSvgStorage) GetPublic(path string) (io.ReadCloser, error) {
+	file, err := os.Open(filepath.Join(f.basepath, f.publicSubPath, path))
+	if err != nil {
+		return nil, err
+	}
+	return file, nil
+}
+
+func (f FileSvgStorage) GetPublicDir() http.Dir {
+	return http.Dir(filepath.Join(f.basepath, f.publicSubPath))
+}
+
+func CreateTemp(data io.Reader, filetype string) (string, error) {
+	file, err := os.CreateTemp(TempDir, "*."+filetype)
+	if err != nil {
+		return "", err
+	}
+	defer file.Close()
+	_, err = io.Copy(file, data)
+	if err != nil {
+		return "", err
+	}
+
+	return file.Name(), nil
+}

pkg/svg/templates.go

@@ -0,0 +1,8 @@
+package svg
+
+type TemplateData struct {
+	Id           string
+	TemplateKeys []string
+}
+
+var Storage SvgStorage