💩👌 make lint happy

plugin_diana/text.go → plugin_diana/data/text.go

@@ -1,4 +1,4 @@
-package diana
+package data
 
 import (
 	"io"
@@ -13,7 +13,7 @@ const (
 
 var (
 	compo Composition
-	array []string
+	Array []string
 )
 
 func init() {
@@ -23,13 +23,13 @@ func init() {
 		if err != nil {
 			panic(err)
 		}
-		if loadText() == nil {
-			array = compo.Array
+		if LoadText() == nil {
+			Array = compo.Array
 		}
 	}()
 }
 
-func loadText() error {
+func LoadText() error {
 	if _, err := os.Stat(pbfile); err == nil || os.IsExist(err) {
 		f, err := os.Open(pbfile)
 		if err == nil {
@@ -46,9 +46,9 @@ func loadText() error {
 	return nil
 }
 
-func addText(txt string) error {
+func AddText(txt string) error {
 	if txt != "" {
-		array = append(array, txt)
+		compo.Array = append(compo.Array, txt)
 		data, err := compo.Marshal()
 		if err == nil {
 			if _, err := os.Stat(datapath); err == nil || os.IsExist(err) {

plugin_diana/text.proto → plugin_diana/data/text.proto

@@ -1,5 +1,5 @@
 syntax = "proto3";
-package diana;
+package data;
 
 message composition {
     repeated string array = 1;

plugin_diana/tools/convert.go

@@ -0,0 +1,44 @@
+// Package convert 转换txt到pb
+package main
+
+import (
+	"bufio"
+	"os"
+
+	"github.com/FloatTech/ZeroBot-Plugin/plugin_diana/data"
+)
+
+var (
+	compo data.Composition
+)
+
+func init() {
+	compo.Array = make([]string, 0, 64)
+}
+
+// 参数：txt文件位置 pb文件位置
+func main() {
+	file, err := os.Open(os.Args[1])
+	if err != nil {
+		panic(err)
+	}
+	defer file.Close()
+
+	scanner := bufio.NewScanner(file)
+	// optionally, resize scanner's capacity for lines over 64K, see next example
+	for scanner.Scan() {
+		//fmt.Println(scanner.Text())
+		compo.Array = append(compo.Array, scanner.Text())
+	}
+
+	if err := scanner.Err(); err != nil {
+		panic(err)
+	}
+
+	data, _ := compo.Marshal()
+	f, err1 := os.OpenFile(os.Args[2], os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
+	if err1 == nil {
+		defer f.Close()
+		f.Write(data)
+	}
+}