|
|
@@ -3,6 +3,8 @@ package helpers
|
|
|
import (
|
|
|
"database/sql"
|
|
|
"encoding/json"
|
|
|
+ "log"
|
|
|
+ "path"
|
|
|
"errors"
|
|
|
"fmt"
|
|
|
"os"
|
|
|
@@ -11,7 +13,6 @@ import (
|
|
|
|
|
|
"git.aetherial.dev/aeth/keiji/pkg/env"
|
|
|
"github.com/google/uuid"
|
|
|
- "github.com/mattn/go-sqlite3"
|
|
|
"github.com/redis/go-redis/v9"
|
|
|
)
|
|
|
|
|
|
@@ -23,22 +24,32 @@ type DatabaseSchema struct {
|
|
|
}
|
|
|
|
|
|
type MenuLinkPair struct {
|
|
|
- MenuLink string `json:"menu_link"`
|
|
|
- LinkText string `json:"link_text"`
|
|
|
+ MenuLink string `json:"link"`
|
|
|
+ LinkText string `json:"text"`
|
|
|
}
|
|
|
|
|
|
type NavBarItem struct {
|
|
|
- Png string `json:"png"`
|
|
|
+ Png []byte `json:"png"`
|
|
|
Link string `json:"link"`
|
|
|
+ Redirect string `json:"redirect"`
|
|
|
}
|
|
|
|
|
|
+type Asset struct {
|
|
|
+ Name string
|
|
|
+ Data []byte
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+type Identifier string
|
|
|
+
|
|
|
type Document struct {
|
|
|
- PostId string
|
|
|
+ Row int
|
|
|
+ Ident string `json:"id"`
|
|
|
Title string `json:"title"`
|
|
|
Created string `json:"created"`
|
|
|
Body string `json:"body"`
|
|
|
Category string `json:"category"`
|
|
|
- Sample string
|
|
|
+ Sample string `json:"sample"`
|
|
|
}
|
|
|
|
|
|
/*
|
|
|
@@ -59,23 +70,27 @@ func (d *Document) MakeSample() string {
|
|
|
}
|
|
|
|
|
|
type Image struct {
|
|
|
+ Ident string
|
|
|
Location string
|
|
|
Title string
|
|
|
Desc string
|
|
|
+ Created string
|
|
|
+ Category string
|
|
|
+ Data []byte
|
|
|
}
|
|
|
|
|
|
type DocumentIO interface {
|
|
|
- GetDocument(id string) (Document, error)
|
|
|
- GetImage(id string) (Image, error)
|
|
|
+ GetDocument(id Identifier) (Document, error)
|
|
|
+ GetImage(id Identifier) (Image, error)
|
|
|
UpdateDocument(doc Document) error
|
|
|
- DeleteDocument(id string) error
|
|
|
+ DeleteDocument(id Identifier) error
|
|
|
AddDocument(doc Document) error
|
|
|
- AddImage(img ImageStoreItem) error
|
|
|
- GetByCategory(category string) []string
|
|
|
+ AddImage(img Image) error
|
|
|
+ GetByCategory(category string) []Document
|
|
|
AllDocuments() []Document
|
|
|
GetDropdownElements() []MenuLinkPair
|
|
|
GetNavBarLinks() []NavBarItem
|
|
|
- ExportAll()
|
|
|
+ GetAssets() []Asset
|
|
|
}
|
|
|
|
|
|
var (
|
|
|
@@ -99,10 +114,10 @@ func NewSQLiteRepo(db *sql.DB) *SQLiteRepo {
|
|
|
|
|
|
// Creates a new SQL table for text posts
|
|
|
func (r *SQLiteRepo) Migrate() error {
|
|
|
- query := `
|
|
|
+ postsTable := `
|
|
|
CREATE TABLE IF NOT EXISTS posts(
|
|
|
- id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
|
- postid TEXT NOT NULL,
|
|
|
+ row INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
|
+ id TEXT NOT NULL,
|
|
|
title TEXT NOT NULL,
|
|
|
created TEXT NOT NULL,
|
|
|
body TEXT NOT NULL UNIQUE,
|
|
|
@@ -110,55 +125,167 @@ func (r *SQLiteRepo) Migrate() error {
|
|
|
sample TEXT NOT NULL
|
|
|
);
|
|
|
`
|
|
|
+ imagesTable := `
|
|
|
+ CREATE TABLE IF NOT EXISTS images(
|
|
|
+ row INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
|
+ id TEXT NOT NULL,
|
|
|
+ title TEXT NOT NULL,
|
|
|
+ location TEXT NOT NULL,
|
|
|
+ description TEXT NOT NULL,
|
|
|
+ created TEXT NOT NULL,
|
|
|
+ category TEXT NOT NULL
|
|
|
+ );
|
|
|
+ `
|
|
|
+ menuItemsTable := `
|
|
|
+ CREATE TABLE IF NOT EXISTS menu(
|
|
|
+ row INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
|
+ link TEXT NOT NULL,
|
|
|
+ text TEXT NOT NULL
|
|
|
+ );
|
|
|
+ `
|
|
|
+ navbarItemsTable := `
|
|
|
+ CREATE TABLE IF NOT EXISTS navbar(
|
|
|
+ row INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
|
+ png BLOB NOT NULL,
|
|
|
+ link TEXT NOT NULL,
|
|
|
+ redirect TEXT
|
|
|
+ );`
|
|
|
+ assetTable := `
|
|
|
+ CREATE TABLE IF NOT EXISTS assets(
|
|
|
+ row INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
|
+ name TEXT NOT NULL,
|
|
|
+ data BLOB NOT NULL
|
|
|
+ );
|
|
|
+ `
|
|
|
+ seedQueries := []string{postsTable, imagesTable, menuItemsTable, navbarItemsTable, assetTable}
|
|
|
+ for i := range seedQueries {
|
|
|
+ _, err := r.db.Exec(seedQueries[i])
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+func (s *SQLiteRepo) Seed(menu string, pngs string, dir string) {
|
|
|
+ b, err := os.ReadFile(menu)
|
|
|
+
|
|
|
+ if err != nil {log.Fatal(err)}
|
|
|
+ entries := strings.Split(string(b), "\n")
|
|
|
+ for i := range entries {
|
|
|
+ if entries[i] == "" {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ info := strings.Split(entries[i], "=")
|
|
|
+ err := s.AddMenuItem(MenuLinkPair{MenuLink: info[0], LinkText: info[1]})
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ b, err = os.ReadFile(pngs)
|
|
|
+ if err != nil {log.Fatal(err)}
|
|
|
+ entries = strings.Split(string(b), "\n")
|
|
|
+ for i := range entries {
|
|
|
+ if entries[i] == "" {continue}
|
|
|
+ info := strings.Split(entries[i], "=")
|
|
|
+ b, err := os.ReadFile(path.Join(dir, info[0]))
|
|
|
+ if err != nil {log.Fatal(err)}
|
|
|
+ err = s.AddNavbarItem(NavBarItem{Png: b, Link: info[0], Redirect: info[1]})
|
|
|
+ if err != nil {log.Fatal(err)}
|
|
|
+ }
|
|
|
+ assets, err := os.ReadDir(dir)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
+ for i := range assets {
|
|
|
+ b, err := os.ReadFile(path.Join(dir, assets[i].Name()))
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
+ err = s.AddAsset(assets[i].Name(), b)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
|
|
|
- _, err := r.db.Exec(query)
|
|
|
- return err
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
-/*
|
|
|
-Create an entry in the hosts table
|
|
|
|
|
|
- :param host: a Host entry from a port scan
|
|
|
+/*
|
|
|
+Get all dropdown menu elements
|
|
|
*/
|
|
|
-func (r *SQLiteRepo) Create(post Document) error {
|
|
|
- _, err := r.db.Exec("INSERT INTO posts(postid, title, created, body, category, sample) values(?,?,?,?,?)", uuid.New().String(), post.Title, post.Created, post.Body, post.Category, post.MakeSample())
|
|
|
- if err != nil {
|
|
|
- var sqliteErr sqlite3.Error
|
|
|
- if errors.As(err, &sqliteErr) {
|
|
|
- if errors.Is(sqliteErr.ExtendedCode, sqlite3.ErrConstraintUnique) {
|
|
|
- return ErrDuplicate
|
|
|
- }
|
|
|
+func (s *SQLiteRepo) GetDropdownElements() []MenuLinkPair {
|
|
|
+ rows, err := s.db.Query("SELECT * FROM menu")
|
|
|
+ var menuItems []MenuLinkPair
|
|
|
+ defer rows.Close()
|
|
|
+ for rows.Next() {
|
|
|
+ var id int
|
|
|
+ var item MenuLinkPair
|
|
|
+ err = rows.Scan(&id, &item.MenuLink, &item.LinkText)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
- return err
|
|
|
+ menuItems = append(menuItems, item)
|
|
|
}
|
|
|
+ return menuItems
|
|
|
+
|
|
|
|
|
|
- return nil
|
|
|
}
|
|
|
|
|
|
-// Get all Hosts from the host table
|
|
|
-func (r *SQLiteRepo) AllDocuments() []Document {
|
|
|
- rows, err := r.db.Query("SELECT * FROM posts")
|
|
|
- if err != nil {
|
|
|
- fmt.Printf("There was an issue getting all posts. %s", err.Error())
|
|
|
- return nil
|
|
|
- }
|
|
|
+
|
|
|
+/*
|
|
|
+Get all nav bar items
|
|
|
+*/
|
|
|
+func (s *SQLiteRepo) GetNavBarLinks() []NavBarItem {
|
|
|
+
|
|
|
+ rows, err := s.db.Query("SELECT * FROM navbar")
|
|
|
+ var navbarItems []NavBarItem
|
|
|
defer rows.Close()
|
|
|
+ for rows.Next() {
|
|
|
+ var item NavBarItem
|
|
|
+ var id int
|
|
|
+ err = rows.Scan(&id, &item.Png, &item.Link, &item.Redirect)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
+ navbarItems= append(navbarItems, item)
|
|
|
+ }
|
|
|
+ return navbarItems
|
|
|
|
|
|
- var all []Document
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+/*
|
|
|
+get all assets from the asset table
|
|
|
+*/
|
|
|
+func (s *SQLiteRepo) GetAssets() []Asset {
|
|
|
+ rows, err := s.db.Query("SELECT * FROM assets")
|
|
|
+ var assets []Asset
|
|
|
+ defer rows.Close()
|
|
|
for rows.Next() {
|
|
|
- var post Document
|
|
|
- if err := rows.Scan(&post.PostId, &post.Title, &post.Created, &post.Body, &post.Sample); err != nil {
|
|
|
- fmt.Printf("There was an error getting all documents. %s", err.Error())
|
|
|
- return nil
|
|
|
+ var item Asset
|
|
|
+ var id int
|
|
|
+ err = rows.Scan(&id, &item.Name, &item.Data)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
- all = append(all, post)
|
|
|
+ assets = append(assets, item)
|
|
|
}
|
|
|
- return all
|
|
|
+ return assets
|
|
|
+
|
|
|
}
|
|
|
|
|
|
-// Get a blogpost by its postid
|
|
|
-func (r *SQLiteRepo) GetByIP(postId string) (Document, error) {
|
|
|
- row := r.db.QueryRow("SELECT * FROM posts WHERE postid = ?", postId)
|
|
|
+
|
|
|
+
|
|
|
+/*
|
|
|
+Retrieve a document from the sqlite db
|
|
|
+
|
|
|
+ :param id: the Identifier of the post
|
|
|
+*/
|
|
|
+func (s *SQLiteRepo) GetDocument(id Identifier) (Document, error) {
|
|
|
+ row := s.db.QueryRow("SELECT * FROM posts WHERE id = ?", id)
|
|
|
|
|
|
var post Document
|
|
|
if err := row.Scan(&post); err != nil {
|
|
|
@@ -168,49 +295,201 @@ func (r *SQLiteRepo) GetByIP(postId string) (Document, error) {
|
|
|
return post, err
|
|
|
}
|
|
|
return post, nil
|
|
|
+
|
|
|
}
|
|
|
|
|
|
-// Update a record by its ID
|
|
|
-func (r *SQLiteRepo) Update(id int64, updated Document) error {
|
|
|
- if id == 0 {
|
|
|
- return errors.New("invalid updated ID")
|
|
|
+/*
|
|
|
+Get all documents by category
|
|
|
+*/
|
|
|
+func (s *SQLiteRepo) GetByCategory(category string) []Document {
|
|
|
+ rows, err := s.db.Query("SELECT * FROM posts WHERE category = ?", category)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
}
|
|
|
- res, err := r.db.Exec("UPDATE posts SET title = ?, body = ?, desc = ? WHERE id = ?", updated.Title, updated.Body, updated.MakeSample(), id)
|
|
|
+ var docs []Document
|
|
|
+ defer rows.Close()
|
|
|
+ for rows.Next() {
|
|
|
+ var doc Document
|
|
|
+ err := rows.Scan(&doc.Row, &doc.Ident, &doc.Title, &doc.Created, &doc.Body, &doc.Category, &doc.Sample)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
+ docs = append(docs, doc)
|
|
|
+ }
|
|
|
+ err = rows.Err()
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
+ return docs
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+/*
|
|
|
+get image data from the images table
|
|
|
+ :param id: the serial identifier of the post
|
|
|
+*/
|
|
|
+func (s *SQLiteRepo) GetImage(id Identifier) (Image, error) {
|
|
|
+ row := s.db.QueryRow("SELECT * FROM images WHERE id = ?", id)
|
|
|
+ var title string
|
|
|
+ var location string
|
|
|
+ var desc string
|
|
|
+ var category string
|
|
|
+ var created string
|
|
|
+ if err := row.Scan(&title, &location, &desc, &created, &category); err != nil {
|
|
|
+ if errors.Is(err, sql.ErrNoRows) {
|
|
|
+ return Image{}, ErrNotExists
|
|
|
+ }
|
|
|
+ return Image{}, err
|
|
|
+ }
|
|
|
+ data, err := os.ReadFile(location)
|
|
|
+ if err != nil {
|
|
|
+ return Image{}, err
|
|
|
+ }
|
|
|
+ return Image{Title: title, Location: location, Desc: desc, Data: data, Created: created, Category: category}, nil
|
|
|
+}
|
|
|
+
|
|
|
+/*
|
|
|
+Add an image to the database
|
|
|
+ :param title: the title of the image
|
|
|
+ :param location: the location to save the image to
|
|
|
+ :param desc: the description of the image, if any
|
|
|
+ :param data: the binary data for the image
|
|
|
+*/
|
|
|
+func (s *SQLiteRepo) AddImage(img Image) error {
|
|
|
+ err := os.WriteFile(path.Join(img.Location, img.Ident), img.Data, os.ModeAppend)
|
|
|
if err != nil {
|
|
|
return err
|
|
|
}
|
|
|
+ _, err = s.db.Exec("INSERT INTO images (id, title, location, desc, created, category) VALUES (?,?,?,?,?,?,?)", img.Ident, img.Title, img.Location, img.Desc, img.Created, img.Category)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ return nil
|
|
|
+}
|
|
|
|
|
|
- rowsAffected, err := res.RowsAffected()
|
|
|
+
|
|
|
+
|
|
|
+func (s *SQLiteRepo) UpdateDocument(doc Document) error {
|
|
|
+ tx, err := s.db.Begin()
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ stmt,_ := tx.Prepare("UPDATE posts set title=? body=? category=? WHERE id=?")
|
|
|
+ _, err = stmt.Exec(doc.Title, doc.Body, doc.Category, doc.Ident)
|
|
|
if err != nil {
|
|
|
+ tx.Rollback()
|
|
|
return err
|
|
|
}
|
|
|
+ tx.Commit()
|
|
|
+ return nil
|
|
|
+}
|
|
|
|
|
|
- if rowsAffected == 0 {
|
|
|
- return ErrUpdateFailed
|
|
|
+func (s *SQLiteRepo) AddMenuItem(item MenuLinkPair) error {
|
|
|
+ tx, err := s.db.Begin()
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
}
|
|
|
+ stmt,_ := tx.Prepare("INSERT INTO menu(link, text) VALUES (?,?)")
|
|
|
+ _, err = stmt.Exec(item.MenuLink, item.LinkText)
|
|
|
+ if err != nil {
|
|
|
+ tx.Rollback()
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ tx.Commit()
|
|
|
+ return nil
|
|
|
+
|
|
|
+}
|
|
|
|
|
|
+func (s *SQLiteRepo) AddNavbarItem(item NavBarItem) error {
|
|
|
+ tx, err := s.db.Begin()
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ stmt,_ := tx.Prepare("INSERT INTO navbar(png, link, redirect) VALUES (?,?,?)")
|
|
|
+ _, err = stmt.Exec(item.Png, item.Link, item.Redirect)
|
|
|
+ if err != nil {
|
|
|
+ tx.Rollback()
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ tx.Commit()
|
|
|
return nil
|
|
|
+
|
|
|
}
|
|
|
|
|
|
-// Delete a record by its ID
|
|
|
-func (r *SQLiteRepo) Delete(id int64) error {
|
|
|
- res, err := r.db.Exec("DELETE FROM posts WHERE id = ?", id)
|
|
|
+func (s *SQLiteRepo) AddAsset(name string, data []byte) error {
|
|
|
+ tx, err := s.db.Begin()
|
|
|
if err != nil {
|
|
|
return err
|
|
|
}
|
|
|
+ stmt,_ := tx.Prepare("INSERT INTO assets(name, data) VALUES (?,?)")
|
|
|
+ _, err = stmt.Exec(name, data)
|
|
|
+ if err != nil {
|
|
|
+ tx.Rollback()
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ tx.Commit()
|
|
|
+ return nil
|
|
|
+}
|
|
|
|
|
|
- rowsAffected, err := res.RowsAffected()
|
|
|
+
|
|
|
+func (s *SQLiteRepo) AddDocument(doc Document) error {
|
|
|
+ tx, err := s.db.Begin()
|
|
|
if err != nil {
|
|
|
return err
|
|
|
}
|
|
|
+ stmt,_ := tx.Prepare("INSERT INTO posts (id, title, created, body, category,sample) VALUES (?,?,?,?,?,?)")
|
|
|
+ _, err = stmt.Exec(doc.Ident, doc.Title, doc.Created, doc.Body, doc.Category, doc.Sample)
|
|
|
+ if err != nil {
|
|
|
+ tx.Rollback()
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ tx.Commit()
|
|
|
+ return nil
|
|
|
|
|
|
- if rowsAffected == 0 {
|
|
|
- return ErrDeleteFailed
|
|
|
+}
|
|
|
+
|
|
|
+/*
|
|
|
+Delete a document from the db
|
|
|
+ :param id: the identifier of the document to remove
|
|
|
+*/
|
|
|
+func (s *SQLiteRepo) DeleteDocument(id Identifier) error {
|
|
|
+ tx, err := s.db.Begin()
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
}
|
|
|
+ stmt,_ := tx.Prepare("DELETE FROM posts WHERE id=?")
|
|
|
+ _, err = stmt.Exec(id)
|
|
|
+ if err != nil {
|
|
|
+ tx.Rollback()
|
|
|
+ return err}
|
|
|
+ tx.Commit()
|
|
|
+ return nil
|
|
|
+
|
|
|
+}
|
|
|
|
|
|
- return err
|
|
|
+// Get all Hosts from the host table
|
|
|
+func (s *SQLiteRepo) AllDocuments() []Document {
|
|
|
+ rows, err := s.db.Query("SELECT * FROM posts")
|
|
|
+ if err != nil {
|
|
|
+ fmt.Printf("There was an issue getting all posts. %s", err.Error())
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ defer rows.Close()
|
|
|
+
|
|
|
+ var all []Document
|
|
|
+ for rows.Next() {
|
|
|
+ var post Document
|
|
|
+ if err := rows.Scan(&post.Ident, &post.Title, &post.Created, &post.Body, &post.Sample); err != nil {
|
|
|
+ fmt.Printf("There was an error getting all documents. %s", err.Error())
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ all = append(all, post)
|
|
|
+ }
|
|
|
+ return all
|
|
|
}
|
|
|
|
|
|
+
|
|
|
+
|
|
|
type InvalidSkipArg struct{ Skip int }
|
|
|
|
|
|
func (i *InvalidSkipArg) Error() string {
|
|
|
@@ -235,18 +514,17 @@ Create a new ImageStoreItem
|
|
|
:param title: the canonical title to give the image
|
|
|
:param desc: the description to associate to the image
|
|
|
*/
|
|
|
-func NewImageStoreItem(fname string, title string, desc string) *ImageStoreItem {
|
|
|
+func NewImageStoreItem(title string, desc string) Image {
|
|
|
id := uuid.New()
|
|
|
- img := ImageStoreItem{
|
|
|
- Identifier: id.String(),
|
|
|
- Filename: fname,
|
|
|
+ img := Image{
|
|
|
+ Ident: id.String(),
|
|
|
Title: title,
|
|
|
Category: DIGITAL_ART,
|
|
|
- AbsolutePath: fmt.Sprintf("%s/%s", GetImageStore(), fname),
|
|
|
+ Location: GetImageStore(),
|
|
|
Created: time.Now().UTC().String(),
|
|
|
Desc: desc,
|
|
|
}
|
|
|
- return &img
|
|
|
+ return img
|
|
|
}
|
|
|
|
|
|
/*
|
