contractAmin/main.go at master · mover-code/contractAmin · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
package main

import (
	"context"
	"flag"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"os"
	"os/signal"
	"time"

	"github.com/GoAdminGroup/example/handler"
	_ "github.com/GoAdminGroup/example/theme2"
	_ "github.com/GoAdminGroup/go-admin/adapter/gin" // web framework adapter
	"github.com/GoAdminGroup/go-admin/examples/datamodel"
	_ "github.com/GoAdminGroup/go-admin/modules/db/drivers/sqlite" // sql driver
	"github.com/GoAdminGroup/go-admin/plugins/admin"
	_ "github.com/GoAdminGroup/themes/adminlte" // ui theme

	"github.com/GoAdminGroup/components/login"
	"github.com/GoAdminGroup/example/models"
	"github.com/GoAdminGroup/example/pages"
	"github.com/GoAdminGroup/example/tables"
	"github.com/GoAdminGroup/go-admin/engine"
	"github.com/GoAdminGroup/go-admin/template"
	"github.com/GoAdminGroup/go-admin/template/chartjs"
	"github.com/gin-gonic/gin"
)

var configFile = flag.String("f", "etc/config.json", "the config file")
var port = flag.String("p", "9900", "port")

func main() {
	flag.Parse()
	fmt.Println("port:", *port, "config file:", *configFile)
	startServer()
}

func startServer() {
	gin.SetMode(gin.ReleaseMode)
	gin.DefaultWriter = ioutil.Discard

	login.Init(login.Config{
		Theme:         "theme2",
		CaptchaDigits: 4, // 使用图片验证码，这里代表多少个验证码数字
		// 使用腾讯验证码，需提供appID与appSecret
		// TencentWaterProofWallData: login.TencentWaterProofWallData{
		//    AppID:"",
		//    AppSecret: "",
		// }
	})

	r := gin.Default()
	r.GET("/admin/getContract", handler.Contract)

	eng := engine.Default()

	template.AddComp(chartjs.NewChart())
	adminPlugin := admin.NewAdmin(datamodel.Generators)
	if err := eng.AddConfigFromJSON(*configFile).
		AddPlugins(adminPlugin).
		AddGenerators(tables.Generators).
		AddGenerator("external", tables.GetExternalTable).
		Use(r); err != nil {
		panic(err)
	}
	// cfg := config.ReadFromJson("./config.json")
	// fmt.Println(fmt.Sprintf("%+v", cfg.Databases.GroupByDriver()))
	// eng.PluginList
	adminPlugin.SetCaptcha(map[string]string{"driver": login.CaptchaDriverKeyDefault})
	// captcha.Add(login.CaptchaDriverKeyDefault, new(login.DigitsCaptcha))

	// config mysql use
	// models.Init(eng.DefaultConnection(), eng.MysqlConnection())
	models.Init(eng.DefaultConnection(), nil)
	r.Static("/uploads", "./uploads")
	r.Static("/admin/web3", "./html/contractReader")

	eng.HTML("GET", "/admin", pages.DashboardPage)
	eng.HTML("GET", "/admin/form", pages.GetFormContent)
	eng.HTML("GET", "/admin/table", pages.GetTableContent)
	eng.HTMLFile("GET", "/admin/hello", "./html/hello.tmpl", nil)
	eng.HTMLFile("GET", "/admin/dao", "./html/contract.tmpl", nil)

	srv := &http.Server{
		Addr:    ":" + *port,
		Handler: r,
	}

	go func() {
		if err := srv.ListenAndServe(); err != nil {
			log.Printf("listen: %s\n", err)
		}
	}()
	log.Printf("server start listen: http://127.0.0.1:%s/admin\n", *port)

	quit := make(chan os.Signal)
	signal.Notify(quit, os.Interrupt)
	<-quit

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	if err := srv.Shutdown(ctx); err != nil {
		log.Fatal("Server Shutdown:", err)
	}
	log.Println("Server exiting")
}

func Cors() gin.HandlerFunc {
	return func(c *gin.Context) {
		method := c.Request.Method
		origin := c.Request.Header.Get("Origin") //请求头部
		if origin != "" {
			//接收客户端发送的origin （重要！）
			c.Writer.Header().Set("Access-Control-Allow-Origin", origin)
			//服务器支持的所有跨域请求的方法
			c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE,UPDATE")
			//允许跨域设置可以返回其他子段，可以自定义字段
			c.Header("Access-Control-Allow-Headers", "Authorization, Content-Length, X-CSRF-Token, Token,session")
			// 允许浏览器（客户端）可以解析的头部 （重要）
			c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers")
			//设置缓存时间
			c.Header("Access-Control-Max-Age", "172800")
			//允许客户端传递校验信息比如 cookie (重要)
			c.Header("Access-Control-Allow-Credentials", "true")
		}

		//允许类型校验
		if method == "OPTIONS" {
			c.JSON(http.StatusOK, "ok!")
		}

		defer func() {
			if err := recover(); err != nil {
				log.Printf("Panic info is: %v", err)
			}
		}()

		c.Next()
	}
}