// debuglevel: debug, info, notice, warn, error, crit, alert, emerg
// ssl: on, off
type Site struct {
    Id          int64
    SiteName    string `xorm:"unique"`
    Port        int
    BackendAddr []string
    Ssl         string    `xorm:"varchar(10) notnull default 'off'"`
    DebugLevel  string    `xorm:"varchar(10) notnull default 'error'"`
    LastChange  time.Time `xorm:"updated"`
    Version     int       `xorm:"version"` // 乐观锁
}

func ListSite() (sites []Site, err error) {
    sites = make([]Site, 0)
    err = Engine.Find(&sites)    
    log.Println(err, sites)    
    return sites, err
}

func NewSite(siteName string, Port int, BackendAddr []string, SSL string, DebugLevel string) (err error) {    
    if SSL == "" {
        SSL = "off"
    }    
    if DebugLevel == "" {
        DebugLevel = "error"
    }

    _, err = Engine.Insert(&Site{SiteName: siteName, Port: Port, BackendAddr: BackendAddr, Ssl: SSL, DebugLevel: DebugLevel})   
 return err
}

func NewSite(ctx *macaron.Context, sess session.Store, x csrf.CSRF) {    if sess.Get("uid") != "" {
        ctx.Data["csrf_token"] = x.GetToken()
        ctx.HTML(200, "newSite")
    } else {
        ctx.Redirect("/login/")
    }
}

func DoNewSite(ctx *macaron.Context, sess session.Store) {    
    if sess.Get("uid") != nil {        
        log.Println(sess.Get("uid"))
        siteName := ctx.Req.Form.Get("sitename")
        port := ctx.Req.Form.Get("port")
        Port, _ := strconv.Atoi(port)
        backaddr := ctx.Req.Form.Get("backendaddr")
        backendaddr := strings.Split(backaddr, "\r\n")
        BackendAddr := make([]string, 0)        
            for _, v := range backendaddr {            
                if v == "" {                
                continue
            }
            v = strings.TrimSpace(v)
            BackendAddr = append(BackendAddr, v)
        }

        ssl := ctx.Req.Form.Get("ssl")
        debugLevel := ctx.Req.Form.Get("debuglevel")        
        log.Println(siteName, BackendAddr, ssl, debugLevel)
        models.NewSite(siteName, Port, BackendAddr, ssl, debugLevel)
        ctx.Redirect("/admin/site/list/")
    } else {
        ctx.Redirect("/login/")
    }
}

var (
    Engine *xorm.Engine
    err    error
)

func init() {    
    // 从conf/app.ini获取数据库的配置信息
    sec := setting.Cfg.Section("database")    
    // 连接数据库
    Engine, err = xorm.NewEngine("mysql", fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8",
        sec.Key("USER").String(),
        sec.Key("PASSWD").String(),
        sec.Key("HOST").String(),
        sec.Key("NAME").String()))    
    
    if err != nil {        
        log.Panicf("Faild to connect to database, err:%v", err)
    }    
    
    // 新建site、user和rules表
    Engine.Sync2(new(Site))
    Engine.Sync2(new(User))
    Engine.Sync2(new(Rules))    
    // 如果user表为空，则新建一个默认账户， 
    ret, err := Engine.IsTableEmpty(new(User))   
    if err == nil && ret {        
        log.Printf("create new user:%v, password:%v\n", "admin", "x@xsec.io")
        NewUser("admin", "x@xsec.io")
    }    
     
    // 如果规则为空，则插入默认的初始化规则
    ret, err = Engine.IsTableEmpty(new(Rules))    
        if err == nil && ret {        
        log.Println("Insert default waf rules")
        Engine.Exec(DefaultRules)
    }
}

m.Group("/admin", func() {       
     m.Get("/index/", routers.Admin)        
     
     m.Group("/site/", func() {            
     m.Get("", routers.Admin)            
     m.Get("/list/", routers.Admin)            
     m.Get("/new/", routers.NewSite)            
     m.Post("/new/", csrf.Validate, routers.DoNewSite)           
     m.Get("/edit/:id", routers.EditSite)            
     m.Post("/edit/:id", csrf.Validate, routers.DoEditSite)            
     m.Get("/del/:id", routers.DelSite)            
     m.Get("/sync/", routers.SyncSite)            
     m.Get("/sync/:id", routers.SyncSiteById)           
     m.Get("/json/", routers.SiteJSON)
     })
    })    
      m.Group("/api", func() {        
      m.Get("/site/sync/", routers.SyncSiteApi)        
      m.Get("/rule/sync/", routers.SyncRuleApi)
    })    
      
      log.Printf("xsec waf admin %s", setting.AppVer)    
      log.Printf("Run mode %s", strings.Title(macaron.Env))    
      log.Printf("Server is running on %s", fmt.Sprintf("0.0.0.0:%v", setting.HTTPPort))    
      log.Println(http.ListenAndServe(fmt.Sprintf("0.0.0.0:%v", setting.HTTPPort), m))