add projects and metrics routes

api/database/database.go

@@ -2,76 +2,62 @@ package database
 
 import (
 	"fmt"
+	"os"
+	"path"
 
 	"database/sql"
+
 	_ "github.com/mattn/go-sqlite3"
 )
 
+const (
+	SQL_PATH = "sql"
+
+	TABLE_ADMIN_LOG = "admin_log" // stores administrator logs
+	TABLE_METRICS   = "metrics"   // stores API usage metrcis
+	TABLE_NEWS      = "news"      // stores news posts
+	TABLE_SERVICES  = "services"  // stores services
+	TABLE_PROJECTS  = "projects"  // stores projects
+)
+
+var tables []string = []string{
+	TABLE_ADMIN_LOG, TABLE_METRICS, TABLE_NEWS,
+	TABLE_SERVICES, TABLE_PROJECTS,
+}
+
 type Type struct {
 	sql  *sql.DB
 	rows *sql.Rows
 }
 
-func (db *Type) Load() (err error) {
-	if db.sql, err = sql.Open("sqlite3", "data.db"); err != nil {
-		return fmt.Errorf("cannot access the database: %s", err.Error())
+func (db *Type) create_table(table string) error {
+	var (
+		err   error
+		query []byte
+	)
+
+	query_path := path.Join(SQL_PATH, table+".sql")
+
+	if query, err = os.ReadFile(query_path); err != nil {
+		return fmt.Errorf("failed to read %s for table %s: %", query_path, table, err.Error())
 	}
 
-	// see database/visitor.go
-	_, err = db.sql.Exec(`
-    CREATE TABLE IF NOT EXISTS visitor_count(
-      id    TEXT NOT NULL UNIQUE,
-      count INTEGER NOT NULL
-    );
-  `)
-
-	if err != nil {
-		return fmt.Errorf("failed to create the visitor_count table: %s", err.Error())
-	}
-
-	// see database/service.go
-	_, err = db.sql.Exec(`
-    CREATE TABLE IF NOT EXISTS services(
-      name       TEXT NOT NULL UNIQUE,
-      desc       TEXT NOT NULL,
-			check_time INTEGER NOT NULL,
-			check_res  INTEGER NOT NULL,
-			check_url  TEXT NOT NULL,
-      clear      TEXT,
-      onion      TEXT,
-      i2p        TEXT
-    );
-  `)
-
-	if err != nil {
-		return fmt.Errorf("failed to create the services table: %s", err.Error())
-	}
-
-	// see database/news.go
-	_, err = db.sql.Exec(`
-    CREATE TABLE IF NOT EXISTS news(
-			id      TEXT NOT NULL UNIQUE,
-			title   TEXT NOT NULL,
-			author  TEXT NOT NULL,
-			time    INTEGER NOT NULL,
-			content TEXT NOT NULL
-		);
-  `)
-
-	if err != nil {
-		return fmt.Errorf("failed to create the news table: %s", err.Error())
-	}
-
-	// see database/admin.go
-	_, err = db.sql.Exec(`
-    CREATE TABLE IF NOT EXISTS admin_log(
-			action TEXT NOT NULL,
-			time   INTEGER NOT NULL
-    );
-  `)
-
-	if err != nil {
-		return fmt.Errorf("failed to create the admin_log table: %s", err.Error())
+	if _, err = db.sql.Exec(string(query)); err != nil {
+		return fmt.Errorf("failed to create the %s table: %s", table, err.Error())
+	}
+
+	return nil
+}
+
+func (db *Type) Load() (err error) {
+	if db.sql, err = sql.Open("sqlite3", "data.db"); err != nil {
+		return fmt.Errorf("failed access the database: %s", err.Error())
+	}
+
+	for _, table := range tables {
+		if err = db.create_table(table); err != nil {
+			return err
+		}
 	}
 
 	return nil