chrly/db/filesystem.go

package db

import (
	"github.com/elyby/chrly/http"
	"os"
	"path"
	"strings"

	"github.com/elyby/chrly/model"
	"github.com/elyby/chrly/mojangtextures"
)

type FilesystemFactory struct {
	BasePath     string
	CapesDirName string
}

func (f FilesystemFactory) CreateSkinsRepository() (http.SkinsRepository, error) {
	panic("skins repository not supported for this storage type")
}

func (f FilesystemFactory) CreateCapesRepository() (http.CapesRepository, error) {
	if err := f.validateFactoryConfig(); err != nil {
		return nil, err
	}

	return &filesStorage{path: path.Join(f.BasePath, f.CapesDirName)}, nil
}

func (f FilesystemFactory) CreateMojangUuidsRepository() (mojangtextures.UuidsStorage, error) {
	panic("implement me")
}

func (f FilesystemFactory) validateFactoryConfig() error {
	if f.BasePath == "" {
		return &ParamRequired{"basePath"}
	}

	if f.CapesDirName == "" {
		f.CapesDirName = "capes"
	}

	return nil
}

type filesStorage struct {
	path string
}

func (repository *filesStorage) FindByUsername(username string) (*model.Cape, error) {
	if username == "" {
		return nil, nil
	}

	capePath := path.Join(repository.path, strings.ToLower(username)+".png")
	file, err := os.Open(capePath)
	if err != nil {
		if os.IsNotExist(err) {
			return nil, nil
		}

		return nil, err
	}

	return &model.Cape{
		File: file,
	}, nil
}