owl-blogs/infra/binary_file_repository.go

package infra

import (
	"fmt"
	"owl-blogs/app/repository"
	"owl-blogs/domain/model"
	"strings"

	"github.com/jmoiron/sqlx"
)

type sqlBinaryFile struct {
	Id      string  `db:"id"`
	Name    string  `db:"name"`
	EntryId *string `db:"entry_id"`
	Data    []byte  `db:"data"`
}

type DefaultBinaryFileRepo struct {
	db *sqlx.DB
}

// NewBinaryFileRepo creates a new binary file repository
// It creates the table if not exists
func NewBinaryFileRepo(db Database) repository.BinaryRepository {
	sqlxdb := db.Get()

	// Create table if not exists
	sqlxdb.MustExec(`
		CREATE TABLE IF NOT EXISTS binary_files (
			id VARCHAR(255) PRIMARY KEY,
			name VARCHAR(255) NOT NULL,
			entry_id VARCHAR(255),
			data BLOB NOT NULL
		);
	`)

	return &DefaultBinaryFileRepo{db: sqlxdb}
}

// Create implements repository.BinaryRepository
func (repo *DefaultBinaryFileRepo) Create(name string, data []byte, entry model.Entry) (*model.BinaryFile, error) {
	parts := strings.Split(name, ".")
	fileName := strings.Join(parts[:len(parts)-1], ".")
	fileExt := parts[len(parts)-1]
	id := fileName + "." + fileExt

	// check if id exists
	var count int
	err := repo.db.Get(&count, "SELECT COUNT(*) FROM binary_files WHERE id = ?", id)
	if err != nil {
		return nil, err
	}

	if count > 0 {
		counter := 1
		for {
			id = fmt.Sprintf("%s-%d.%s", fileName, counter, fileExt)
			err := repo.db.Get(&count, "SELECT COUNT(*) FROM binary_files WHERE id = ?", id)
			if err != nil {
				return nil, err
			}
			if count == 0 {
				break
			}
			counter++
		}
	}

	var entryId *string
	if entry != nil {
		eId := entry.ID()
		entryId = &eId
	}

	_, err = repo.db.Exec("INSERT INTO binary_files (id, name, entry_id, data) VALUES (?, ?, ?, ?)", id, name, entryId, data)
	if err != nil {
		return nil, err
	}
	return &model.BinaryFile{Id: id, Name: name, Data: data}, nil
}

// FindById implements repository.BinaryRepository
func (repo *DefaultBinaryFileRepo) FindById(id string) (*model.BinaryFile, error) {
	var sqlFile sqlBinaryFile
	err := repo.db.Get(&sqlFile, "SELECT * FROM binary_files WHERE id = ?", id)
	if err != nil {
		return nil, err
	}
	return &model.BinaryFile{Id: sqlFile.Id, Name: sqlFile.Name, Data: sqlFile.Data}, nil
}

// FindByNameForEntry implements repository.BinaryRepository
func (repo *DefaultBinaryFileRepo) FindByNameForEntry(name string, entry model.Entry) (*model.BinaryFile, error) {
	var sqlFile sqlBinaryFile
	err := repo.db.Get(&sqlFile, "SELECT * FROM binary_files WHERE name = ? AND entry_id = ?", name, entry.ID())
	if err != nil {
		return nil, err
	}
	return &model.BinaryFile{Id: sqlFile.Id, Name: sqlFile.Name, Data: sqlFile.Data}, nil
}