go-admin/main.go

package main

import (
	"io/ioutil"
	"net/http"

	_ "github.com/GoAdminGroup/go-admin/adapter/gin"               // adapter
	_ "github.com/GoAdminGroup/go-admin/modules/db/drivers/sqlite" // sql driver
	_ "github.com/GoAdminGroup/themes/adminlte"                    // theme

	"github.com/GoAdminGroup/example/tables"
	"github.com/GoAdminGroup/go-admin/engine"
	"github.com/GoAdminGroup/go-admin/plugins/admin"
	"github.com/GoAdminGroup/go-admin/plugins/example"
	"github.com/GoAdminGroup/go-admin/template"
	"github.com/GoAdminGroup/go-admin/template/chartjs"
	"github.com/GoAdminGroup/go-admin/template/types"
	"github.com/gin-gonic/gin"
)

func main() {
	r := gin.Default()

	gin.SetMode(gin.ReleaseMode)
	gin.DefaultWriter = ioutil.Discard

	eng := engine.Default()

	adminPlugin := admin.NewAdmin(tables.Generators)

	// add generator, first parameter is the url prefix of table when visit.
	// example:
	//
	// "user" => http://localhost:9033/admin/info/user
	//
	adminPlugin.AddGenerator("user", tables.GetUserTable)
	adminPlugin.AddGenerator("external", tables.GetExternalTable)

	// customize a plugin
	// 自己定制一个插件👇

	examplePlugin := example.NewExample()

	template.AddComp(chartjs.NewChart())

	// you can also add config like:
	// 您也可以像下面这样的方式去引入数据库👇
	//
	// import "github.com/GoAdminGroup/go-admin/modules/config"
	//
	// cfg := config.Config{
	//	 Databases: config.DatabaseList{
	//		"default": {
	//			Host:       "127.0.0.1",
	//			Port:       "3306",
	//			User:       "root",
	//			Pwd:        "root",
	//			Name:       "godmin",
	//			MaxIdleCon: 50,
	//			MaxOpenCon: 150,
	//			Driver:     db.DriverMysql,
	//		},
	//	},
	//	UrlPrefix: "admin",
	//	IndexUrl:  "/",
	//	Debug:     true,
	//	Language:  language.CN,
	// }
	//
	// eng.AddConfig(cfg)

	if err := eng.AddConfigFromJSON("./config.json").
		AddPlugins(adminPlugin, examplePlugin).
		Use(r); err != nil {
		panic(err)
	}

	r.Static("/uploads", "./uploads")

	// customize your index pages
	// 下面这样定制您的首页👇

	r.GET("/admin", func(ctx *gin.Context) {
		engine.Content(ctx, func(ctx interface{}) (types.Panel, error) {
			return DashboardPage()
		})
	})

	r.GET("/admin/external_link", func(ctx *gin.Context) {
		engine.Content(ctx, func(ctx interface{}) (types.Panel, error) {
			return types.Panel{
				Content: `<iframe style="width:100%;height:800px;" src="https://gitee.com/go-admin/go-admin"></iframe>`,
			}, nil
		})
	})

	r.POST("/admin/popup", func(ctx *gin.Context) {

		user, ok := eng.User(ctx)

		if !ok {
			ctx.JSON(http.StatusOK, gin.H{
				"code": 401,
				"msg":  "auth fail",
			})
			return
		}

		if !user.CheckPermission("*") {
			//ctx.JSON(http.StatusOK, gin.H{
			//	"code": 401,
			//	"msg":  "没有权限",
			//})
			//return
		}

		ctx.JSON(http.StatusOK, gin.H{
			"code": 0,
			"data": "<h2>hello world</h2>",
		})
	})

	_ = r.Run(":9033")
}