package utils

import (
	"errors"
	"os"
	"path"

	"github.com/pelletier/go-toml"
	"gorm.io/gorm"
)

func StopIfErr(err error)	{
	if err != nil	{
		ErrPrint(err.Error())
		os.Exit(1)
	}
}

func GetEnvy() (*toml.Tree, error)	{
	cwd, err := os.Getwd()
	if err != nil {
		return nil, errors.New("Not able to fetch working directory: " + err.Error())
	}

	_, err = os.Stat(path.Join(cwd, ".envy"))

	// file exists
	if err == nil {
		envyFile, err := toml.LoadFile(path.Join(cwd, ".envy"))
		if err != nil	{
			return nil, errors.New("Error occured while reading .envy file:" + err.Error())
		}
		return envyFile, nil
	}

	// file does not exist
	if os.IsNotExist(err)	{
		return nil, errors.New(".envy file doesn't exist in the current directory.")
	} else	{
		return nil, errors.New("Some error occured:" + err.Error())
	}
}


func GetProject(name string) (Project, error)	{
	InitDb()

	var dbProject Project
	result := DB.Where("name = ?", name).First(&dbProject)

	if result.Error != nil	{
		if errors.Is(result.Error, gorm.ErrRecordNotFound)	{
			return Project{}, errors.New("Project with the name " + name + " not found in the database.")
		} else	{
			return Project{}, errors.New("Some error occured: " + result.Error.Error())
		}
	}
	return dbProject, nil
}


func GetEnvironments(project uint) ([]Environment, error)	{
	InitDb()

	var dbEnvs []Environment
	result := DB.Where("project_id = ?", project).Find(&dbEnvs)

	if result.Error != nil	{
		return nil, errors.New("Some error occured: " + result.Error.Error())
	}
	return dbEnvs, nil
}


func GetEnvironment(project uint, environment string) (Environment, error)	{
	InitDb()

	var dbEnv Environment
	result := DB.Where("project_id = ? AND name = ?", project, environment).First(&dbEnv)

	if result.Error != nil	{
		if errors.Is(result.Error, gorm.ErrRecordNotFound)	{
			return Environment{}, errors.New("Environment with the name " + environment + " not found in the database.")
		} else	{
			return Environment{}, errors.New("Some error occured: " + result.Error.Error())
		}
	}
	return dbEnv, nil
}