84 lines
2.6 KiB
Go
84 lines
2.6 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"log"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func InitDB(db *sql.DB) {
|
|
schema := `
|
|
CREATE TABLE IF NOT EXISTS currencies (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
code TEXT NOT NULL UNIQUE,
|
|
name TEXT NOT NULL
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS companies (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT NOT NULL UNIQUE,
|
|
shares_outstanding INTEGER NOT NULL,
|
|
price REAL NOT NULL,
|
|
currency_id INTEGER NOT NULL,
|
|
FOREIGN KEY (currency_id) REFERENCES currencies(id)
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS periods (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
type TEXT NOT NULL CHECK(type IN ('Q', 'H', 'Y')),
|
|
year INTEGER NOT NULL,
|
|
idx INTEGER NOT NULL,
|
|
start_date TEXT NOT NULL,
|
|
end_date TEXT NOT NULL,
|
|
UNIQUE(type, year, idx)
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS revenue_reports (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
company_id INTEGER NOT NULL,
|
|
period_id INTEGER NOT NULL,
|
|
FOREIGN KEY (company_id) REFERENCES companies(id),
|
|
FOREIGN KEY (period_id) REFERENCES periods(id),
|
|
UNIQUE(company_id, period_id)
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS revenue_entries (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
report_id INTEGER NOT NULL,
|
|
currency_id INTEGER NOT NULL,
|
|
category TEXT NOT NULL CHECK(category IN ('product', 'location', 'total')),
|
|
label TEXT NOT NULL,
|
|
value REAL NOT NULL,
|
|
FOREIGN KEY (report_id) REFERENCES revenue_reports(id),
|
|
FOREIGN KEY (currency_id) REFERENCES currencies(id)
|
|
);
|
|
|
|
-- one row per category type per company
|
|
-- e.g. Apple has "product" and "location"
|
|
CREATE TABLE IF NOT EXISTS category_defs (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
company_id INTEGER NOT NULL,
|
|
name TEXT NOT NULL,
|
|
FOREIGN KEY (company_id) REFERENCES companies(id),
|
|
UNIQUE(company_id, name)
|
|
);
|
|
|
|
-- the actual labels belonging to each category
|
|
-- e.g. category_def "product" → "iPhone", "Mac", "Services"
|
|
CREATE TABLE IF NOT EXISTS category_labels (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
category_def_id INTEGER NOT NULL,
|
|
label TEXT NOT NULL,
|
|
FOREIGN KEY (category_def_id) REFERENCES category_defs(id),
|
|
UNIQUE(category_def_id, label)
|
|
);
|
|
`
|
|
|
|
if _, err := db.Exec(schema); err != nil {
|
|
log.Fatal("Failed to create tables:", err)
|
|
}
|
|
fmt.Println("Tables ready")
|
|
}
|