132 lines
3.1 KiB
Go
132 lines
3.1 KiB
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, productLocation database.ProductLocation) (int64, error) {
|
|
lastPosition, err := database.GetLastPositionInStorageArea(db, productLocation.StorageAreaId)
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
|
|
productLocation.Position = lastPosition + 1
|
|
|
|
id, err := database.InsertProductLocation(db, productLocation)
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
|
|
return id, nil
|
|
}
|
|
|
|
func storeInEmptySlot(db *sql.DB, productLocation database.ProductLocation, productLocationId int) error {
|
|
productLocation.Id = productLocationId
|
|
err := database.InsertProductInExistingLocation(db, productLocation)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func StoreProduct(db *sql.DB, productLocation database.ProductLocation) (int64, error) {
|
|
storageAreaType, err := database.GetStorageAreaTypeById(db, productLocation.StorageAreaId)
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
|
|
if storageAreaType == database.StorageAreaTypeBinder {
|
|
nextEmptySlotId, err := database.GetNextEmptySlotInBinder(db, productLocation.StorageAreaId)
|
|
|
|
if err == database.ErrNoEmptySlotsInBinder {
|
|
id, err := storeAfterLastCard(db, productLocation)
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
|
|
return id, nil
|
|
} else if err != nil {
|
|
return -1, err
|
|
} else {
|
|
err = storeInEmptySlot(db, productLocation, nextEmptySlotId)
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
}
|
|
|
|
return int64(nextEmptySlotId), nil
|
|
}
|
|
|
|
id, err := storeAfterLastCard(db, productLocation)
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
|
|
return id, nil
|
|
}
|
|
|
|
func Replace(db *sql.DB, productLocationId int, cardtraderBlueprintId int) error {
|
|
productLocation := database.ProductLocation{
|
|
Id: productLocationId,
|
|
CardtraderBlueprintId: cardtraderBlueprintId,
|
|
}
|
|
|
|
return database.InsertProductInExistingLocation(db, productLocation)
|
|
}
|