Git structure files

app/handlers/api/api.go

@@ -0,0 +1,101 @@
+package api
+
+import (
+	"database/sql"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/url"
+
+	"gitea.voser.cloud/Golang/GoDown/handlers/sqldb"
+	"github.com/gin-gonic/gin"
+)
+
+// Global Variables
+var DB *sql.DB
+
+// check error
+func check(err error) {
+	if err != nil {
+		panic(err.Error())
+	}
+}
+
+func GET_api_db(c *gin.Context) {
+	dbquery := fmt.Sprintf("SELECT * FROM testdb.media")
+	media, err := getJSON(dbquery)
+	check(err)
+	c.Data(http.StatusOK, "application/json", media)
+}
+
+func GET_api_add_ytid(c *gin.Context) {
+	newSong_ytid := c.Param("ytid")
+	newSong_url := "https://www.youtube.com/watch?v=" + newSong_ytid
+	_, err := url.ParseRequestURI(newSong_url)
+	check(err)
+	is_available := ytid_in_db(newSong_ytid)
+	c.String(http.StatusOK, is_available)
+}
+
+func ytid_in_db(ytid string) string {
+	sqlString := fmt.Sprintf("SELECT COUNT(ytid) FROM testdb.media WHERE ytid LIKE \"%v\";", ytid)
+	var count int
+	err := DB.QueryRow(sqlString).Scan(&count)
+	check(err)
+	if count == 0 {
+		return "Song is not available"
+	}
+	if count == 1 {
+		return "Song is available"
+	}
+	error_return := fmt.Sprintf("Rows returned: %s --> there is an Issue within the database", string(count))
+	return error_return
+}
+
+func GET_api_get_ytid(c *gin.Context) {
+	ytid := c.Param("ytid")
+	sqlString := fmt.Sprintf("SELECT * FROM testdb.media WHERE ytid LIKE \"%v\";", ytid)
+	json, err := getJSON(sqlString)
+	check(err)
+	c.Data(http.StatusOK, "application/json", json)
+}
+
+func getJSON(sqlString string) ([]byte, error) {
+	DB = sqldb.Opendb()
+	defer DB.Close()
+	rows, err := DB.Query(sqlString)
+	check(err)
+	defer rows.Close()
+	columns, err := rows.Columns()
+	if err != nil {
+		return []byte("Error while generating json"), err
+	}
+	count := len(columns)
+	tableData := make([]map[string]interface{}, 0)
+	values := make([]interface{}, count)
+	valuePtrs := make([]interface{}, count)
+	for rows.Next() {
+		for i := 0; i < count; i++ {
+			valuePtrs[i] = &values[i]
+		}
+		rows.Scan(valuePtrs...)
+		entry := make(map[string]interface{})
+		for i, col := range columns {
+			var v interface{}
+			val := values[i]
+			b, ok := val.([]byte)
+			if ok {
+				v = string(b)
+			} else {
+				v = val
+			}
+			entry[col] = v
+		}
+		tableData = append(tableData, entry)
+	}
+	jsonData, err := json.Marshal(tableData)
+	if err != nil {
+		return []byte("Error while generating json"), err
+	}
+	return jsonData, nil
+}

app/handlers/sqldb/sqldb.go

@@ -0,0 +1,70 @@
+package sqldb
+
+import (
+	"database/sql"
+	"fmt"
+
+	"github.com/spf13/viper"
+)
+
+var DB *sql.DB
+var dbUser, dbPass, dbHost, dbPort, dbName string
+
+// check error
+func check(err error) {
+	if err != nil {
+		panic(err.Error())
+	}
+}
+
+// read db cred
+func ReaddbCred() {
+	dbCred := viper.New()
+	dbCred.SetConfigName("db-cred")
+	dbCred.AddConfigPath("../")
+	dbCred.SetConfigType("yaml")
+	err := dbCred.ReadInConfig()
+	check(err)
+	dbUser = dbCred.GetString("dbUser")
+	dbPass = dbCred.GetString("dbPass")
+	dbHost = dbCred.GetString("dbHost")
+	dbPort = dbCred.GetString("dbPort")
+	dbName = dbCred.GetString("dbName")
+}
+
+func Opendb() (db *sql.DB) {
+	// open db
+	sqlopen := fmt.Sprintf(dbUser + ":" + dbPass + "@tcp(" + dbHost + ":" + dbPort + ")/" + dbName)
+	var err error
+	DB, err = sql.Open("mysql", sqlopen)
+	check(err)
+	fmt.Println("Successfully connected to the database")
+	return DB
+}
+
+func Dbinit() {
+	dbinit, err := DB.Query("CREATE TABLE IF NOT EXISTS `media` (`id` INT NOT NULL AUTO_INCREMENT, `ytid` VARCHAR(100), `url` VARCHAR(255), `localpath` TEXT(65535), `size` INT, `sha256sum` VARCHAR(255), `length` INT, `title` VARCHAR(255), `author` VARCHAR(255), PRIMARY KEY (`id`));")
+	if err != nil {
+		fmt.Println("Error while creating table")
+		panic(err.Error())
+	}
+	defer dbinit.Close()
+	fmt.Println("DB-Init successfull")
+}
+
+func Dbentrycount() {
+	var count int
+	dbcount, err := DB.Query("SELECT COUNT(*) FROM testdb.media;")
+	if err != nil {
+		fmt.Println("Error while counting rows from table")
+		panic(err.Error())
+	}
+	defer dbcount.Close()
+	for dbcount.Next() {
+		err = dbcount.Scan(&count)
+		if err != nil {
+			panic(err.Error())
+		}
+	}
+	fmt.Printf("There are %v entries in the database\n", count)
+}