Move all DB setup to its own file

1 files changed, 37 insertions, 1 deletions

diff --git a/setup.go b/setup.go
index 6366921..4506282 100644
--- a/setup.go
+++ b/setup.go
@@ -12,6 +12,42 @@ import (
 	"github.com/goccy/go-json"
 )
 
+func setupDatabase() (*sql.DB, error) {
+	db, err := sql.Open("sqlite3", dictionary)
+	if err != nil {
+		return nil, fmt.Errorf("opening DB '%s': %s", dictionary, err)
+	}
+	defer db.Close()
+
+	_, err = db.Exec("create table IF NOT EXISTS words (word text not null, definition text);")
+	if err != nil {
+		return nil, fmt.Errorf("creating table: %s", err)
+	}
+
+	row := db.QueryRow(`SELECT count(*) as count from words`)
+	var count int
+	err = row.Scan(&count)
+	if err != nil {
+		return nil, fmt.Errorf("counting rows: %s", err)
+	}
+
+	// Only populate the database if it is empty.
+	if count > 0 {
+		return db, nil
+	}
+
+	// Faster import performance.
+	_, err = db.Exec("PRAGMA synchronous = OFF;")
+	if err != nil {
+		return nil, fmt.Errorf("setting risky writes: %s", err)
+	}
+
+	if err = populateDictionary(db); err != nil {
+		return nil, fmt.Errorf("failed to prepare dictionary: %s", err)
+	}
+	return db, nil
+}
+
 type rawDictionaryEntry struct {
 	Word      string   `json:"word"`
 	LangCode  string   `json:"lang_code"`
@@ -49,7 +85,7 @@ type SenseForDictionaryEntry struct {
 	Example string
 }
 
-func readDictionary(db *sql.DB) error {
+func populateDictionary(db *sql.DB) error {
 	log.Printf("preparing list of dictionary words...")
 
 	// Set up the template