package database

import (
	"context"

	"certmanager/internal/descriptor"
	"certmanager/pkg/auxid"

	_ "github.com/mattn/go-sqlite3"
)

func (db *Database) InsertService(ctx context.Context, service *descriptor.Service) (int64, error) {
	var err error
	var res int64
	service.ID = auxid.GenID()
	request := `INSERT INTO service(id, issuer_id, name, cert, key, revoked)
                        VALUES ($1, $2, $3, $4, $5, $6)`
	_, err = db.db.Exec(request, service.ID, service.IssuerID, service.Name, service.Cert,
		service.Key, service.Revoked)
	if err != nil {
		return res, err
	}
	res = service.ID
	return res, err
}

func (db *Database) UpdateServiceByID(ctx context.Context, serviceID int64, service *descriptor.Service) error {
	var err error
	request := `UPDATE service SET revoked = $1 WHERE id = $2`
	_, err = db.db.Exec(request, service.Revoked, serviceID)
	if err != nil {
		return err
	}
	return err
}

func (db *Database) ListServices(ctx context.Context) ([]descriptor.Service, error) {
	var err error
	request := `SELECT id, name, revoked FROM service`
	res := make([]descriptor.Service, 0)
	err = db.db.Select(&res, request)
	if err != nil {
		return res, err
	}
	return res, err
}

func (db *Database) GetServiceByID(ctx context.Context, serviceID int64) (bool, *descriptor.Service, error) {
	var err error
	var res *descriptor.Service
	var exists bool
	request := `SELECT id, name, cert, key, revoked FROM service WHERE id = $1 LiMIT 1`
	dbRes := make([]descriptor.Service, 0)
	err = db.db.Select(&dbRes, request, serviceID)
	if err != nil {
		return exists, res, err
	}
	if len(dbRes) == 0 {
		return exists, res, err
	}
	exists = true
	res = &dbRes[0]
	return exists, res, err
}

func (db *Database) GetServiceByName(ctx context.Context, serviceName string) (bool, *descriptor.Service, error) {
	var err error
	var res *descriptor.Service
	var exists bool
	request := `SELECT id, name, cert, key, revoked FROM service WHERE name = $1 LIMIT 1`
	dbRes := make([]descriptor.Service, 0)
	err = db.db.Select(&dbRes, request, serviceName)
	if err != nil {
		return exists, res, err
	}
	if len(dbRes) == 0 {
		return exists, res, err
	}
	exists = true
	res = &dbRes[0]
	return exists, res, err
}

func (db *Database) DeleteServiceByID(ctx context.Context, serviceID int64) error {
	var err error
	request := `DELETE FROM service WHERE id = $1`
	_, err = db.db.Exec(request, serviceID)
	if err != nil {
		return err
	}
	return err
}

func (db *Database) DeleteServiceByName(ctx context.Context, serviceName string) error {
	var err error
	request := `DELETE FROM service WHERE name = $1`
	_, err = db.db.Exec(request, serviceName)
	if err != nil {
		return err
	}
	return err
}