main.go

package main

import (
	"devread/crawler"
	"devread/db"
	_ "devread/docs"
	"devread/handle_log"
	"devread/handler"
	"devread/helper"
	"devread/repository/repo_impl"
	"devread/router"

	"os"
	"time"

	"github.com/joho/godotenv"
	"github.com/labstack/echo/v4"
	echoSwagger "github.com/swaggo/echo-swagger"
)

func init() {
	if err := godotenv.Load(".env-pro"); err != nil {
		return
	}
}

// @title DevRead API
// @version 1.0
// @description Ứng dụng tổng hợp kiến thức cho developer
// @description Tổng hợp bài viết hay nhất trên các blog IT như viblo, toidicodedao, yellowcodebooks, thefullsnack, quan-cam, codeaholicguy,...

// @securityDefinitions.apikey jwt
// @in header
// @name Authorization

// @host devread.herokuapp.com
// @BasePath /

func main() {
	// write log
	log, _ := handle_log.WriteLog()

	// // Dùng dưới local
	// redisHost := os.Getenv("REDIS_HOST")
	// redisPort := os.Getenv("REDIS_PORT")

	// Dùng trên server heroku
	redisUrl := os.Getenv("REDIS_URL")

	// postgres details
	host := os.Getenv("DB_HOST")
	port := os.Getenv("DB_PORT")
	password := os.Getenv("DB_PASSWORD")
	username := os.Getenv("DB_USERNAME")
	dbname := os.Getenv("DB_NAME")

	// connect redis
	client := &db.RedisDB{
		// Dùng dưới local
		// Host:     redisHost,
		// Port:     redisPort,

		// Dùng trên server heroku
		Url: redisUrl,

		Logger: log,
	}
	client.NewRedisDB()

	// connect postgres
	sql := &db.Sql{
		Host:     host,
		Port:     port,
		UserName: username,
		Password: password,
		DbName:   dbname,
		Logger:   log,
	}
	sql.Connect()
	defer sql.Close()

	e := echo.New()
	e.GET("/swagger/*", echoSwagger.WrapHandler)

	customValidator := helper.NewCustomValidator()
	customValidator.RegisterValidate()

	e.Validator = customValidator

	userHandler := handler.UserHandler{
		UserRepo: repo_impl.NewUserRepo(sql),
		AuthRepo: repo_impl.NewAuthenRepo(client),
		Logger:   log,
	}

	postHandler := handler.PostHandler{
		PostRepo:     repo_impl.NewPostRepo(sql),
		AuthRepo:     repo_impl.NewAuthenRepo(client),
		BookmarkRepo: repo_impl.NewBookmarkRepo(sql),
		Logger:       log,
	}

	api := router.API{
		Echo:        e,
		UserHandler: userHandler,
		PostHandler: postHandler,
	}
	api.SetupRouter()

	// time start crawler
	go crawler.VibloPost(postHandler.PostRepo)
	go crawler.ToidicodedaoPost(postHandler.PostRepo)
	go crawler.ThefullsnackPost(postHandler.PostRepo)
	go crawler.QuancamPostV1(postHandler.PostRepo)
	go crawler.CodeaholicguyPost(postHandler.PostRepo)
	go crawler.YellowcodePost(postHandler.PostRepo)

	// schedule crawler
	go schedule(60*time.Minute, postHandler, 1)
	go schedule(24*time.Hour, postHandler, 2)
	go schedule(96*time.Hour, postHandler, 3)
	go schedule(96*time.Hour, postHandler, 4)
	go schedule(96*time.Hour, postHandler, 5)
	go schedule(96*time.Hour, postHandler, 6)

	e.Logger.Fatal(e.Start(":" + os.Getenv("PORT")))
}

func schedule(timeSchedule time.Duration, handler handler.PostHandler, crowIlnndex int) {
	ticker := time.NewTicker(timeSchedule)
	func() {
		for {
			switch crowIlnndex {
			case 1:
				<-ticker.C
				crawler.VibloPost(handler.PostRepo)
			case 2:
				<-ticker.C
				crawler.ToidicodedaoPost(handler.PostRepo)
			case 3:
				<-ticker.C
				crawler.ThefullsnackPost(handler.PostRepo)
			case 4:
				<-ticker.C
				crawler.QuancamPostV1(handler.PostRepo)
			case 5:
				<-ticker.C
				crawler.CodeaholicguyPost(handler.PostRepo)
			case 6:
				<-ticker.C
				crawler.YellowcodePost(handler.PostRepo)
			}
		}
	}()
}