123 lines
3.0 KiB
Go
123 lines
3.0 KiB
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 checkIfPackageExists(pkg Package, db *sql.DB) bool {
|
|
row := db.QueryRow("Select name from dashboard_package where name = ?", pkg.Name)
|
|
err := row.Scan(&pkg.Name)
|
|
if !errors.Is(err, sql.ErrNoRows) {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
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 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
|
|
}
|