package scanner

import (
	"database/sql"
	"github.com/viktorstrate/photoview/api/graphql/models"
	"log"
	"path"
)

func ScanPhoto(tx *sql.Tx, photoPath string, albumId int, content_type *string) error {

	log.Printf("Scanning image: %s\n", photoPath)

	photoName := path.Base(photoPath)

	// Check if image already exists
	row := tx.QueryRow("SELECT (photo_id) FROM photo WHERE path = ?", photoPath)
	var photo_id int64
	if err := row.Scan(&photo_id); err != sql.ErrNoRows {
		if err == nil {
			log.Printf("Image already scanned: %s\n", photoPath)
			return nil
		} else {
			return err
		}
	}

	result, err := tx.Exec("INSERT INTO photo (title, path, album_id) VALUES (?, ?, ?)", photoName, photoPath, albumId)
	if err != nil {
		log.Printf("ERROR: Could not insert photo into database")
		return err
	}
	photo_id, err = result.LastInsertId()
	if err != nil {
		return err
	}

	row = tx.QueryRow("SELECT * FROM photo WHERE photo_id = ?", photo_id)
	photo, err := models.NewPhotoFromRow(row)
	if err != nil {
		return err
	}

	if err := ProcessPhoto(tx, photo, content_type); err != nil {
		return err
	}

	return nil
}