v5.0 cleanup

api/database/database.go

@@ -2,18 +2,10 @@ package database
 
 import (
 	"database/sql"
-	"github.com/ngn13/website/api/global"
 )
 
-type Type struct {
-	Sql   *sql.DB
-	Votes []global.Vote
-}
-
-func (t *Type) Setup() error {
-	t.Votes = []global.Vote{}
-
-	_, err := t.Sql.Exec(`
+func Setup(db *sql.DB) error {
+	_, err := db.Exec(`
     CREATE TABLE IF NOT EXISTS posts(
       id      TEXT NOT NULL UNIQUE,
       title   TEXT NOT NULL,
@@ -29,7 +21,7 @@ func (t *Type) Setup() error {
 		return err
 	}
 
-	_, err = t.Sql.Exec(`
+	_, err = db.Exec(`
     CREATE TABLE IF NOT EXISTS services( 
       name    TEXT NOT NULL UNIQUE,
       desc    TEXT NOT NULL,
@@ -37,19 +29,16 @@ func (t *Type) Setup() error {
     );
   `)
 
-	return err
-}
-
-func (t *Type) Open(p string) error {
-	var err error
-
-	if t.Sql, err = sql.Open("sqlite3", p); err != nil {
+	if err != nil {
 		return err
 	}
 
-	return t.Setup()
-}
+	_, err = db.Exec(`
+    CREATE TABLE IF NOT EXISTS votes( 
+      hash       TEXT NOT NULL UNIQUE,
+      is_upvote  INTEGER NOT NULL
+    );
+  `)
 
-func (t *Type) Close() {
-	t.Sql.Close()
+	return err
 }

api/database/post.go

@@ -0,0 +1,71 @@
+package database
+
+import (
+	"database/sql"
+
+	"github.com/ngn13/website/api/util"
+)
+
+type Post struct {
+	ID      string `json:"id"`
+	Title   string `json:"title"`
+	Author  string `json:"author"`
+	Date    string `json:"date"`
+	Content string `json:"content"`
+	Public  int    `json:"public"`
+	Vote    int    `json:"vote"`
+}
+
+func (p *Post) Load(rows *sql.Rows) error {
+	return rows.Scan(&p.ID, &p.Title, &p.Author, &p.Date, &p.Content, &p.Public, &p.Vote)
+}
+
+func (p *Post) Get(db *sql.DB, id string) (bool, error) {
+	var (
+		success bool
+		rows    *sql.Rows
+		err     error
+	)
+
+	if rows, err = db.Query("SELECT * FROM posts WHERE id = ?", id); err != nil {
+		return false, err
+	}
+	defer rows.Close()
+
+	if success = rows.Next(); !success {
+		return false, nil
+	}
+
+	if err = p.Load(rows); err != nil {
+		return false, err
+	}
+
+	return true, nil
+}
+
+func (p *Post) Remove(db *sql.DB) error {
+	_, err := db.Exec("DELETE FROM posts WHERE id = ?", p.ID)
+	return err
+}
+
+func (p *Post) Save(db *sql.DB) error {
+	p.ID = util.TitleToID(p.Title)
+
+	_, err := db.Exec(
+		"INSERT INTO posts(id, title, author, date, content, public, vote) values(?, ?, ?, ?, ?, ?, ?)",
+		p.ID, p.Title, p.Author, p.Date, p.Content, p.Public, p.Vote,
+	)
+
+	return err
+}
+
+func (p *Post) Update(db *sql.DB) error {
+	p.ID = util.TitleToID(p.Title)
+
+	_, err := db.Exec(
+		"UPDATE posts SET title = ?, author = ?, date = ?, content = ?, public = ?, vote = ? WHERE id = ?",
+		p.Title, p.Author, p.Date, p.Content, p.Public, p.Vote, p.ID,
+	)
+
+	return err
+}

api/database/service.go

@@ -0,0 +1,56 @@
+package database
+
+import (
+	"database/sql"
+)
+
+type Service struct {
+	Name string `json:"name"`
+	Desc string `json:"desc"`
+	Url  string `json:"url"`
+}
+
+func (s *Service) Load(rows *sql.Rows) error {
+	return rows.Scan(&s.Name, &s.Desc, &s.Url)
+}
+
+func (s *Service) Get(db *sql.DB, name string) (bool, error) {
+	var (
+		success bool
+		rows    *sql.Rows
+		err     error
+	)
+
+	if rows, err = db.Query("SELECT * FROM services WHERE name = ?", name); err != nil {
+		return false, err
+	}
+	defer rows.Close()
+
+	if success = rows.Next(); !success {
+		return false, nil
+	}
+
+	if err = s.Load(rows); err != nil {
+		return false, err
+	}
+
+	return true, nil
+}
+
+func (s *Service) Remove(db *sql.DB) error {
+	_, err := db.Exec(
+		"DELETE FROM services WHERE name = ?",
+		s.Name,
+	)
+
+	return err
+}
+
+func (s *Service) Save(db *sql.DB) error {
+	_, err := db.Exec(
+		"INSERT INTO services(name, desc, url) values(?, ?, ?)",
+		s.Name, s.Desc, s.Url,
+	)
+
+	return err
+}

api/database/vote.go

@@ -0,0 +1,49 @@
+package database
+
+import "database/sql"
+
+type Vote struct {
+	Hash     string
+	IsUpvote bool
+}
+
+func (v *Vote) Load(rows *sql.Rows) error {
+	return rows.Scan(&v.Hash, &v.IsUpvote)
+}
+
+func (v *Vote) Get(db *sql.DB, hash string) (bool, error) {
+	var (
+		success bool
+		rows    *sql.Rows
+		err     error
+	)
+
+	if rows, err = db.Query("SELECT * FROM votes WHERE hash = ?", hash); err != nil {
+		return false, err
+	}
+	defer rows.Close()
+
+	if success = rows.Next(); !success {
+		return false, nil
+	}
+
+	if err = v.Load(rows); err != nil {
+		return false, err
+	}
+
+	return true, nil
+}
+
+func (v *Vote) Update(db *sql.DB) error {
+	_, err := db.Exec("UPDATE votes SET is_upvote = ? WHERE hash = ?", v.IsUpvote, v.Hash)
+	return err
+}
+
+func (v *Vote) Save(db *sql.DB) error {
+	_, err := db.Exec(
+		"INSERT INTO votes(hash, is_upvote) values(?, ?)",
+		v.Hash, v.IsUpvote,
+	)
+
+	return err
+}