package sql

import (
	"database/sql"
	"fmt"

	"github.com/filebrowser/filebrowser/v2/share"
)

type shareBackend struct {
	db *sql.DB
}

type linkRecord interface {
	Scan(dest ...interface{}) error
}

func InitSharesTable(db *sql.DB) error {
	sql := fmt.Sprintf("create table if not exists \"%s\" (hash string, path string, userid integer, expire integer, passwordhash string, token string)", SharesTable)
	_, err := db.Exec(sql)
	checkError(err, "Fail to InitSharesTable")
	return err
}

func parseLink(row linkRecord) (*share.Link, error) {
	path := ""
	hash := ""
	userid := uint(0)
	expire := int64(0)
	passwordhash := ""
	token := ""
	err := row.Scan(&path, &hash, &userid, &expire, &passwordhash, &token)
	if checkError(err, "Fail to parse record for share.Link") {
		return nil, err
	}
	link := share.Link{}
	link.Path = path
	link.Hash = hash
	link.UserID = userid
	link.Expire = expire
	link.PasswordHash = passwordhash
	link.Token = token
	return &link, nil
}

func queryLinks(db *sql.DB, condition string) ([]*share.Link, error) {
	sql := fmt.Sprintf("select hash, path, userid, expire, passwordhash, token from \"%s\"", SharesTable)
	if len(condition) > 0 {
		sql = sql + " where " + condition
	}
	rows, err := db.Query(sql)
	if checkError(err, "Fail to Query links") {
		return nil, err
	}
	var links []*share.Link = []*share.Link{}
	for rows.Next() {
		link, err := parseLink(rows)
		if checkError(err, "Fail to parse record for share.Link") {
			continue
		}
		links = append(links, link)
	}
	return links, nil
}

func (s shareBackend) All() ([]*share.Link, error) {
	return queryLinks(s.db, "")
}

func (s shareBackend) FindByUserID(id uint) ([]*share.Link, error) {
	condition := fmt.Sprintf("userid=%d", id)
	return queryLinks(s.db, condition)
}

func (s shareBackend) GetByHash(hash string) (*share.Link, error) {
	sql := fmt.Sprintf("select hash, path, userid, expire, passwordhash, token from \"%s\" where hash='%s'", SharesTable, hash)
	return parseLink(s.db.QueryRow(sql))
}

func (s shareBackend) GetPermanent(path string, id uint) (*share.Link, error) {
	sql := fmt.Sprintf("select hash, path, userid, expire, passwordhash, token from \"%s\" where path='%s' and userid=%d", SharesTable, path, id)
	return parseLink(s.db.QueryRow(sql))
}

func (s shareBackend) Gets(path string, id uint) ([]*share.Link, error) {
	condition := fmt.Sprintf("userid=%d and path='%s'", id, path)
	return queryLinks(s.db, condition)
}
func (s shareBackend) Save(l *share.Link) error {
	sql := fmt.Sprintf("insert into \"%s\" (hash, path, userid, expire, passwordhash, token) values('%s', '%s', %d, %d, '%s', '%s')", SharesTable, l.Hash, l.Path, l.UserID, l.Expire, l.PasswordHash, l.Token)
	_, err := s.db.Exec(sql)
	checkError(err, "Fail to Save share")
	return err
}
func (s shareBackend) Delete(hash string) error {
	sql := fmt.Sprintf("DELETE FROM \"%s\" WHERE hash='%s'", SharesTable, hash)
	_, err := s.db.Exec(sql)
	checkError(err, "Fail to Delete share")
	return err
}