update tests

2025-05-07 17:03:30 +02:00
parent 8dd2459240
commit 583a68fd90
16 changed files with 1883 additions and 0 deletions
--- a/Database/os.go
+++ b/Database/os.go
@@ -0,0 +1,128 @@
+package database
+
+import (
+	"database/sql"
+	"errors"
+	"log"
+
+	"gopkg.in/guregu/null.v4"
+)
+
+type OS struct {
+	ID           int         `json:"id"`
+	Distribution null.String `json:"distribution"`
+	Version      null.String `json:"version"`
+	EndOfSupport null.String `json:"EndOfSupport"`
+}
+
+func GetOS(db *sql.DB) ([]OS, error) {
+	var oses []OS
+	rows, err := db.Query("SELECT * FROM dashboard_os")
+	if err != nil {
+		log.Println("Error querying list of OS", err)
+	}
+	defer func(rows *sql.Rows) {
+		err := rows.Close()
+		if err != nil {
+			log.Println("Error closing rows stream")
+		}
+	}(rows)
+
+	for rows.Next() {
+		var os OS
+		err := rows.Scan(&os.ID, &os.Distribution, &os.Version, &os.EndOfSupport)
+		if err != nil {
+			log.Println("Error building list", err)
+		}
+		oses = append(oses, os)
+	}
+	return oses, err
+}
+
+func GetOSbyID(db *sql.DB, id int64) (OS, error) {
+	var os OS
+	if err := db.QueryRow("SELECT * FROM dashboard_os where id = ?", id).Scan(&os.ID, &os.Distribution, &os.Version, &os.EndOfSupport); err != nil {
+		if errors.Is(err, sql.ErrNoRows) {
+			return os, nil
+		}
+		log.Println("Error getting OS by ID", err)
+	}
+	return os, nil
+}
+
+func GetDistributionList(db *sql.DB) ([]null.String, error) {
+	var list []null.String
+	rows, err := db.Query("SELECT DISTINCT distribution FROM dashboard_os")
+	if err != nil {
+		log.Println("Error getting distribution list", err)
+	}
+	for rows.Next() {
+		var os OS
+		err := rows.Scan(&os.Distribution)
+		if err != nil {
+			log.Println("Error building distribution list", err)
+		}
+		list = append(list, os.Distribution)
+	}
+	return list, nil
+}
+
+func GetVersionsByDistributionList(db *sql.DB, d string) ([]null.String, error) {
+	var list []null.String
+	rows, err := db.Query("SELECT DISTINCT version FROM dashboard_os where distribution = ?", d)
+	if err != nil {
+		log.Println("Error getting distribution list", err)
+	}
+	for rows.Next() {
+		var os OS
+		err := rows.Scan(&os.Distribution)
+		if err != nil {
+			log.Println("Error building distribution list", err)
+		}
+		list = append(list, os.Distribution)
+	}
+	return list, nil
+}
+
+func checkIfOsExists(os OS, db *sql.DB) bool {
+	row := db.QueryRow("SELECT distribution, version FROM dashboard_os WHERE distribution = ? AND version = ?", os.Distribution, os.Version)
+	err := row.Scan(&os.Distribution, &os.Version)
+	return !errors.Is(err, sql.ErrNoRows)
+}
+
+func CreateOS(os OS, db *sql.DB) error {
+	if checkIfOsExists(os, db) {
+		log.Println("OS already exists")
+		return nil
+	}
+	q, err := db.Prepare("INSERT INTO `dashboard_os` (`distribution`,`version`,`end_of_support`) VALUES (?,?,?)")
+	if err != nil {
+		log.Println("Error creating query", err)
+		return err
+	}
+	_, err = q.Exec(&os.Distribution, &os.Version, &os.EndOfSupport)
+	if err != nil {
+		log.Println("Error inserting OS", err)
+		return err
+	}
+	return nil
+}
+
+func DeleteOS(os OS, db *sql.DB) (int, error) {
+	if !checkIfOsExists(os, db) {
+		log.Println("OS does not exist, nothing to delete")
+		return 404, nil
+	}
+	q, err := db.Prepare("DELETE FROM dashboard_os WHERE distribution = ? AND version = ?")
+	if err != nil {
+		log.Println("Error deleting OS", err)
+		return 500, err
+	}
+	_, err = q.Exec(&os.Distribution, &os.Version)
+	if err != nil {
+		log.Println("Error deleting OS", err)
+		return 500, err
+	}
+	log.Println("OS deleted")
+	return 200, nil
+}
--- a/Database/package.go
+++ b/Database/package.go
@@ -0,0 +1,151 @@
+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
+}
--- a/Database/server.go
+++ b/Database/server.go
@@ -0,0 +1,117 @@
+package database
+
+import (
+	"database/sql"
+	"errors"
+	"log"
+
+	"gopkg.in/guregu/null.v4"
+)
+
+type Server struct {
+	ID       int         `json:"id"`
+	Hostname null.String `json:"hostname"`
+	OsId     null.String `json:"OsId"`
+}
+
+func GetServersList(db *sql.DB) ([]Server, error) {
+	var servers []Server
+	rows, err := db.Query("SELECT * FROM dashboard_server")
+	if err != nil {
+		log.Println("Error querying servers list", err)
+	}
+	defer func(rows *sql.Rows) {
+		err := rows.Close()
+		if err != nil {
+			log.Println("Error closing rows query", err)
+		}
+	}(rows)
+
+	for rows.Next() {
+		var s Server
+		err := rows.Scan(&s.ID, &s.Hostname, &s.OsId)
+		if err != nil {
+			log.Println("Error building servers list", err)
+		}
+		servers = append(servers, s)
+	}
+	return servers, err
+}
+
+func GetServersbyID(db *sql.DB, id int64) Server {
+	var s Server
+	rows := db.QueryRow("SELECT * FROM dashboard_server WHERE id = ?", id)
+	err := rows.Scan(&s.ID, &s.Hostname, &s.OsId)
+	if err != nil {
+		return Server{}
+	}
+	return s
+}
+
+func GetServersbyOS(db *sql.DB, id int64) ([]Server, error) {
+	var servers []Server
+	rows, err := db.Query("SELECT * FROM dashboard_server WHERE os_id = ?", id)
+	if err != nil {
+		log.Println("Error querying servers list by os_id", err)
+		return nil, err
+	}
+	defer func(rows *sql.Rows) {
+		err := rows.Close()
+		if err != nil {
+			log.Println("Error closing rows query", err)
+		}
+	}(rows)
+
+	for rows.Next() {
+		var s Server
+		err := rows.Scan(&s.ID, &s.Hostname, &s.OsId)
+		if err != nil {
+			log.Println("Error building servers list", err)
+		}
+		servers = append(servers, s)
+	}
+	return servers, err
+}
+
+func checkIfServerExists(server Server, db *sql.DB) bool {
+	row := db.QueryRow("SELECT hostname, os_id FROM dashboard_server WHERE hostname = ? AND os_id = ?", server.Hostname, server.OsId)
+	err := row.Scan(&server.Hostname, &server.OsId)
+	return !errors.Is(err, sql.ErrNoRows)
+}
+
+func CreateServer(server Server, db *sql.DB) error {
+	if !checkIfServerExists(server, db) {
+		log.Println("Server already exists")
+		return nil
+	}
+	q, err := db.Prepare("INSERT INTO dashboard_server (hostname, os_id) VALUES ( ?, ?)")
+	if err != nil {
+		log.Println("Error creating query", err)
+		return err
+	}
+	_, err = q.Exec(server.Hostname, server.OsId)
+	if err != nil {
+		log.Println("Error inserting Server", err)
+		return err
+	}
+	return nil
+}
+
+func DeleteServer(server Server, db *sql.DB) (int, error) {
+	if !checkIfServerExists(server, db) {
+		log.Println("Server does not exist, nothing to delete")
+		return 404, nil
+	}
+	q, err := db.Prepare("DELETE FROM dashboard_server WHERE hostname = ? AND os_id = ?")
+	if err != nil {
+		log.Println("Error deleting Server", err)
+		return 500, err
+	}
+	_, err = q.Exec(server.Hostname, server.OsId)
+	if err != nil {
+		log.Println("Error deleting Server", err)
+		return 500, err
+	}
+	log.Println("Server deleted")
+	return 200, nil
+}
--- a/Database/utils.go
+++ b/Database/utils.go
@@ -0,0 +1,24 @@
+package database
+
+import (
+	"database/sql"
+	"log"
+	"os"
+
+	"github.com/go-sql-driver/mysql"
+)
+
+var GetDatabaseConnection = func() *sql.DB {
+	cfg := mysql.Config{
+		User:   os.Getenv("DATABASE_USER"),
+		Passwd: os.Getenv("DATABASE_PASSWORD"),
+		Net:    "tcp",
+		Addr:   os.Getenv("DATABASE_HOST"),
+		DBName: os.Getenv("DATABASE_NAME"),
+	}
+	DbConn, err := sql.Open("mysql", cfg.FormatDSN())
+	if err != nil {
+		log.Println("Database connection error : ", err)
+	}
+	return DbConn
+}