init
This commit is contained in:
99
local/utl/common/common.go
Normal file
99
local/utl/common/common.go
Normal 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
|
||||
}
|
||||
8
local/utl/configs/configs.go
Normal file
8
local/utl/configs/configs.go
Normal 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
22
local/utl/cron/cron.go
Normal 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
39
local/utl/db/db.go
Normal 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")
|
||||
}
|
||||
}
|
||||
26
local/utl/discord/discord.go
Normal file
26
local/utl/discord/discord.go
Normal 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!")
|
||||
}
|
||||
|
||||
}
|
||||
43
local/utl/server/server.go
Normal file
43
local/utl/server/server.go
Normal 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
|
||||
}
|
||||
Reference in New Issue
Block a user