keiji/pkg/helpers/storage.go

package helpers

import (
	"fmt"
	"os"
	"time"

	"git.aetherial.dev/aeth/keiji/pkg/env"
	"github.com/google/uuid"
)

type InvalidSkipArg struct {Skip int}

func (i *InvalidSkipArg) Error() string {
	return fmt.Sprintf("Invalid skip amount was passed: %v", i.Skip)
}


type ImageStoreItem struct {
	Identifier		string	`json:"identifier"`
	Filename		string	`json:"filename"`
	AbsolutePath	string	`json:"absolute_path"`
	Title			string	`json:"title" form:"title"`
	Created			string	`json:"created"`
	Desc			string	`json:"description" form:"description"`
	Category		string	`json:"category"`
}

/*
Create a new ImageStoreItem
	:param fname: the name of the file to be saved
	:param title: the canonical title to give the image
	:param desc: the description to associate to the image
*/
func NewImageStoreItem(fname string, title string, desc string) *ImageStoreItem {
	id := uuid.New()
	img := ImageStoreItem{
		Identifier: id.String(),
		Filename: fname,
		Title: title,
		Category: DIGITAL_ART,
		AbsolutePath: fmt.Sprintf("%s/%s", GetImageStore(), fname),
		Created: time.Now().UTC().String(),
		Desc: desc,
	}
	return &img
}


/*
Function to return the location of the image store. Wrapping the env call in
a function so that refactoring is easier
*/
func GetImageStore() string {
	return os.Getenv(env.IMAGE_STORE)
}

/*
Return all of the filenames of the images that exist in the imagestore location
	:param limit: the limit of filenames to return
	:param skip: the index to start getting images from
*/
func GetImagePaths(limit int, skip int) ([]string, error) {
	f, err := os.ReadDir(GetImageStore())
	if err != nil {
		return nil, err
	}
	if len(f) < skip {
		return nil, &InvalidSkipArg{Skip: skip}
	}
	if len(f) < limit {
		return nil, &InvalidSkipArg{Skip: limit}
	}
	fnames := []string{}
	for i := skip; i < (skip + limit); i++ {
		fnames = append(fnames, fmt.Sprintf("/api/v1/images/%s", f[i].Name()))
	}
	return fnames, err
}