Merge pull request #5 from flrnd/add-database-init

Add database init command and bump version to 1.0

Author: flrnd

cmd/init.go

@@ -0,0 +1,27 @@
+package cmd
+
+import (
+	"os"
+
+	"github.com/flrnd/parrot/db"
+	"github.com/flrnd/parrot/util"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	rootCmd.AddCommand(initCmd)
+}
+
+var initCmd = &cobra.Command{
+	Use:   "init my_diceware_word_list.asc",
+	Short: "Initialize the diceware databse",
+	Long:  "Initialize the diceware database. Download a file list from https://theworld.com/~reinhold/diceware.html or create your own.",
+	Args:  cobra.MinimumNArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		f, err := os.Open(args[0])
+
+		util.Check(err)
+
+		db.Init(f)
+	},
+}

cmd/version.go

@@ -7,7 +7,7 @@ import (
 )
 
 var appName = "parrot"
-var appVersion = "v0.0.3"
+var appVersion = "v1.0.0"
 
 func init() {
 	rootCmd.AddCommand(versionCmd)

db/database.go

@@ -1,8 +1,12 @@
 package db
 
 import (
+	"bufio"
 	"database/sql"
+	"fmt"
 	"os"
+	"strconv"
+	"strings"
 
 	"github.com/flrnd/parrot/util"
 	_ "github.com/mattn/go-sqlite3"
@@ -15,6 +19,53 @@ func Path() string {
 	return homeDir + dbPath
 }
 
+func Init(f *os.File) {
+	homeDir, _ := os.UserHomeDir()
+	databasePath := Path()
+
+	if _, err := os.Stat(databasePath); os.IsNotExist(err) {
+		os.MkdirAll(homeDir+"/.config/parrot", 0755)
+		file, err := os.Create(databasePath)
+		util.Check(err)
+
+		file.Close()
+	}
+
+	// Open the database
+	parrotDB, err := sql.Open("sqlite3", databasePath)
+	util.Check(err)
+	defer parrotDB.Close()
+
+	scanner := bufio.NewScanner(f)
+	wordsDatabase, err := sql.Open("sqlite3", databasePath)
+
+	util.Check(err)
+
+	defer wordsDatabase.Close()
+
+	createWordsTable := `create table words(id integer not null primary key, word text) without rowid;`
+
+	if createTable, err := wordsDatabase.Prepare(createWordsTable); err != nil {
+		util.Check(err)
+	} else {
+		createTable.Exec()
+	}
+
+	stmt, err := wordsDatabase.Prepare("INSERT INTO words(id, word) values(?,?)")
+	util.Check(err)
+
+	fmt.Printf("Initializing the database with %s\n", f.Name())
+
+	for scanner.Scan() {
+		word := scanner.Text()
+		tuple := strings.Split(word, "	")
+		id, err := strconv.Atoi(tuple[0])
+		util.Check(err)
+
+		stmt.Exec(id, tuple[1])
+	}
+}
+
 func Find(id int, db *sql.DB) string {
 	rows, err := db.Query("select id, word from words where id = ?", id)
 	util.Check(err)