102 lines
2.3 KiB
Go
102 lines
2.3 KiB
Go
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
|
|
}
|