Adding create/disable/delete packages
This commit is contained in:
@@ -88,7 +88,6 @@ 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)
|
||||
if !errors.Is(err, sql.ErrNoRows) {
|
||||
log.Println("OS exists")
|
||||
return true
|
||||
}
|
||||
return false
|
||||
|
||||
@@ -43,3 +43,80 @@ func GetPackagebyID(db *sql.DB, id int64) (Package, error) {
|
||||
}
|
||||
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
|
||||
}
|
||||
|
||||
@@ -75,7 +75,6 @@ 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)
|
||||
if !errors.Is(err, sql.ErrNoRows) {
|
||||
log.Println("Server exists")
|
||||
return true
|
||||
}
|
||||
return false
|
||||
|
||||
Reference in New Issue
Block a user