bookhoard/internal/services/conversion_service.go

package services

import (
	"context"
	"crypto/sha256"
	"encoding/hex"
	"fmt"
	"io"
	"os"
	"os/exec"
	"path/filepath"
	"time"

	"bookhoard/internal/database"
	"github.com/jackc/pgx/v5/pgtype"
)

type ConvertedKEPUB struct {
	Path   string
	SHA256 string
	Cached bool
}

type ConversionService struct {
	db                 *database.Queries
	cacheDir           string
	conversionTool     string
	conversionCacheTTL time.Duration
}

func NewConversionService(db *database.Queries, cacheDir string) *ConversionService {
	return &ConversionService{
		db:                 db,
		cacheDir:           cacheDir,
		conversionTool:     "/usr/bin/kepubify",
		conversionCacheTTL: 24 * time.Hour,
	}
}

func (s *ConversionService) ConvertEPUBToKEPUB(ctx context.Context, mediaItemID pgtype.UUID, epubPath string) (*ConvertedKEPUB, error) {
	existing, err := s.db.GetMediaItemFormatByType(ctx, database.GetMediaItemFormatByTypeParams{
		MediaItemID: mediaItemID,
		FormatType:  "kepub",
	})
	if err == nil && existing.FilePath.Valid {
		if time.Since(existing.CreatedAt.Time) < s.conversionCacheTTL {
			return &ConvertedKEPUB{
				Path:   existing.FilePath.String,
				SHA256: existing.FileSha256.String,
				Cached: true,
			}, nil
		}
	}

	kepubPath := filepath.Join(s.cacheDir, fmt.Sprintf("%s.kepub.epub", mediaItemID.Bytes[:]))

	if err := os.MkdirAll(s.cacheDir, 0755); err != nil {
		return nil, fmt.Errorf("failed to create cache directory: %w", err)
	}

	if err := s.convertEPUB(epubPath, kepubPath); err != nil {
		return nil, fmt.Errorf("conversion failed: %w", err)
	}

	kepubSHA256, err := s.calculateSHA256(kepubPath)
	if err != nil {
		return nil, fmt.Errorf("hash calculation failed: %w", err)
	}

	epubFormat, err := s.db.GetMediaItemFormatByType(ctx, database.GetMediaItemFormatByTypeParams{
		MediaItemID: mediaItemID,
		FormatType:  "epub",
	})
	if err != nil {
		return nil, fmt.Errorf("EPUB format not found: %w", err)
	}

	fileinfo, err := os.Stat(kepubPath)
	if err != nil {
		return nil, fmt.Errorf("failed to stat converted file: %w", err)
	}

	var fileSize pgtype.Int8
	fileSize.Scan(int64(fileinfo.Size()))

	_, err = s.db.CreateMediaItemFormat(ctx, database.CreateMediaItemFormatParams{
		MediaItemID:           mediaItemID,
		FormatType:            "kepub",
		FilePath:              pgtype.Text{String: kepubPath, Valid: true},
		FileSha256:            pgtype.Text{String: kepubSHA256, Valid: true},
		FileSizeBytes:         fileSize,
		MimeType:              pgtype.Text{String: "application/vnd.kobo+xml+zip", Valid: true},
		ConvertedFromFormatID: pgtype.UUID{Bytes: epubFormat.ID.Bytes, Valid: true},
	})
	if err != nil {
		return nil, fmt.Errorf("failed to store converted format: %w", err)
	}

	return &ConvertedKEPUB{
		Path:   kepubPath,
		SHA256: kepubSHA256,
		Cached: false,
	}, nil
}

func (s *ConversionService) convertEPUB(epubPath, kepubPath string) error {
	if _, err := os.Stat(s.conversionTool); err == nil {
		cmd := exec.Command(s.conversionTool, "-i", epubPath, "-o", kepubPath)
		if output, err := cmd.CombinedOutput(); err != nil {
			return fmt.Errorf("kepubify failed: %w, output: %s", err, string(output))
		}
		return nil
	}

	cmd := exec.Command("ebook-convert", epubPath, kepubPath, "--output-format", "kepub")
	if output, err := cmd.CombinedOutput(); err != nil {
		return fmt.Errorf("ebook-convert failed: %w, output: %s", err, string(output))
	}
	return nil
}

func (s *ConversionService) calculateSHA256(filePath string) (string, error) {
	file, err := os.Open(filePath)
	if err != nil {
		return "", err
	}
	defer file.Close()

	hasher := sha256.New()
	if _, err := io.Copy(hasher, file); err != nil {
		return "", err
	}

	return hex.EncodeToString(hasher.Sum(nil)), nil
}