102 lines
2.6 KiB
Go
102 lines
2.6 KiB
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
|
|
}
|