mirror of
https://github.com/tgdrive/teldrive.git
synced 2024-11-10 17:14:03 +08:00
69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"mime"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/divyam234/teldrive/database"
|
|
"github.com/divyam234/teldrive/routes"
|
|
"github.com/divyam234/teldrive/ui"
|
|
"github.com/divyam234/teldrive/utils"
|
|
|
|
"github.com/divyam234/cors"
|
|
"github.com/divyam234/teldrive/utils/cache"
|
|
"github.com/divyam234/teldrive/utils/cron"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/go-co-op/gocron"
|
|
)
|
|
|
|
func main() {
|
|
|
|
gin.SetMode(gin.ReleaseMode)
|
|
|
|
router := gin.Default()
|
|
|
|
utils.InitConfig()
|
|
|
|
utils.InitializeLogger()
|
|
|
|
database.InitDB()
|
|
|
|
cache.InitCache()
|
|
|
|
scheduler := gocron.NewScheduler(time.UTC)
|
|
|
|
scheduler.Every(1).Hour().Do(cron.FilesDeleteJob)
|
|
|
|
scheduler.Every(12).Hour().Do(cron.UploadCleanJob)
|
|
|
|
scheduler.StartAsync()
|
|
|
|
router.Use(cors.New(cors.Config{
|
|
AllowMethods: []string{"GET", "POST", "PUT", "PATCH", "DELETE", "HEAD", "OPTIONS"},
|
|
AllowHeaders: []string{"Authorization", "Content-Length", "Content-Type"},
|
|
AllowCredentials: true,
|
|
AllowOriginFunc: func(origin string) bool {
|
|
return true
|
|
},
|
|
MaxAge: 12 * time.Hour,
|
|
}))
|
|
|
|
mime.AddExtensionType(".js", "application/javascript")
|
|
|
|
router.Use(gin.ErrorLogger())
|
|
|
|
routes.AddRoutes(router)
|
|
|
|
ui.AddRoutes(router)
|
|
|
|
config := utils.GetConfig()
|
|
certDir := filepath.Join(config.ExecDir, "sslcerts")
|
|
ok, _ := utils.PathExists(certDir)
|
|
if ok && config.Https {
|
|
router.RunTLS(fmt.Sprintf(":%d", config.Port), filepath.Join(certDir, "cert.pem"), filepath.Join(certDir, "key.pem"))
|
|
} else {
|
|
router.Run(fmt.Sprintf(":%d", config.Port))
|
|
}
|
|
}
|