74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package actions
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"regexp"
|
|
"strconv"
|
|
|
|
"tomatentum.net/svg-templater/internal/database"
|
|
"tomatentum.net/svg-templater/pkg/svg"
|
|
)
|
|
|
|
const FILEFORMAT string = "%s-%d"
|
|
|
|
func AddPage(id string, svgbuf []byte) (svg.TemplatePage, error) {
|
|
data := svg.TemplatePage{TemplateId: id}
|
|
populateKeys(&data, svgbuf)
|
|
|
|
page, err := database.InsertSVGPage(&data)
|
|
|
|
data.Page = page
|
|
|
|
if err != nil {
|
|
return svg.TemplatePage{}, err
|
|
}
|
|
|
|
_, err = svg.Storage.Create(fmt.Sprintf(FILEFORMAT, id, page), bytes.NewReader(svgbuf))
|
|
if err != nil {
|
|
return svg.TemplatePage{}, err
|
|
}
|
|
|
|
log.Println("Created SVG Template " + data.TemplateId + " Page " + strconv.Itoa(page))
|
|
return data, nil
|
|
}
|
|
|
|
func GetPage(id string, page int) ([]byte, error) {
|
|
file, err := svg.Storage.Get(fmt.Sprintf(FILEFORMAT, id, page))
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer file.Close()
|
|
|
|
return io.ReadAll(file)
|
|
}
|
|
|
|
func DeletePage(id string, page int) (bool, error) {
|
|
if success, err := database.DeleteSVGPage(id, page); err != nil || !success {
|
|
return success, err
|
|
}
|
|
|
|
if err := svg.Storage.Delete(fmt.Sprintf(FILEFORMAT, id, page)); err != nil {
|
|
return false, err
|
|
}
|
|
log.Println("Deleted SVG Template " + id + " page " + strconv.Itoa(page))
|
|
return true, nil
|
|
}
|
|
|
|
func populateKeys(data *svg.TemplatePage, svgblob []byte) {
|
|
regex := regexp.MustCompile(svg.KeyRegex)
|
|
result := regex.FindAllSubmatch(svgblob, -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)
|
|
}
|