TheMathemagicians/sevenkeys/logic/storage.go

package logic

import (
	"database/sql"
	"errors"
	"sevenkeys/database"
)

var ErrCouldNotGetStorageAreaId error = errors.New("Could not get storage area ID.")

func GetStorageAreaId(db *sql.DB, storageAreaName string) (int, error) {
	id, err := database.GetStorageAreaIdByName(db, storageAreaName)
	if err == nil {
		return id, nil
	}

	if err == database.ErrStorageAreaDoesNotExist {
		storageOptions, err := GetStorageAreaSearchOptions(db)
		if err != nil {
			return id, err
		}

		id, _, err := GenericSearch(storageOptions)
		if err == nil {
			return id, nil
		}

		return id, ErrCouldNotGetStorageAreaId
	}

	return id, ErrCouldNotGetStorageAreaId
}

func CreateStorageArea(db *sql.DB, storageArea database.StorageArea) error {
	// TODO: Check if there's already a storage are with the same name
	// TODO: Check if the type entered is valid
	err := database.InsertStorageArea(db, storageArea)
	if err != nil {
		return err
	}

	return nil
}

type StorageAreaSearchOptions map[string]int

func GetStorageAreaSearchOptions(db *sql.DB) (StorageAreaSearchOptions, error) {
	var options StorageAreaSearchOptions = make(map[string]int)

	storageAreas, err := database.GetAllStorageAreas(db)
	if err != nil {
		return options, err
	}

	for _, area := range storageAreas {
		options[area.Name] = area.Id
	}

	return options, nil
}

func storeAfterLastCard(db *sql.DB, cardLocation database.CardLocation) error {
	lastPosition, err := database.GetLastPositionInStorageArea(db, cardLocation.StorageAreaId)
	if err != nil {
		return err
	}

	cardLocation.Position = lastPosition + 1

	err = database.InsertCardLocation(db, cardLocation)
	if err != nil {
		return err
	}

	return nil
}

func storeInEmptySlot(db *sql.DB, cardLocation database.CardLocation, cardLocationId int) error {
	cardLocation.Id = cardLocationId
	err := database.InsertCardInExistingLocation(db, cardLocation)
	if err != nil {
		return err
	}

	return nil
}

func StoreCard(db *sql.DB, cardLocation database.CardLocation) error {
	storageAreaType, err := database.GetStorageAreaTypeById(db, cardLocation.StorageAreaId)
	if err != nil {
		return err
	}

	if storageAreaType == database.StorageAreaTypeBinder {
		nextEmptySlotId, err := database.GetNextEmptySlotInBinder(db, cardLocation.StorageAreaId)

		if err == sql.ErrNoRows {
			err = storeAfterLastCard(db, cardLocation)
			if err != nil {
				return err
			}
		} else if err != nil {
			return err
		} else {
			err = storeInEmptySlot(db, cardLocation, nextEmptySlotId)
			if err != nil {
				return err
			}
		}

		return nil
	}

	err = storeAfterLastCard(db, cardLocation)
	if err != nil {
		return err
	}

	return nil
}