This commit is contained in:
+20
-30
@@ -1,32 +1,23 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"log"
|
||||
|
||||
"tomatentum.net/svg-templater/pkg/svg"
|
||||
)
|
||||
|
||||
const SVGTABLECREATE string = `
|
||||
CREATE TABLE IF NOT EXISTS svg (
|
||||
id varchar(16) PRIMARY KEY NOT NULL,
|
||||
name varchar(32),
|
||||
templatekeys longtext NOT NULL
|
||||
name varchar(32)
|
||||
);`
|
||||
|
||||
const INSERTSVGSQL string = "INSERT INTO svg VALUES (?, ?, ?);"
|
||||
const INSERTSVGSQL string = "INSERT INTO svg VALUES (?, ?);"
|
||||
const GETSPECIFICSVGSQL string = "SELECT * FROM svg WHERE id = ?;"
|
||||
const GETSVGSQL string = "SELECT * FROM svg;"
|
||||
const DELETESVGSQL string = "DELETE FROM svg WHERE id = ?;"
|
||||
const RENAMESVGSQL string = "UPDATE svg SET name = ? WHERE id = ?;"
|
||||
|
||||
func InsertSVG(data *svg.TemplateData) error {
|
||||
json, err := json.Marshal(data.TemplateKeys)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if _, err := database.Exec(INSERTSVGSQL, data.Id, data.Name, string(json)); err != nil {
|
||||
if _, err := database.Exec(INSERTSVGSQL, data.Id, data.Name); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -42,18 +33,19 @@ func GetSVG() ([]svg.TemplateData, error) {
|
||||
templates := make([]svg.TemplateData, 0)
|
||||
for result.Next() {
|
||||
var (
|
||||
id string
|
||||
name string
|
||||
keysjson []byte
|
||||
keys []string
|
||||
id string
|
||||
name string
|
||||
)
|
||||
if err := result.Scan(&id, &name, &keysjson); err != nil {
|
||||
if err := result.Scan(&id, &name); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := json.Unmarshal(keysjson, &keys); err != nil {
|
||||
pages, err := GetSVGPages(id)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
templates = append(templates, svg.TemplateData{Id: id, Name: name, TemplateKeys: keys})
|
||||
|
||||
templates = append(templates, svg.TemplateData{Id: id, Name: name, Pages: pages})
|
||||
}
|
||||
|
||||
if err := result.Err(); err != nil {
|
||||
@@ -65,18 +57,17 @@ func GetSVG() ([]svg.TemplateData, error) {
|
||||
func GetSpecificSVG(id string) (svg.TemplateData, error) {
|
||||
result := database.QueryRow(GETSPECIFICSVGSQL, id)
|
||||
|
||||
var (
|
||||
name string
|
||||
keysjson []byte
|
||||
keys []string
|
||||
)
|
||||
if err := result.Scan(&id, &name, &keysjson); err != nil {
|
||||
var name string
|
||||
if err := result.Scan(&id, &name); err != nil {
|
||||
return svg.TemplateData{}, err
|
||||
}
|
||||
if err := json.Unmarshal(keysjson, &keys); err != nil {
|
||||
|
||||
pages, err := GetSVGPages(id)
|
||||
|
||||
if err != nil {
|
||||
return svg.TemplateData{}, err
|
||||
}
|
||||
return svg.TemplateData{Id: id, Name: name, TemplateKeys: keys}, nil
|
||||
return svg.TemplateData{Id: id, Name: name, Pages: pages}, nil
|
||||
}
|
||||
|
||||
func DeleteSvg(id string) (bool, error) {
|
||||
@@ -92,12 +83,11 @@ func DeleteSvg(id string) (bool, error) {
|
||||
if num == 0 {
|
||||
return false, nil
|
||||
}
|
||||
return true, nil
|
||||
return DeleteAllSVGPages(id)
|
||||
}
|
||||
|
||||
func RenameSvg(id string, name string) error {
|
||||
res, err := database.Exec(RENAMESVGSQL, name, id)
|
||||
log.Println(id, name, res)
|
||||
_, err := database.Exec(RENAMESVGSQL, name, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user