mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-20 06:20:08 +08:00
53 lines
925 B
Go
53 lines
925 B
Go
// Package convert 转换txt到pb
|
||
package main
|
||
|
||
import (
|
||
"bufio"
|
||
"fmt"
|
||
"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()
|
||
_, err2 := f.Write(data)
|
||
if err2 == nil {
|
||
fmt.Println("成功")
|
||
} else {
|
||
panic(err2)
|
||
}
|
||
} else {
|
||
panic(err1)
|
||
}
|
||
}
|