Implement basic search-and-store functionality

sevenkeys/database/cardstoragelocation.go

@@ -25,7 +25,7 @@ func GetLastPositionInBox(db *sql.DB, storageBox string) (int, error) {
 	return lastPosition, nil
 }
 
-func InsertCardStorageLocation(db *sql.DB, location CardStorageLocation) error {
+func InsertCardStorageLocation(db *sql.DB, storageLocation CardStorageLocation) error {
 	query := `INSERT INTO CardStorageLocation
 		(CardPrintingId, StorageBox, Position, Source)
 		VALUES (?, ?, ?, ?);`
@@ -35,7 +35,7 @@ func InsertCardStorageLocation(db *sql.DB, location CardStorageLocation) error {
 		return err
 	}
 
-	_, err = insert.Exec(location.CardPrintingId, location.StorageBox, location.Position, location.Source)
+	_, err = insert.Exec(storageLocation.CardPrintingId, storageLocation.StorageBox, storageLocation.Position, storageLocation.Source)
 	if err != nil {
 		return err
 	}

sevenkeys/logic/search.go

@@ -3,7 +3,11 @@ package logic
 import (
 	"database/sql"
 	"fmt"
+	"io"
+	"os"
+	"os/exec"
 	"sevenkeys/database"
+	"strings"
 )
 
 // The SearchOptions type is a map of `string`s (which can be searched using fzf)
@@ -34,3 +38,29 @@ func GetAllSearchOptions(db *sql.DB) (SearchOptions, error) {
 
 	return searchOptions, err
 }
+
+func Search(searchOptions SearchOptions) (int, string, error) {
+	cmd := exec.Command("fzf")
+	cmd.Stderr = os.Stderr
+
+	fzfStdin, err := cmd.StdinPipe()
+	if err != nil {
+		return -1, "", err
+	}
+
+	go func() {
+		defer fzfStdin.Close()
+		for searchString, _ := range searchOptions {
+			io.WriteString(fzfStdin, searchString+"\n")
+		}
+	}()
+
+	fzfOutput, err := cmd.Output()
+	if err != nil {
+		return -1, "", err
+	}
+
+	key := strings.TrimSuffix(string(fzfOutput), "\n")
+
+	return searchOptions[key], key, nil
+}

sevenkeys/logic/storage.go

@@ -1,12 +1,22 @@
 package logic
 
-import "fmt"
+import (
+	"database/sql"
+	"sevenkeys/database"
+)
 
-func GetStorageLocation() string {
+func StoreCard(db *sql.DB, storageLocation database.CardStorageLocation) error {
-	fmt.Print("Enter storage location: ")
+	lastPosition, err := database.GetLastPositionInBox(db, storageLocation.StorageBox)
+	if err != nil {
+		return err
+	}
 
-	var response string
+	storageLocation.Position = lastPosition + 1
-	fmt.Scan(&response)
 
-	return response
+	err = database.InsertCardStorageLocation(db, storageLocation)
+	if err != nil {
+		return err
+	}
+
+	return nil
 }

sevenkeys/logic/ui.go

@@ -0,0 +1,19 @@
+package logic
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+)
+
+func GetResponse(prompt string) string {
+	fmt.Print(prompt, " ")
+
+	scanner := bufio.NewScanner(os.Stdin)
+	var response string
+	if scanner.Scan() {
+		response = scanner.Text()
+	}
+
+	return response
+}

sevenkeys/main.go

@@ -5,7 +5,6 @@ import (
 	"sevenkeys/database"
 	"sevenkeys/logic"
 	"sevenkeys/logic/scryfall"
-	"time"
 
 	"github.com/inancgumus/screen"
 )
@@ -40,13 +39,38 @@ func main() {
 		fmt.Println("No update required.")
 	}
 
-	storageLocation := logic.GetStorageLocation()
+	storageBox := logic.GetResponse("Enter storage box label:")
+	source := logic.GetResponse("Enter source:")
+	searchOptions, err := logic.GetAllSearchOptions(db)
+	logic.Check(err)
 
 	for {
 		screen.Clear()
 		screen.MoveTopLeft()
-		fmt.Println("Current storage location:", storageLocation)
+		fmt.Println("Storage location:", storageBox)
+		fmt.Println("Source:", source)
 
-		time.Sleep(1 * time.Second)
+		selectedCardPrintingId, selectedSearchOption, err := logic.Search(searchOptions)
+		logic.Check(err)
+
+		fmt.Println("Inserted card:", selectedSearchOption)
+		storageLocation := database.CardStorageLocation{
+			CardPrintingId: selectedCardPrintingId,
+			StorageBox:     storageBox,
+			Source:         source,
+		}
+		err = logic.StoreCard(db, storageLocation)
+		logic.Check(err)
+
+		nextAction := logic.GetResponse("[s]earch again/[r]epeat last insert/[q]uit:")
+		switch nextAction {
+		case "s":
+			continue
+		case "r":
+			err = logic.StoreCard(db, storageLocation)
+			logic.Check(err)
+		case "q":
+			break
+		}
 	}
 }

sevenkeys/printinglist.go

@@ -1,27 +0,0 @@
-package main
-
-/*
-func main() {
-	db := database.GetDatabaseFromConfig("config.json")
-	cardSearchOptions, err := operations.GetCardSearchOptions(db)
-	check(err)
-
-	cmd := exec.Command("fzf")
-	cmd.Stderr = os.Stderr
-
-	fzfStdin, err := cmd.StdinPipe()
-	check(err)
-
-	go func() {
-		defer fzfStdin.Close()
-		for _, option := range cardSearchOptions {
-			io.WriteString(fzfStdin, option+"\n")
-		}
-	}()
-
-	fzfOutput, err := cmd.Output()
-	check(err)
-
-	fmt.Println("Output:", string(fzfOutput))
-}
-*/