Create & Init Project...

app/admin/main/config/pkg/lint/toml/BUILD

@@ -0,0 +1,43 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = [
+        "testdata_test.go",
+        "tomllint_test.go",
+    ],
+    embed = [":go_default_library"],
+    tags = ["automanaged"],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["tomllint.go"],
+    importpath = "go-common/app/admin/main/config/pkg/lint/toml",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/admin/main/config/pkg/lint:go_default_library",
+        "//vendor/github.com/BurntSushi/toml:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/admin/main/config/pkg/lint/toml/testdata_test.go

@@ -0,0 +1,62 @@
+package tomllint
+
+var synataxerrordata = `
+[owner]
+name = "Tom Preston-Werner"
+dob 1979-05-27T07:32:00-08:00 # First class dates
+
+`
+var normaldata = `
+[owner]
+name = "Tom Preston-Werner"
+dob = 1979-05-27T07:32:00-08:00 # First class dates
+
+[database]
+server = "192.168.1.1"
+ports = [ 8001, 8001, 8002 ]
+connection_max = 5000
+enabled = true
+
+[servers]
+
+  # Indentation (tabs and/or spaces) is allowed but not required
+  [servers.alpha]
+  ip = "10.0.0.1"
+  dc = "eqdc10"
+
+  [servers.beta]
+  ip = "10.0.0.2"
+  dc = "eqdc10"
+
+[clients]
+data = [ ["gamma", "delta"], [1, 2] ]
+
+# Line breaks are OK when inside arrays
+hosts = [
+  "alpha",
+  "omega"
+]
+`
+
+var notopkvdata = `
+title = "test123"
+age = 123
+array = [1,2,3]
+[clients]
+data = "helo"
+`
+
+var nocommondata = `
+version = "v1.1"
+user = "nobody"
+`
+
+var noidentify = `
+[identify]
+xxx = 123
+`
+
+var noapp = `
+[app]
+xxx = 123
+`

app/admin/main/config/pkg/lint/toml/tomllint.go

@@ -0,0 +1,100 @@
+package tomllint
+
+import (
+	"io"
+	"regexp"
+	"strconv"
+
+	"github.com/BurntSushi/toml"
+
+	"go-common/app/admin/main/config/pkg/lint"
+)
+
+var lineNumberRe *regexp.Regexp
+
+const filetype = "toml"
+
+type lintFn func(metadata toml.MetaData) []lint.LineErr
+
+var lintFns []lintFn
+
+type tomllint struct{}
+
+// Lint toml file return lint.Error
+func (tomllint) Lint(r io.Reader) lint.Error {
+	var v interface{}
+	var lintErr lint.Error
+	metadata, err := toml.DecodeReader(r, &v)
+	if err != nil {
+		line := -1
+		if match := lineNumberRe.FindStringSubmatch(err.Error()); len(match) == 2 {
+			line, _ = strconv.Atoi(match[1])
+		}
+		lintErr = append(lintErr, lint.LineErr{Line: line, Message: err.Error()})
+		return lintErr
+	}
+	for _, fn := range lintFns {
+		if lineErrs := fn(metadata); lineErrs != nil {
+			lintErr = append(lintErr, lineErrs...)
+		}
+	}
+	if len(lintErr) == 0 {
+		return nil
+	}
+	return lintErr
+}
+
+// not allowed defined kv that type is not Hash at top level
+//func noTopKV(metadata toml.MetaData) []lint.LineErr {
+//	var lineErrs []lint.LineErr
+//	for _, keys := range metadata.Keys() {
+//		if len(keys) != 1 {
+//			continue
+//		}
+//		typeName := metadata.Type(keys...)
+//		if typeName != "Hash" {
+//			lineErrs = append(lineErrs, lint.LineErr{
+//				Line:    -1,
+//				Message: fmt.Sprintf("top level value must be Object, key: %s type is %s", keys[0], typeName),
+//			})
+//		}
+//	}
+//	return lineErrs
+//}
+
+// noApp not allowed app section exists
+func noApp(metadata toml.MetaData) []lint.LineErr {
+	if metadata.IsDefined("app") {
+		return []lint.LineErr{{Line: -1, Message: "请删除无用 App 配置 see: http://git.bilibili.co/platform/go-common/issues/310 (゜-゜)つロ"}}
+	}
+	return nil
+}
+
+// noIdentify not allowed identify config
+func noIdentify(metadata toml.MetaData) []lint.LineErr {
+	if metadata.IsDefined("identify") {
+		return []lint.LineErr{{Line: -1, Message: "请删除无用 Identify 配置 see: http://git.bilibili.co/platform/go-common/issues/310 (゜-゜)つロ"}}
+	}
+	return nil
+}
+
+// noCommon not allowed common config
+func noCommon(metadata toml.MetaData) []lint.LineErr {
+	count := 0
+	commonKey := []string{"version", "user", "pid", "dir", "perf"}
+	for _, key := range commonKey {
+		if metadata.IsDefined(key) {
+			count++
+		}
+	}
+	if count > 0 {
+		return []lint.LineErr{{Line: -1, Message: "请删除无用 Common 配置 see: http://git.bilibili.co/platform/go-common/issues/310 (゜-゜)つロ"}}
+	}
+	return nil
+}
+
+func init() {
+	lint.RegisterLinter(filetype, tomllint{})
+	lintFns = []lintFn{noApp, noIdentify, noCommon}
+	lineNumberRe = regexp.MustCompile("^Near line (\\d+)")
+}

app/admin/main/config/pkg/lint/toml/tomllint_test.go

@@ -0,0 +1,67 @@
+package tomllint
+
+import (
+	"bytes"
+	"strings"
+	"testing"
+)
+
+func TestSyntaxError(t *testing.T) {
+	lint := &tomllint{}
+	r := bytes.NewBufferString(synataxerrordata)
+	lintErr := lint.Lint(r)
+	if lintErr == nil {
+		t.Fatalf("expect lintErr != nil")
+	}
+	if lintErr[0].Line == -1 {
+		t.Errorf("expect get line number")
+	}
+}
+
+func TestTomlLintOK(t *testing.T) {
+	lint := &tomllint{}
+	r := bytes.NewBufferString(normaldata)
+	lintErr := lint.Lint(r)
+	if lintErr != nil {
+		t.Errorf("error %v", lintErr)
+	}
+}
+
+func TestNoCommon(t *testing.T) {
+	lint := &tomllint{}
+	r := bytes.NewBufferString(nocommondata)
+	lintErr := lint.Lint(r)
+	if lintErr == nil {
+		t.Fatalf("expect lintErr != nil")
+	}
+	message := lintErr.Error()
+	if !strings.Contains(message, "Common") {
+		t.Errorf("expect error contains common")
+	}
+}
+
+func TestNoIdentify(t *testing.T) {
+	lint := &tomllint{}
+	r := bytes.NewBufferString(noidentify)
+	lintErr := lint.Lint(r)
+	if lintErr == nil {
+		t.Fatalf("expect lintErr != nil")
+	}
+	message := lintErr.Error()
+	if !strings.Contains(message, "Identify") {
+		t.Errorf("expect error Identify common")
+	}
+}
+
+func TestNoApp(t *testing.T) {
+	lint := &tomllint{}
+	r := bytes.NewBufferString(noapp)
+	lintErr := lint.Lint(r)
+	if lintErr == nil {
+		t.Fatalf("expect lintErr != nil")
+	}
+	message := lintErr.Error()
+	if !strings.Contains(message, "App") {
+		t.Errorf("expect error App common")
+	}
+}