75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
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 (?, ?);"
|
|
const GETSPECIFICSVGSQL string = "SELECT * FROM svg WHERE name = ?;"
|
|
const GETSVGSQL string = "SELECT * FROM svg;"
|
|
|
|
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
|
|
}
|
|
|
|
func GetSVG() ([]svg.TemplateData, error) {
|
|
result, err := database.Query(GETSVGSQL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer result.Close()
|
|
templates := make([]svg.TemplateData, 0)
|
|
for result.Next() {
|
|
var (
|
|
id string
|
|
keysjson []byte
|
|
keys []string
|
|
)
|
|
if err := result.Scan(&id, &keysjson); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := json.Unmarshal(keysjson, &keys); err != nil {
|
|
return nil, err
|
|
}
|
|
templates = append(templates, svg.TemplateData{Id: id, TemplateKeys: keys})
|
|
}
|
|
|
|
if err := result.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
return templates, nil
|
|
}
|
|
|
|
func GetSpecificSVG(id string) (svg.TemplateData, error) {
|
|
result := database.QueryRow(GETSPECIFICSVGSQL, id)
|
|
|
|
var (
|
|
keysjson []byte
|
|
keys []string
|
|
)
|
|
if err := result.Scan(&id, &keysjson); err != nil {
|
|
return svg.TemplateData{}, err
|
|
}
|
|
if err := json.Unmarshal(keysjson, &keys); err != nil {
|
|
return svg.TemplateData{}, err
|
|
}
|
|
return svg.TemplateData{Id: id, TemplateKeys: keys}, nil
|
|
}
|