This commit is contained in:
Fran Jurmanović
2024-09-30 22:00:15 +02:00
parent 518ce27d7c
commit 14a689ad76
19 changed files with 968 additions and 0 deletions

View File

@@ -0,0 +1,99 @@
package common
import (
"fmt"
"log"
"net"
"os"
"os/exec"
"regexp"
"strings"
"github.com/gofiber/fiber/v2"
)
type RouteGroups struct {
Concert fiber.Router
}
func CheckError(err error) {
if err != nil {
log.Printf("Error occured. %v", err)
}
}
var matchFirstCap = regexp.MustCompile("(.)([A-Z][a-z]+)")
var matchAllCap = regexp.MustCompile("([a-z0-9])([A-Z])")
func GetMonthFromLocalized(month string) (int, error) {
switch month {
case "január":
return 1, nil
case "február":
return 2, nil
case "március":
return 3, nil
case "április":
return 4, nil
case "május":
return 5, nil
case "június":
return 6, nil
case "július":
return 7, nil
case "augusztus":
return 8, nil
case "szeptember":
return 9, nil
case "október":
return 10, nil
case "november":
return 11, nil
case "december":
return 12, nil
default:
return 0, fmt.Errorf("Month not recognized %1", month)
}
}
func ToSnakeCase(str string) string {
snake := matchFirstCap.ReplaceAllString(str, "${1}_${2}")
snake = matchAllCap.ReplaceAllString(snake, "${1}_${2}")
return strings.ToLower(snake)
}
func GetIP() string {
addrs, err := net.InterfaceAddrs()
if err != nil {
os.Stderr.WriteString("Oops: " + err.Error() + "\n")
os.Exit(1)
}
for _, a := range addrs {
if ipnet, ok := a.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
if ipnet.IP.To4() != nil {
return ipnet.IP.String()
}
}
}
return ""
}
func Find[T any](lst *[]T, callback func(item *T) bool) *T {
for _, item := range *lst {
if callback(&item) {
return &item
}
}
return nil
}
func RunElevatedCommand(command string, service string) (string, error) {
cmd := exec.Command("powershell", "-nologo", "-noprofile", "-File", "run_sc.ps1", command, service)
output, err := cmd.CombinedOutput()
if err != nil {
log.Panic("error: %v, output: %s", err, string(output))
return "", fmt.Errorf("error: %v, output: %s", err, string(output))
}
return string(output), nil
}

View File

@@ -0,0 +1,8 @@
package configs
const (
Version = "0.0.1"
Prefix = "v1"
Secret = "Donde4sta"
SecretCode = "brasno"
)

22
local/utl/cron/cron.go Normal file
View File

@@ -0,0 +1,22 @@
package cronhu
import (
"log"
"github.com/robfig/cron/v3"
"go.uber.org/dig"
)
func Init(di *dig.Container) *cron.Cron {
c := cron.New()
err := di.Provide(func() *cron.Cron {
return c
})
if err != nil {
log.Panic("unable to initialize cron!")
}
return c
}

39
local/utl/db/db.go Normal file
View File

@@ -0,0 +1,39 @@
package db
import (
"os"
"rockhu-bot/local/model"
"time"
"go.uber.org/dig"
"gorm.io/driver/postgres"
"gorm.io/gorm"
)
func Start(di *dig.Container) {
db, err := gorm.Open(postgres.New(postgres.Config{
DSN: os.Getenv("CONNECTION_STRINGS"),
PreferSimpleProtocol: true,
}), &gorm.Config{
NowFunc: func() time.Time {
utc, _ := time.LoadLocation("")
return time.Now().In(utc)
}})
if err != nil {
panic("failed to connect database")
}
err = di.Provide(func() *gorm.DB {
return db
})
if err != nil {
panic("failed to bind database")
}
Migrate(db)
}
func Migrate(db *gorm.DB) {
err := db.AutoMigrate(&model.ConcertModel{})
if err != nil {
panic("failed to migrate model.ApiModel")
}
}

View File

@@ -0,0 +1,26 @@
package discordrhu
import (
"log"
"os"
"github.com/bwmarrin/discordgo"
"go.uber.org/dig"
)
func Init(di *dig.Container) {
dsc, err := discordgo.New("Bot " + os.Getenv("DISCORD_KEY"))
if err != nil {
log.Panic("unable to start discord session!")
}
err = di.Provide(func() *discordgo.Session {
return dsc
})
if err != nil {
log.Panic("unable to provide discord session!")
}
}

View File

@@ -0,0 +1,43 @@
package server
import (
"rockhu-bot/local/api"
"rockhu-bot/local/utl/common"
"fmt"
"log"
"os"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/fiber/v2/middleware/cors"
"github.com/gofiber/swagger"
"go.uber.org/dig"
)
func Start(di *dig.Container) *fiber.App {
app := fiber.New(fiber.Config{
EnablePrintRoutes: true,
})
app.Use(cors.New())
app.Get("/swagger/*", swagger.HandlerDefault)
file, err := os.OpenFile("logs.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0666)
if err != nil {
log.Print("Cannot open file logs.log")
}
log.SetOutput(file)
api.Init(di, app)
app.Get("/ping", func(c *fiber.Ctx) error {
return c.SendString("pong")
})
port := os.Getenv("PORT")
err = app.Listen(":" + port)
if err != nil {
msg := fmt.Sprintf("Running on %s:%s", common.GetIP(), port)
println(msg)
}
return app
}