main.go

// Copyright 2023 The casbin Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package main

import (
	"fmt"

	"github.com/beego/beego"
	"github.com/beego/beego/plugins/cors"
	_ "github.com/beego/beego/session/redis"
	"github.com/casbin/caswaf/casdoor"
	"github.com/casbin/caswaf/ip"
	"github.com/casbin/caswaf/object"
	"github.com/casbin/caswaf/proxy"
	"github.com/casbin/caswaf/routers"
	"github.com/casbin/caswaf/run"
	"github.com/casbin/caswaf/service"
	"github.com/casbin/caswaf/util"
)

func main() {
	util.InitSelfGuard()
	object.InitFlag()
	object.InitAdapter()
	object.CreateTables()
	casdoor.InitCasdoorConfig()
	proxy.InitHttpClient()
	ip.InitIpDb()
	object.InitSiteMap()
	object.InitRuleMap()
	run.InitAppMap()
	run.InitRdsClient()
	run.InitSelfStart()
	object.StartMonitorSitesLoop()

	beego.InsertFilter("*", beego.BeforeRouter, cors.Allow(&cors.Options{
		AllowOrigins:     []string{"*"},
		AllowMethods:     []string{"GET", "POST", "DELETE", "PUT", "PATCH", "OPTIONS"},
		AllowHeaders:     []string{"Origin", "X-Requested-With", "Content-Type", "Accept"},
		ExposeHeaders:    []string{"Content-Length"},
		AllowCredentials: true,
	}))

	//beego.DelStaticPath("/static")
	beego.SetStaticPath("/static", "web/build/static")
	// https://studygolang.com/articles/2303
	beego.InsertFilter("/", beego.BeforeRouter, routers.TransparentStatic) // must has this for default page
	beego.InsertFilter("/*", beego.BeforeRouter, routers.TransparentStatic)
	beego.InsertFilter("/api/*", beego.BeforeRouter, routers.ApiFilter)

	if beego.AppConfig.String("redisEndpoint") == "" {
		beego.BConfig.WebConfig.Session.SessionProvider = "file"
		beego.BConfig.WebConfig.Session.SessionProviderConfig = "./tmp"
	} else {
		beego.BConfig.WebConfig.Session.SessionProvider = "redis"
		beego.BConfig.WebConfig.Session.SessionProviderConfig = beego.AppConfig.String("redisEndpoint")
	}
	beego.BConfig.WebConfig.Session.SessionGCMaxLifetime = 3600 * 24 * 365

	port := beego.AppConfig.DefaultInt("httpport", 17000)

	// Stop old instances on all ports before starting new services
	// Check gateway ports first since they bind first in service.Start()
	gatewayEnabled, err := beego.AppConfig.Bool("gatewayEnabled")
	if err != nil {
		panic(err)
	}

	if gatewayEnabled {
		gatewayHttpPort, err := beego.AppConfig.Int("gatewayHttpPort")
		if err != nil {
			panic(err)
		}
		err = util.StopOldInstance(gatewayHttpPort)
		if err != nil {
			panic(err)
		}

		gatewayHttpsPort, err := beego.AppConfig.Int("gatewayHttpsPort")
		if err != nil {
			panic(err)
		}
		err = util.StopOldInstance(gatewayHttpsPort)
		if err != nil {
			panic(err)
		}
	}

	err = util.StopOldInstance(port)
	if err != nil {
		panic(err)
	}

	service.Start()

	beego.Run(fmt.Sprintf(":%v", port))
}