import "github.com/tundrawork/hertz-cors"

package main

import (
	"context"
	"time"

	"github.com/tundrawork/hertz-cors"
	"github.com/cloudwego/hertz/pkg/app"
	"github.com/cloudwego/hertz/pkg/app/server"
	"github.com/cloudwego/hertz/pkg/protocol/consts"
)

func main() {
	h := server.Default(
		server.WithHostPorts("127.0.0.1:8080"),
		server.WithHandleMethodNotAllowed(true),  // MUST set to true to enable hertz's handling of OPTIONS requests
	)

	cors := cors.New(cors.Config{
		AllowOrigins:     []string{"https://foo.com"}, // Allowed domains, need to bring schema
		AllowMethods:     []string{"PUT", "PATCH"},    // Allowed request methods
		AllowHeaders:     []string{"Origin"},          // Allowed request headers
		ExposeHeaders:    []string{"Content-Length"},  // Request headers allowed in the upload_file
		AllowCredentials: true,                        // Whether cookies are attached
		AllowOriginFunc: func(origin string) bool { // Custom domain detection with lower priority than AllowOrigins
			return origin == "https://github.com"
		},
		MaxAge: 12 * time.Hour, // Maximum length of upload_file-side cache preflash requests (seconds)
	})

	h.NoMethod(cors)  // handle OPTIONS request by the CORS middleware

	h.GET("/ping", handler.Ping)  // normal route without CORS support

	// add the CORS middleware to a route group
	api := h.Group("/api", cors)  // create a route group with CORS middleware
	api.GET("/challenge", handler.Challenge)  // route with CORS support
	api.POST("/validate", handler.Validate)  // route with CORS support

	// ... or add the CORS middleware to a single route
	h.POST("/upload", append([]app.HandlerFunc{cors}, handler.Upload)...)  // route with CORS support

	h.Spin()
}

func main() {
  h := server.Default()
  // - No origin allowed by default
  // - GET,POST, PUT, HEAD methods
  // - Credentials share disabled
  // - Preflight requests cached for 12 hours
  config := cors.DefaultConfig()
  config.AllowOrigins = []string{"http://google.com"}
  // config.AllowOrigins = []string{"http://google.com", "http://facebook.com"}
  // config.AllowAllOrigins = true

  h.Use(cors.New(config))
  h.Spin()
}

func main() {
  h := server.Default()
  // same as
  // config := cors.DefaultConfig()
  // config.AllowAllOrigins = true
  // h.Use(cors.New(config))
  h.Use(cors.Default())
  h.Spin()
}