package main

import (
	_ "git.oschina.net/liukexing/RbacManager/models/center"
	"git.oschina.net/liukexing/RbacManager/models/flatui"
	_ "git.oschina.net/liukexing/RbacManager/routers"

	"fmt"
	"github.com/astaxie/beego"
	"github.com/astaxie/beego/orm"
	_ "github.com/go-sql-driver/mysql"
	_ "github.com/lib/pq"
	//_ "github.com/mattn/go-sqlite3"
)

func main() {
	FuncMapInit()
	DataBaseInit()
	FilterInit()
	beego.Run()
}

func FuncMapInit() {
	//beego.AddFuncMap("htmlbuild", flatui.Htmlbuild)
	//beego.AddFuncMap("themebuild", flatui.Themebuild)
}

func DataBaseInit() {
	var dns string
	db_type := beego.AppConfig.String("db_type")
	db_host := beego.AppConfig.String(db_type + "::db_host")
	db_port := beego.AppConfig.String(db_type + "::db_port")
	db_user := beego.AppConfig.String(db_type + "::db_user")
	db_pass := beego.AppConfig.String(db_type + "::db_pass")
	db_name := beego.AppConfig.String(db_type + "::db_name")
	db_path := beego.AppConfig.String(db_type + "::db_path")
	db_sslmode := beego.AppConfig.String(db_type + "::sslmode")
	switch db_type {
	case "mysql":
		orm.RegisterDriver("mysql", orm.DRMySQL)
		dns = fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8", db_user, db_pass, db_host, db_port, db_name)
		break
	case "postgres":
		orm.RegisterDriver("postgres", orm.DRPostgres)
		dns = fmt.Sprintf("dbname=%s host=%s  user=%s  password=%s  port=%s  sslmode=%s", db_name, db_host, db_user, db_pass, db_port, db_sslmode)
	case "sqlite3":
		orm.RegisterDriver("sqlite3", orm.DRSqlite)
		if db_path == "" {
			db_path = "./"
		}
		dns = fmt.Sprintf("%s%s.db", db_path, db_name)
		break
	default:
		beego.Critical("Database driver is not allowed:", db_type)
	}
	orm.RegisterDataBase("default", db_type, dns)

	//orm.RegisterDataBase("default", "mysql", "root:@/herbalifecenter?charset=utf8", 30)
	flatui.Syncdb()
}

func FilterInit() {
	beego.InsertFilter("/*", beego.BeforeExec, flatui.GetUserAgent)
}