commit bbed7f94f484d92c2632b4d05ed69b2ae6643c33 Author: Gregory Salaun Date: Sat Apr 15 16:23:34 2023 +0700 first commit diff --git a/.env b/.env new file mode 100644 index 0000000..605dfd3 --- /dev/null +++ b/.env @@ -0,0 +1,4 @@ +DOWNLOAD_FOLDER=/home/rouggy/torrents/rtorrent/Race +HTTP_PORT=3000 +DB_NAME=racer +TMDB_APIKEY=41d05b7a36ba961740f7c05cc4ef634b diff --git a/.idea/.gitignore b/.idea/.gitignore new file mode 100644 index 0000000..13566b8 --- /dev/null +++ b/.idea/.gitignore @@ -0,0 +1,8 @@ +# Default ignored files +/shelf/ +/workspace.xml +# Editor-based HTTP Client requests +/httpRequests/ +# Datasource local storage ignored files +/dataSources/ +/dataSources.local.xml diff --git a/.idea/RaceBot.iml b/.idea/RaceBot.iml new file mode 100644 index 0000000..5e764c4 --- /dev/null +++ b/.idea/RaceBot.iml @@ -0,0 +1,9 @@ + + + + + + + + + \ No newline at end of file diff --git a/.idea/modules.xml b/.idea/modules.xml new file mode 100644 index 0000000..86e44cb --- /dev/null +++ b/.idea/modules.xml @@ -0,0 +1,8 @@ + + + + + + + + \ No newline at end of file diff --git a/.idea/vcs.xml b/.idea/vcs.xml new file mode 100644 index 0000000..94a25f7 --- /dev/null +++ b/.idea/vcs.xml @@ -0,0 +1,6 @@ + + + + + + \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..e69de29 diff --git a/api/server.go b/api/server.go new file mode 100644 index 0000000..69d9cb5 --- /dev/null +++ b/api/server.go @@ -0,0 +1,43 @@ +package api + +import ( + "fmt" + "git.rouggy.com/rouggy/RaceBot/config" + "git.rouggy.com/rouggy/RaceBot/controllers" + "github.com/gin-gonic/gin" +) + +type Server struct { + listenAddr string +} + +func NewServer(listenAddr string) *Server { + return &Server{ + listenAddr: listenAddr, + } +} + +func (s *Server) Start() error { + gin.SetMode(gin.ReleaseMode) + r := gin.Default() + SetupRoutes(r) + fmt.Println("Server is running on port:", config.HTTPPort) + r.Run(":" + config.HTTPPort) +} + +func SetupRoutes(app *gin.Engine) { + + api := app.Group("/api") + v1 := api.Group("/v1") + + v1.GET("/races", controllers.GetAllRaces) + //v1.GET("/races/:id", controllers.GetReleaseById) + //v1.POST("/races", controllers.CreateRace) + //v1.PUT("/races/:id", controllers.UpdateRace) + //v1.DELETE("/races/:id", controllers.DeleteRace) + //v1.GET("/releases", controllers.GetReleases) + //v1.GET("/releases/:id", controllers.GetReleaseById) + v1.POST("/preraces", controllers.CreatePreRace) + //v1.PUT("/releases/:id", controllers.UpdateRelease) + //v1.DELETE("/releases/:id", controllers.DeleteRelease) +} diff --git a/config/config.go b/config/config.go new file mode 100644 index 0000000..1942c1a --- /dev/null +++ b/config/config.go @@ -0,0 +1,38 @@ +package config + +import ( + "log" + "os" + + "github.com/joho/godotenv" +) + +var DownloadFolder string +var HTTPPort string +var DBName string +var TMDbApiKey string + +func Load() { + if err := godotenv.Load(); err != nil { + log.Println("Error loading .env file ...") + log.Println("Using default configuration") + } + + if DownloadFolder = os.Getenv("DOWNLOAD_FOLDER"); DownloadFolder == "" { + DownloadFolder = "/home/rouggy/torrents/rtorrent/Race" + } + + if HTTPPort = os.Getenv("HTTP_PORT"); HTTPPort == "" { + HTTPPort = "9000" + } + + if DBName = os.Getenv("DB_NAME"); DBName == "" { + DBName = "racerdb" + } + + if TMDbApiKey = os.Getenv("TMDB_APIKEY"); TMDbApiKey == "" { + log.Println("TMDb Api Key cannot be empty...") + os.Exit(1) + } + +} diff --git a/controllers/race.go b/controllers/race.go new file mode 100644 index 0000000..7db955e --- /dev/null +++ b/controllers/race.go @@ -0,0 +1,41 @@ +package controllers + +import ( + "fmt" + "git.rouggy.com/rouggy/RaceBot/database" + "git.rouggy.com/rouggy/RaceBot/helper" + "git.rouggy.com/rouggy/RaceBot/models" + "github.com/gin-gonic/gin" + "time" +) + +func CreatePreRace(c *gin.Context) { + start := time.Now() + r := models.NewRace() + rel := models.NewRelease() + + if err := c.ShouldBind(&r); err != nil { + fmt.Println("Error: %v", err) + } + + rel.ParseString(r.TorrentName) + + searchRace, _ := r.GetRaceByName(r.TorrentName) + + if searchRace.ID == 0 { + r.SaveRace() + helper.RespondJSON(c, 200, "New race saved", r) + fmt.Println("Elapsed: ", time.Since(start)) + } else { + helper.RespondJSON(c, 200, "Race already existing", searchRace) + fmt.Println("Elapsed: ", time.Since(start)) + } +} + +func GetAllRaces(c *gin.Context) { + var r []models.Race + db := database.GetDB() + db.Find(&r) + + helper.RespondJSON(c, 200, "Get all races", r) +} diff --git a/controllers/release.go b/controllers/release.go new file mode 100644 index 0000000..2d32936 --- /dev/null +++ b/controllers/release.go @@ -0,0 +1 @@ +package controllers diff --git a/database/database.go b/database/database.go new file mode 100644 index 0000000..8161dad --- /dev/null +++ b/database/database.go @@ -0,0 +1,13 @@ +package database + +import ( + "github.com/jinzhu/gorm" +) + +var db *gorm.DB +var err error + +// GetDB : get current connection +func GetDB() *gorm.DB { + return db +} diff --git a/database/sqlite.go b/database/sqlite.go new file mode 100644 index 0000000..b5da249 --- /dev/null +++ b/database/sqlite.go @@ -0,0 +1,20 @@ +package database + +import ( + "fmt" + "git.rouggy.com/rouggy/RaceBot/config" + "github.com/jinzhu/gorm" + _ "github.com/mattn/go-sqlite3" +) + +// SQLiteDBConnect : Create Connection to database +func SQLiteDBConnect() { + //Connect to database, exit when errored + db, err = gorm.Open("sqlite3", "./"+config.DBName+".db") + if err != nil { + panic("[Database] Failed to connect to database") + } + fmt.Println("[Database] Database successfully connected") + //If set true then print all executed queries to the console + db.LogMode(true) +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..6dfd92d --- /dev/null +++ b/go.mod @@ -0,0 +1,38 @@ +module git.rouggy.com/rouggy/RaceBot + +go 1.20 + +require ( + github.com/bytedance/sonic v1.8.0 // indirect + github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect + github.com/gin-contrib/sse v0.1.0 // indirect + github.com/gin-gonic/gin v1.9.0 // indirect + github.com/go-playground/locales v0.14.1 // indirect + github.com/go-playground/universal-translator v0.18.1 // indirect + github.com/go-playground/validator/v10 v10.11.2 // indirect + github.com/goccy/go-json v0.10.0 // indirect + github.com/jinzhu/gorm v1.9.16 // indirect + github.com/jinzhu/inflection v1.0.0 // indirect + github.com/jinzhu/now v1.1.5 // indirect + github.com/joho/godotenv v1.5.1 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/klauspost/cpuid/v2 v2.0.9 // indirect + github.com/leodido/go-urn v1.2.1 // indirect + github.com/mattn/go-isatty v0.0.17 // indirect + github.com/mattn/go-sqlite3 v1.14.16 // indirect + github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect + github.com/modern-go/reflect2 v1.0.2 // indirect + github.com/moistari/rls v0.5.9 // indirect + github.com/pelletier/go-toml/v2 v2.0.6 // indirect + github.com/twitchyliquid64/golang-asm v0.15.1 // indirect + github.com/ugorji/go/codec v1.2.9 // indirect + golang.org/x/arch v0.0.0-20210923205945-b76863e36670 // indirect + golang.org/x/crypto v0.5.0 // indirect + golang.org/x/net v0.7.0 // indirect + golang.org/x/sync v0.1.0 // indirect + golang.org/x/sys v0.5.0 // indirect + golang.org/x/text v0.8.0 // indirect + google.golang.org/protobuf v1.28.1 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect + gorm.io/gorm v1.24.6 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..8de287b --- /dev/null +++ b/go.sum @@ -0,0 +1,109 @@ +github.com/PuerkitoBio/goquery v1.5.1/go.mod h1:GsLWisAFVj4WgDibEWF4pvYnkVQBpKBKeU+7zCJoLcc= +github.com/andybalholm/cascadia v1.1.0/go.mod h1:GsXiBklL0woXo1j/WYWtSYYC4ouU9PqHO0sqidkEA4Y= +github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM= +github.com/bytedance/sonic v1.8.0 h1:ea0Xadu+sHlu7x5O3gKhRpQ1IKiMrSiHttPF0ybECuA= +github.com/bytedance/sonic v1.8.0/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U= +github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY= +github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams= +github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/denisenkom/go-mssqldb v0.0.0-20191124224453-732737034ffd/go.mod h1:xbL0rPBG9cCiLr28tMa8zpbdarY27NDyej4t/EjAShU= +github.com/erikstmartin/go-testdb v0.0.0-20160219214506-8d10e4a1bae5/go.mod h1:a2zkGnVExMxdzMo3M0Hi/3sEU+cWnZpSni0O6/Yb/P0= +github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE= +github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI= +github.com/gin-gonic/gin v1.9.0 h1:OjyFBKICoexlu99ctXNR2gg+c5pKrKMuyjgARg9qeY8= +github.com/gin-gonic/gin v1.9.0/go.mod h1:W1Me9+hsUSyj3CePGrd1/QrKJMSJ1Tu/0hFEH89961k= +github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= +github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= +github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= +github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY= +github.com/go-playground/validator/v10 v10.11.2 h1:q3SHpufmypg+erIExEKUmsgmhDTyhcJ38oeKGACXohU= +github.com/go-playground/validator/v10 v10.11.2/go.mod h1:NieE624vt4SCTJtD87arVLvdmjPAeV8BQlHtMnw9D7s= +github.com/go-sql-driver/mysql v1.5.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg= +github.com/goccy/go-json v0.10.0 h1:mXKd9Qw4NuzShiRlOXKews24ufknHO7gx30lsDyokKA= +github.com/goccy/go-json v0.10.0/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= +github.com/golang-sql/civil v0.0.0-20190719163853-cb61b32ac6fe/go.mod h1:8vg3r2VgvsThLBIFL93Qb5yWzgyZWhEmBwUJWevAkK0= +github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= +github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/jinzhu/gorm v1.9.16 h1:+IyIjPEABKRpsu/F8OvDPy9fyQlgsg2luMV2ZIH5i5o= +github.com/jinzhu/gorm v1.9.16/go.mod h1:G3LB3wezTOWM2ITLzPxEXgSkOXAntiLHS7UdBefADcs= +github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= +github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc= +github.com/jinzhu/now v1.0.1/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= +github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ= +github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= +github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0= +github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4= +github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= +github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= +github.com/klauspost/cpuid/v2 v2.0.9 h1:lgaqFMSdTdQYdZ04uHyN2d/eKdOMyi2YLSvlQIBFYa4= +github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg= +github.com/leodido/go-urn v1.2.1 h1:BqpAaACuzVSgi/VLzGZIobT2z4v53pjosyNd9Yv6n/w= +github.com/leodido/go-urn v1.2.1/go.mod h1:zt4jvISO2HfUBqxjfIshjdMTYS56ZS/qv49ictyFfxY= +github.com/lib/pq v1.1.1/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo= +github.com/mattn/go-isatty v0.0.17 h1:BTarxUcIeDqL27Mc+vyvdWYSL28zpIhv3RoTdsLMPng= +github.com/mattn/go-isatty v0.0.17/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= +github.com/mattn/go-sqlite3 v1.14.0/go.mod h1:JIl7NbARA7phWnGvh0LKTyg7S9BA+6gx71ShQilpsus= +github.com/mattn/go-sqlite3 v1.14.16 h1:yOQRA0RpS5PFz/oikGwBEqvAWhWg5ufRz4ETLjwpU1Y= +github.com/mattn/go-sqlite3 v1.14.16/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= +github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/moistari/rls v0.5.9 h1:peRGW+1/HJDUZ76s0v2ukcBLCBUs4/Qf3TKOzRjOOco= +github.com/moistari/rls v0.5.9/go.mod h1:/3P63JjNkaf1MNBoS2tSXqGeqee6l4je+Krakp4ob7c= +github.com/pelletier/go-toml/v2 v2.0.6 h1:nrzqCb7j9cDFj2coyLNLaZuJTLjWjlaz6nvTvIwycIU= +github.com/pelletier/go-toml/v2 v2.0.6/go.mod h1:eumQOmlWiOPt5WriQQqoM5y18pDHwha2N+QD+EUNTek= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= +github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= +github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= +github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= +github.com/ugorji/go/codec v1.2.9 h1:rmenucSohSTiyL09Y+l2OCk+FrMxGMzho2+tjr5ticU= +github.com/ugorji/go/codec v1.2.9/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670 h1:18EFjUmQOcUvxNYSkA6jO9VAiXCnxFY6NyDX0bHDmkU= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20190325154230-a5d413f7728c/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20191205180655-e7c4368fe9dd/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/crypto v0.5.0 h1:U/0M97KRkSFvyD/3FSmdP5W5swImpNgle/EHFhOsQPE= +golang.org/x/crypto v0.5.0/go.mod h1:NK/OQwhpMQP3MwtdjgLlYHnH9ebylxKWv3e0fK+mkQU= +golang.org/x/net v0.0.0-20180218175443-cbe0f9307d01/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20200202094626-16171245cfb2/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20200324143707-d3edc9973b7e/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A= +golang.org/x/net v0.7.0 h1:rJrUqqhjsgNp7KqAIc25s9pZnjU7TUcSY7HcVZjdn1g= +golang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= +golang.org/x/sync v0.1.0 h1:wsuoTGHzEhffawBOhz5CYhcrV4IdKZbEyZjBMuTp12o= +golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.5.0 h1:MUK/U/4lj1t1oPg0HfuXDN/Z1wv31ZJ/YcPiGccS4DU= +golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.7.0 h1:4BRB4x83lYWy72KwLD/qYDuTu7q9PjSagHvijDw7cLo= +golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= +golang.org/x/text v0.8.0 h1:57P1ETyNKtuIjB4SRd15iJxuhj8Gc416Y78H3qgMh68= +golang.org/x/text v0.8.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= +google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175w= +google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gorm.io/gorm v1.24.6 h1:wy98aq9oFEetsc4CAbKD2SoBCdMzsbSIvSUUFJuHi5s= +gorm.io/gorm v1.24.6/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k= +rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4= diff --git a/helper/reponse.go b/helper/reponse.go new file mode 100644 index 0000000..eb6189c --- /dev/null +++ b/helper/reponse.go @@ -0,0 +1,32 @@ +package helper + +import ( + "github.com/gin-gonic/gin" + "net/http" +) + +// ResponseData : response format +type ResponseData struct { + Success bool `json:"success"` + Message string `json:"message"` + Data interface{} `json:"data"` +} + +// RespondJSON : send a proper response json to the client +func RespondJSON(w *gin.Context, status int, message string, payload interface{}) { + var res ResponseData + + if status >= 200 && status < 300 { + res.Success = true + } + + if len(message) != 0 { + res.Message = message + } + + if payload != nil { + res.Data = payload + } + + w.JSON(http.StatusOK, res) +} diff --git a/main.go b/main.go new file mode 100644 index 0000000..3e28e85 --- /dev/null +++ b/main.go @@ -0,0 +1,38 @@ +package main + +import ( + "fmt" + "git.rouggy.com/rouggy/RaceBot/config" + "git.rouggy.com/rouggy/RaceBot/database" + "git.rouggy.com/rouggy/RaceBot/models" + "git.rouggy.com/rouggy/RaceBot/router" + "github.com/gin-gonic/gin" +) + +func main() { + + config.Load() + + database.SQLiteDBConnect() + defer database.GetDB().Close() + SQLiteMigrate() + ServeApplication() + +} + +func ServeApplication() { + gin.SetMode(gin.ReleaseMode) + r := gin.Default() + router.SetupRoutes(r) + fmt.Println("Server is running on port:", config.HTTPPort) + r.Run(":" + config.HTTPPort) + +} + +func SQLiteMigrate() { + db := database.GetDB() + if err := db.AutoMigrate(&models.Release{}, &models.Race{}).Error; err != nil { + panic("[Database] Failed migrating database: ") + } + fmt.Println("[Database] Database successfully migrated") +} diff --git a/models/race.go b/models/race.go new file mode 100644 index 0000000..27f325c --- /dev/null +++ b/models/race.go @@ -0,0 +1,40 @@ +package models + +import ( + "git.rouggy.com/rouggy/RaceBot/database" + "github.com/jinzhu/gorm" +) + +type Race struct { + gorm.Model + Hash string `gorm:"column:hash;type:varchar(50)" form:"hash" json:"hash" binding:"max=50"` + TorrentName string `gorm:"column:torrent_name;varchar(100);not null" form:"torrent_name" json:"torrent_name" binding:"required,max=100"` + Category string `gorm:"column:category;varchar(50);not null" form:"category" json:"category" binding:"max=50"` + ContentPath string `gorm:"column:content_path;varchar(100);not null" form:"content_path" json:"content_path" binding:"max=100"` + RootPath string `gorm:"column:root_path;varchar(100)" form:"root_path" json:"root_path" binding:"max=100"` + Size string `gorm:"column:size;varchar(15)" form:"size" json:"size" binding:"max=15"` + Won bool `gorm:"column:won;bool" form:"won" json:"won"` +} + +func NewRace() *Race { + r := &Race{} + return r +} + +func (r *Race) SaveRace() (*Race, error) { + db := database.GetDB() + err := db.Create(&r).Error + if err != nil { + return &Race{}, err + } + return r, nil +} + +func (r *Race) GetRaceByName(title string) (*Race, error) { + db := database.GetDB() + err := db.Where(Race{TorrentName: title}).Find(&r).Error + if err != nil { + return &Race{}, err + } + return r, nil +} diff --git a/models/release.go b/models/release.go new file mode 100644 index 0000000..635159e --- /dev/null +++ b/models/release.go @@ -0,0 +1,83 @@ +package models + +import ( + "git.rouggy.com/rouggy/RaceBot/database" + "github.com/jinzhu/gorm" + "github.com/moistari/rls" +) + +type Release struct { + gorm.Model + Type string `gorm:"column:type;varchar(10);not null" form:"type" json:"type" binding:"required,max=10"` + TorrentName string `gorm:"column:torrent_name;varchar(100);not null" form:"torrent_name" json:"torrent_name" binding:"required,max=100"` + Title string `gorm:"column:title;varchar(100);not null" form:"title" json:"title" binding:"required,max=100"` + Season int `gorm:"column:season;integer" form:"season" json:"season" binding:"required,max=100"` + Episode int `gorm:"column:episode;integer" form:"episode" json:"episode" binding:"required,max=100"` + Subtitle string `gorm:"column:subtitle;varchar(10)" form:"subtitle" json:"subtitle" binding:"max=10"` + Source string `gorm:"column:source;varchar(15)" form:"source" json:"source" binding:"max=15"` + Resolution string `gorm:"column:resolution;varchar(10)" form:"resolution" json:"resolution" binding:"max=10"` + Codec string `gorm:"many2many:release_codecs;column:codec;varchar(10)" form:"codec" json:"codec" binding:"max=10"` + HDR string `gorm:"many2many:releases_hdr;column:hdr;varchar(10)" form:"hdr" json:"hdr" binding:"max=10"` + Audio string `gorm:"many2many:releases_audio;column:audio;varchar(10)" form:"audio" json:"audio" binding:"max=10"` + Channels string `gorm:"column:channels;varchar(10)" form:"channels" json:"channels" binding:"max=10"` + Group string `gorm:"column:group;varchar(20)" form:"group" json:"group" binding:"max=20"` + Other string `gorm:"many2many:releases_other;column:other;varchar(20)" form:"other" json:"other" binding:"max=20"` + Language string `gorm:"many2many:releases_languages;column:language;varchar(20)" form:"language" json:"language" binding:"max=20"` + ReleaseType string `gorm:"column:release_type;varchar(20)" form:"release_type" json:"release_type" binding:"max=20"` + Year int `gorm:"column:year;integer" form:"year" json:"year" binding:"max=4"` +} + +func NewRelease() *Release { + r := &Release{} + return r +} + +func (r *Release) ParseString(title string) { + rel := rls.ParseString(title) + + r.TorrentName = title + r.Source = rel.Source + r.Resolution = rel.Resolution + if rel.Audio != nil { + r.Audio = rel.Audio[0] + } + r.Audio = rel.Channels + r.Codec = rel.Codec[0] + if rel.HDR != nil { + r.HDR = rel.HDR[0] + } + if rel.Other != nil { + r.Other = rel.Other[0] + } + if rel.Language != nil { + r.Language = rel.Language[0] + } + + if r.Title == "" { + r.Title = rel.Title + } + + if r.Season == 0 { + r.Season = rel.Series + } + if r.Episode == 0 { + r.Episode = rel.Episode + } + + if r.Year == 0 { + r.Year = rel.Year + } + + if r.Group == "" { + r.Group = rel.Group + } +} + +func (r *Release) SaveRelease() (*Release, error) { + db := database.GetDB() + err := db.Create(&r).Error + if err != nil { + return &Release{}, err + } + return r, nil +} diff --git a/racer.db b/racer.db new file mode 100644 index 0000000..cf18a8b Binary files /dev/null and b/racer.db differ diff --git a/router/router.go b/router/router.go new file mode 100644 index 0000000..cfe286e --- /dev/null +++ b/router/router.go @@ -0,0 +1,2 @@ +package router +