main.go

package main

import (
	"bufio"
	"fmt"
	"net"
	"os"

	"github.com/HTTPs-omma/HTTPsBAS-HSProtocol/HSProtocol"
	cors2 "github.com/gin-contrib/cors"
	"github.com/gin-gonic/gin"
	"github.com/gofiber/fiber/v3"
	"github.com/gofiber/fiber/v3/middleware/cors"
	"github.com/joho/godotenv"
	swaggerFiles "github.com/swaggo/files"
	ginSwagger "github.com/swaggo/gin-swagger"
	"github.com/your/repo/Core"
	_ "github.com/your/repo/docs"
	"github.com/your/repo/router"
)

// @title			Swagger Example API
// @version		1.0
// @description	This is a sample server Petstore server.
// @termsOfService	http://swagger.io/terms/
// @contact.name	API Support
// @contact.url	http://www.swagger.io/support
// @contact.email	support@swagger.io
// @license.name	Apache 2.0
// @license.url	http://www.apache.org/licenses/LICENSE-2.0.html
// @host			httpsbas.com:8002
// @BasePath		/
// @Path			api
var testCommand string = "dir /"

func main() {
	err := godotenv.Load()
	if err != nil {
		panic("Error loading .env file")
	}

	// tcp
	go TCPServer()

	// Swagger
	go Swagger()

	// HTTP
	HTTPServer()

}

// https://zzihyeon.tistory.com/76
func Swagger() {
	r := gin.Default()

	// Swagger 엔드포인트
	r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
	r.Use(cors2.New(cors2.Config{
		AllowOrigins:     []string{"*"},                            // 모든 도메인 허용, 보안 상 필요한 경우 특정 도메인만 허용해야 함
		AllowMethods:     []string{"GET", "POST", "PUT", "DELETE"}, // 허용할 HTTP 메서드
		AllowHeaders:     []string{"Origin", "Content-Type", "Accept", "Authorization", "Access-Control-Allow-Origin", "Connection", "Accept-Encoding"},
		ExposeHeaders:    []string{"Content-Length"},
		AllowCredentials: true,
	}))
	r.Run("0.0.0.0:8001")

}

func TCPServer() {
	listener, err := net.Listen("tcp", "0.0.0.0:8080")
	if err != nil {
		fmt.Println("Error starting TCP server:", err)
		return
	}
	defer listener.Close()

	fmt.Println("TCP server listening on port 8080")

	for {
		conn, err := listener.Accept()
		fmt.Println("======= tcp received ===========")
		if err != nil {
			fmt.Println("Error accepting connection:", err)
			continue
		}

		go handleTCPConnection(conn)
	}

}

func handleTCPConnection(conn net.Conn) {
	reader := bufio.NewReader(conn)
	reader.Discard(reader.Buffered()) // 남은 버퍼를 버림
	defer conn.Close()                // 함수 호출 종료 후 Close
	index := 0
	for {
		buffer := make([]byte, 1024*1024)
		n, err := conn.Read(buffer)
		if err != nil {
			fmt.Println("Error reading from connection:", err)
			break
		}
		if index > 100 {
			break
		}
		if n < 2 {
			index += 1
			continue
		}
		index = 0

		defer func() {
			if r := recover(); r != nil {
				fmt.Println("panic 회복:", r)
			}
		}()

		HSMgr := HSProtocol.NewHSProtocolManager()
		hs, err := HSMgr.Parsing(buffer)
		if err != nil {
			fmt.Println("Error parsing:", err)
			ack := &HSProtocol.HS{ // HSProtocol.ACK
				ProtocolID:     hs.ProtocolID,
				Command:        HSProtocol.ERROR_ACK,
				UUID:           hs.UUID,
				HealthStatus:   hs.HealthStatus,
				Identification: hs.Identification,
				TotalLength:    hs.TotalLength,
				Data:           []byte{},
			}
			rstb, _ := HSMgr.ToBytes(ack)
			conn.Write(rstb)
			continue
		}

		// fmt.Println("hs.command : ", hs.Command)
		// fmt.Println("hs.TotalLeng : ", hs.TotalLength)
		// da, _ := HSProtocol.NewHSProtocolManager().ToBytes(hs)
		// fmt.Println("hs len : ", len(da))

		dipt := Core.CommandDispatcher{}
		ack, err := dipt.Action(hs)
		if err != nil {
			ack := &HSProtocol.HS{ // HSProtocol.ACK
				ProtocolID:     hs.ProtocolID,
				Command:        HSProtocol.ERROR_ACK,
				UUID:           hs.UUID,
				HealthStatus:   hs.HealthStatus,
				Identification: hs.Identification,
				TotalLength:    hs.TotalLength,
				Data:           []byte{},
			}
			rstb, _ := HSMgr.ToBytes(ack)
			fmt.Println(err)
			conn.Write(rstb)
			continue
		}
		rstb, err := HSMgr.ToBytes(ack)
		conn.Write(rstb)

		// reader := bufio.NewReader(conn)
		// reader.Discard(reader.Buffered()) // 남은 버퍼를 버림

		// fmt.Println("완료")
		continue
	}
}

func HTTPServer() {
	app := fiber.New()
	app.Get("/view/db", func(c fiber.Ctx) error {
		// HTML 파일을 읽어서 응답으로 반환
		htmlData, err := os.ReadFile("./view/html/viewdata.html")
		if err != nil {
			return c.Status(500).SendString("Error loading page")
		}
		c.Set("Content-Type", "text/html")
		return c.Send(htmlData)
	})

	// 효과적인 Cors 에러 해결
	app.Use(cors.New(cors.Config{
		AllowCredentials: true,
		AllowOriginsFunc: func(origin string) bool { return true },
	}))
	app.Use(cors.New(cors.Config{
		AllowOrigins: []string{"*", "*"},
		AllowHeaders: []string{"Origin", "Content-Type", "Accept"},
	}))
	app.Get("/downloads/:filename", func(c fiber.Ctx) error {
		// 파일 이름을 URL 파라미터로 받음
		filename := c.Params("filename")
		filePath := "./HTTPsBAS-Dropbin/" + filename
		// 파일이 존재하는지 확인
		if _, err := os.Stat(filePath); os.IsNotExist(err) {
			return c.Status(fiber.StatusNotFound).SendString("File not found")
		}

		// 파일을 클라이언트에게 전송 (다운로드)
		return c.Download(filePath)
	})

	router.SetupAPIRoutes(app)
	router.SetupViewRoutes(app)

	// fmt.Println("HTTP server listening on port 80")
	err := app.Listen(":8002")
	if err != nil {
		fmt.Println("Error starting HTTP server:", err)
	}

}