infra-dashboard/app/Database/package.go

package database

import (
	"database/sql"
	"errors"
	"log"

	"gopkg.in/guregu/null.v4"
)

type Package struct {
	ID     int         `json:"id"`
	Name   null.String `json:"name"`
	Active int         `json:"active"`
}

func GetPackage(db *sql.DB) ([]Package, error) {
	var packages []Package
	rows, err := db.Query("SELECT * FROM dashboard_package")
	if err != nil {
		log.Println("Error querying list of OS", err)
	}
	defer rows.Close()

	for rows.Next() {
		var pkg Package
		err := rows.Scan(&pkg.ID, &pkg.Name, &pkg.Active)
		if err != nil {
			log.Println("Error building list", err)
		}
		packages = append(packages, pkg)
	}
	return packages, err
}

func GetPackagebyID(db *sql.DB, id int64) (Package, error) {
	var pkg Package
	if err := db.QueryRow("SELECT * FROM dashboard_package where id = ?", id).Scan(&pkg.ID, &pkg.Name, &pkg.Active); err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return pkg, nil
		}
		log.Println("Error getting OS by ID", err)
	}
	return pkg, nil
}

func GetPackagebyName(db *sql.DB, id int64) (Package, error) {
	var pkg Package
	if err := db.QueryRow("SELECT * FROM dashboard_package where name = ?", id).Scan(&pkg.ID, &pkg.Name, &pkg.Active); err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return pkg, nil
		}
		log.Println("Error getting OS by ID", err)
	}
	return pkg, nil
}

func checkIfPackageExists(pkg Package, db *sql.DB) bool {
	if err := db.QueryRow("Select name from dashboard_package where name = ?", pkg.Name).Scan(&pkg.Name); err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return false
		}
	}
	return true
}

func CreatePackage(pkg Package, db *sql.DB) error {
	if checkIfPackageExists(pkg, db) {
		/* If package exists but is not active we activate it */
		row := db.QueryRow("Select name, active from dashboard_package where name = ? and active = 1", pkg.Name)
		err := row.Scan(&pkg.Name, &pkg.Active)
		if errors.Is(err, sql.ErrNoRows) {
			q, err := db.Prepare("UPDATE dashboard_package set active = 1 where name = ?")
			if err != nil {
				log.Println("Error creating query", err)
			}
			_, err = q.Exec(pkg.Name)
			if err != nil {
				log.Println("Error activating Package", err)
			}
			log.Println("Activating new package", pkg.Name)
			return nil
		}
		return nil
	}
	q, err := db.Prepare("INSERT INTO dashboard_package (name, active) values (?, 1)")
	if err != nil {
		log.Println("Error creating query", err)
		return err
	}
	_, err = q.Exec(pkg.Name)
	if err != nil {
		log.Println("Error creating Package", err)
		return err
	}
	return nil
}

func DisablePackage(pkg Package, db *sql.DB) (int, error) {
	if !checkIfPackageExists(pkg, db) {
		log.Println("Package does not exist", pkg.Name)
		return 404, nil
	}
	q, err := db.Prepare("UPDATE dashboard_package set active = 0 where name = ?")
	if err != nil {
		log.Println("Error creating query", err)
		return 500, err
	}
	_, err = q.Exec(pkg.Name)
	if err != nil {
		log.Println("Error disabling Package", err)
		return 500, err
	}
	return 200, nil
}

func EnablePackage(pkg Package, db *sql.DB) (int, error) {
	if !checkIfPackageExists(pkg, db) {
		log.Println("Package does not exist", pkg.Name)
		return 404, nil
	}
	q, err := db.Prepare("UPDATE dashboard_package set active = 1 where name = ?")
	if err != nil {
		log.Println("Error creating query", err)
		return 500, err
	}
	_, err = q.Exec(pkg.Name)
	if err != nil {
		log.Println("Error disabling Package", err)
		return 500, err
	}
	return 200, nil
}

func DeletePackage(pkg Package, db *sql.DB) (int, error) {
	if !checkIfPackageExists(pkg, db) {
		log.Println("Package does not exist", pkg.Name)
		return 404, nil
	}
	q, err := db.Prepare("DELETE FROM dashboard_package where name = ?")
	if err != nil {
		log.Println("Error creating query", err)
		return 500, err
	}
	_, err = q.Exec(pkg.Name)
	if err != nil {
		log.Println("Error deleting Package", err)
		return 500, err
	}
	return 200, nil
}