data.go

package data

import (
	"bytes"
	"compress/gzip"
	"encoding/gob"
	"fmt"
	"io/ioutil"
	"strings"

	"github.com/ericraio/landlord/src/card"
	"github.com/ericraio/landlord/src/scryfall"
	"github.com/markbates/pkger"
)

var DATA []scryfall.ScryfallCard

type CardData struct {
	Cards []scryfall.ScryfallCard
}

func NewCardData() *CardData {
	return &CardData{}
}

func (cd *CardData) FindByName(cardName string) *card.Card {
	for _, c := range cd.Cards {
		if strings.EqualFold(c.Name, cardName) {
			return c.ConvertToCard()
		}
	}
	return nil
}

func (cd *CardData) Load() error {
	if len(cd.Cards) > 0 {
		return nil
	}
	if len(DATA) > 0 {
		cd.Cards = DATA
		return nil
	}

	fi, err := pkger.Open("/data/all_cards.landlord")
	if err != nil {
		fi, err = pkger.Open("../../data/all_cards.landlord")
		if err != nil {
			return err
		}
	}
	defer func() {
		err := fi.Close()
		if err != nil {
			fmt.Println(err)
		}
	}()

	fz, err := gzip.NewReader(fi)
	if err != nil {
		return err
	}
	defer func() {
		err := fz.Close()
		if err != nil {
			fmt.Println(err)
		}
	}()

	s, err := ioutil.ReadAll(fz)
	if err != nil {
		return err
	}

	buf := bytes.NewBuffer(s)

	dec := gob.NewDecoder(buf)
	if err := dec.Decode(&cd.Cards); err != nil {
		return err
	}
	DATA = cd.Cards
	return nil
}