TheMathemagicians/sevenkeys/logic/search.go

package logic

import (
	"database/sql"
	"fmt"
	"io"
	"os"
	"os/exec"
	"sevenkeys/database"
	"strings"
)

type Triadic int

const (
	True Triadic = iota
	False
	Either
)

type SearchCriteria struct {
	SetCode  string
	Foil     Triadic
	Promo    Triadic
	Language string
}

// The InsertSearchOptions type is a map of `string` keys (which can be searched using
// fzf) to `string` values (which represent a primary key in the CardPrintings table)
type InsertSearchOptions map[string]string

func filterPrinting(printing database.CardPrinting, searchCriteria SearchCriteria) bool {
	if searchCriteria.SetCode != "" && printing.SetCode != searchCriteria.SetCode {
		return true
	}

	if searchCriteria.Foil == False && printing.IsFoil {
		return true
	}
	if searchCriteria.Foil == True && !printing.IsFoil {
		return true
	}

	if searchCriteria.Promo == False && printing.IsPromo {
		return true
	}
	if searchCriteria.Promo == True && !printing.IsPromo {
		return true
	}

	if searchCriteria.Language != "" && printing.Language != searchCriteria.Language {
		return true
	}

	return false
}

func GetAllSearchOptions(db *sql.DB, searchCriteria SearchCriteria) (InsertSearchOptions, error) {
	var searchOptions InsertSearchOptions = make(map[string]string)

	cardPrintings, err := database.GetAllCardPrintings(db)
	if err != nil {
		return searchOptions, err
	}

	for _, printing := range cardPrintings {
		// Filter based on search criteria
		filter := filterPrinting(printing, searchCriteria)
		if filter {
			continue
		}

		// Construct search option string
		searchString := fmt.Sprintf("%s (%s %s) [%s]", printing.Name, printing.SetCode, printing.CollectorNumber, printing.Language)

		if printing.IsFoil {
			searchString += " FOIL"
		}

		if printing.IsPromo {
			searchString += " PROMO"
		}

		searchOptions[searchString] = printing.Id
	}

	return searchOptions, err
}

func GenericSearch[pk string | int](options map[string]pk) (pk, error) {
	var value pk

	cmd := exec.Command("fzf")
	cmd.Stderr = os.Stderr

	fzfStdin, err := cmd.StdinPipe()
	if err != nil {
		return value, err
	}

	go func() {
		defer fzfStdin.Close()
		for option, _ := range options {
			io.WriteString(fzfStdin, option+"\n")
		}
	}()

	fzfOutput, err := cmd.Output()
	if err != nil {
		return value, err
	}

	searchResult := strings.TrimSuffix(string(fzfOutput), "\n")
	value = options[searchResult]
	return value, nil
}