feat(config): 添加 JSON/TOML 配置读取与 YAML/JSON/TOML 导出功能, 优化部分内部代码结构, 更新 Makefile:添加 Zig 外链检测、可选 UPX 压缩、构建流程优化
This commit is contained in:
65
Makefile
65
Makefile
@@ -1,23 +1,64 @@
|
|||||||
APP_NAME = netctl
|
APP_NAME := netctl
|
||||||
SRC = .
|
SRC := .
|
||||||
OUTPUT = ./bin/$(APP_NAME).exe
|
OUTPUT := ./bin/$(APP_NAME).exe
|
||||||
GO = go
|
GO := go
|
||||||
|
|
||||||
# 获取 Git commit
|
# Version info
|
||||||
|
VERSION := 1.0.1
|
||||||
GIT_COMMIT := $(shell git rev-parse --short HEAD 2>/dev/null || echo none)
|
GIT_COMMIT := $(shell git rev-parse --short HEAD 2>/dev/null || echo none)
|
||||||
# 获取当前时间 UTC
|
|
||||||
BUILD_TIME := $(shell date -u +%Y-%m-%dT%H:%M:%SZ)
|
BUILD_TIME := $(shell date -u +%Y-%m-%dT%H:%M:%SZ)
|
||||||
|
|
||||||
.PHONY: build run clean
|
# Detect tools
|
||||||
|
DETECTED_ZIG := $(shell command -v zig 2>/dev/null)
|
||||||
|
DETECTED_UPX := $(shell command -v upx 2>/dev/null)
|
||||||
|
|
||||||
|
# User config
|
||||||
|
ZIG ?= 1 # auto enable zig linker if exists
|
||||||
|
BRUTE ?= 0 # UPX brute mode disabled by default
|
||||||
|
|
||||||
|
.PHONY: all build run clean
|
||||||
|
|
||||||
|
all: build
|
||||||
|
|
||||||
# 编译命令
|
|
||||||
build:
|
build:
|
||||||
$(GO) build -ldflags "-X 'netcli/cmd.Version=1.0.0' -X 'netcli/cmd.Commit=$(GIT_COMMIT)' -X 'netcli/cmd.BuildTime=$(BUILD_TIME)'" -o $(OUTPUT) $(SRC)
|
@mkdir -p ./bin
|
||||||
|
|
||||||
|
@if [ "$(ZIG)" = "1" ] && [ -n "$(DETECTED_ZIG)" ]; then \
|
||||||
|
echo "Zig detected -> Using Zig as Go external linker"; \
|
||||||
|
export CGO_ENABLED=0; \
|
||||||
|
export CC="zig cc"; \
|
||||||
|
export CXX="zig c++"; \
|
||||||
|
$(GO) build -trimpath -ldflags "-s -w \
|
||||||
|
-X 'netcli/cmd.Version=$(VERSION)' \
|
||||||
|
-X 'netcli/cmd.Commit=$(GIT_COMMIT)' \
|
||||||
|
-X 'netcli/cmd.BuildTime=$(BUILD_TIME)'" \
|
||||||
|
-o $(OUTPUT) $(SRC); \
|
||||||
|
else \
|
||||||
|
echo "Zig not used → Using Go native linker"; \
|
||||||
|
$(GO) build -trimpath -ldflags "-s -w \
|
||||||
|
-X 'netcli/cmd.Version=$(VERSION)' \
|
||||||
|
-X 'netcli/cmd.Commit=$(GIT_COMMIT)' \
|
||||||
|
-X 'netcli/cmd.BuildTime=$(BUILD_TIME)'" \
|
||||||
|
-o $(OUTPUT) $(SRC); \
|
||||||
|
fi
|
||||||
|
|
||||||
|
@if [ -n "$(DETECTED_UPX)" ]; then \
|
||||||
|
if [ "$(BRUTE)" = "1" ]; then \
|
||||||
|
echo "UPX brute mode enabled (--ultra-brute)..."; \
|
||||||
|
upx --ultra-brute $(OUTPUT); \
|
||||||
|
else \
|
||||||
|
echo "UPX detected → Applying normal compression (--best --lzma)..."; \
|
||||||
|
upx --best --lzma $(OUTPUT); \
|
||||||
|
fi \
|
||||||
|
else \
|
||||||
|
echo "UPX not found → Skipping compression"; \
|
||||||
|
fi
|
||||||
|
|
||||||
# 直接运行
|
|
||||||
run:
|
run:
|
||||||
$(GO) run -ldflags "-X 'netcli/cmd.Version=1.0.0' -X 'netcli/cmd.Commit=$(GIT_COMMIT)' -X 'netcli/cmd.BuildTime=$(BUILD_TIME)'" $(SRC)
|
$(GO) run -ldflags "-s -w \
|
||||||
|
-X 'netcli/cmd.Version=$(VERSION)' \
|
||||||
|
-X 'netcli/cmd.Commit=$(GIT_COMMIT)' \
|
||||||
|
-X 'netcli/cmd.BuildTime=$(BUILD_TIME)'" $(SRC)
|
||||||
|
|
||||||
# 清理编译文件
|
|
||||||
clean:
|
clean:
|
||||||
rm -f $(OUTPUT)
|
rm -f $(OUTPUT)
|
||||||
|
|||||||
19
cmd/dhcp.go
19
cmd/dhcp.go
@@ -2,10 +2,10 @@ package cmd
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
stdnet "net"
|
"net"
|
||||||
|
|
||||||
"netcli/internal/constants"
|
"netcli/internal/constants"
|
||||||
"netcli/internal/net"
|
netmod "netcli/internal/net"
|
||||||
|
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
@@ -16,30 +16,25 @@ var dhcpCmd = &cobra.Command{
|
|||||||
Short: "将指定网卡改为 DHCP 自动获取 IP",
|
Short: "将指定网卡改为 DHCP 自动获取 IP",
|
||||||
RunE: func(cmd *cobra.Command, args []string) error {
|
RunE: func(cmd *cobra.Command, args []string) error {
|
||||||
// 检查管理员权限
|
// 检查管理员权限
|
||||||
if !net.IsAdmin() {
|
if !netmod.IsAdmin() {
|
||||||
fmt.Println("程序需要管理员权限,尝试提升...")
|
fmt.Println("程序需要管理员权限,尝试提升...")
|
||||||
return net.RunAsAdmin()
|
return netmod.RunAsAdmin()
|
||||||
}
|
}
|
||||||
|
|
||||||
// 获取所有网卡
|
// 获取所有网卡
|
||||||
ifaces, err := stdnet.Interfaces()
|
ifaces, err := net.Interfaces()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("获取网卡失败: %w", err)
|
return fmt.Errorf("获取网卡失败: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
var nics []net.NetIfWrap
|
|
||||||
for _, i := range ifaces {
|
|
||||||
nics = append(nics, net.NetIfWrap{i})
|
|
||||||
}
|
|
||||||
|
|
||||||
// 用户选择网卡
|
// 用户选择网卡
|
||||||
nic, err := net.ChooseNic(nics, constants.DefaultDHCPMsg)
|
nic, err := netmod.ChooseNic(ifaces, constants.DefaultDHCPMsg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// 执行 netsh 命令设置 DHCP
|
// 执行 netsh 命令设置 DHCP
|
||||||
if err := net.SetDHCP(nic); err != nil {
|
if err := netmod.SetDHCP(nic); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
89
cmd/export.go
Normal file
89
cmd/export.go
Normal file
@@ -0,0 +1,89 @@
|
|||||||
|
package cmd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"netcli/config"
|
||||||
|
|
||||||
|
"github.com/pelletier/go-toml/v2"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
"gopkg.in/yaml.v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
var exportCmd = &cobra.Command{
|
||||||
|
Use: "export",
|
||||||
|
Short: "导出默认配置文件",
|
||||||
|
Long: "可以将默认配置导出为 YAML, TOML, JSON 格式文件,支持自定义路径和格式, 生效顺序: YAML, TOML, JSON",
|
||||||
|
RunE: func(cmd *cobra.Command, args []string) error {
|
||||||
|
// 获取参数
|
||||||
|
filePath, _ := cmd.Flags().GetString("file")
|
||||||
|
format, _ := cmd.Flags().GetString("format")
|
||||||
|
|
||||||
|
// 默认文件名和格式
|
||||||
|
if filePath == "" {
|
||||||
|
filePath = "config.yaml"
|
||||||
|
}
|
||||||
|
if format == "" {
|
||||||
|
format = "yaml"
|
||||||
|
}
|
||||||
|
|
||||||
|
format = strings.ToLower(format)
|
||||||
|
if format != "yaml" && format != "yml" && format != "toml" && format != "json" {
|
||||||
|
return fmt.Errorf("不支持的格式: %s, 可选: yaml, toml, json", format)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根据格式修改文件后缀
|
||||||
|
ext := strings.ToLower(filepath.Ext(filePath))
|
||||||
|
if ext == "" || ext != "."+format {
|
||||||
|
filePath = strings.TrimSuffix(filePath, ext) + "." + format
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取默认配置
|
||||||
|
defaultConfig := config.DefaultConfig()
|
||||||
|
|
||||||
|
// 创建文件
|
||||||
|
f, err := os.Create(filePath)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("创建文件失败: %w", err)
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
// 写入不同格式
|
||||||
|
switch format {
|
||||||
|
case "yaml", "yml":
|
||||||
|
enc := yaml.NewEncoder(f)
|
||||||
|
defer enc.Close()
|
||||||
|
if err := enc.Encode(defaultConfig); err != nil {
|
||||||
|
return fmt.Errorf("YAML 编码失败: %w", err)
|
||||||
|
}
|
||||||
|
case "json":
|
||||||
|
enc := json.NewEncoder(f)
|
||||||
|
enc.SetIndent("", " ")
|
||||||
|
if err := enc.Encode(defaultConfig); err != nil {
|
||||||
|
return fmt.Errorf("JSON 编码失败: %w", err)
|
||||||
|
}
|
||||||
|
case "toml":
|
||||||
|
data, err := toml.Marshal(defaultConfig)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("TOML 编码失败: %w", err)
|
||||||
|
}
|
||||||
|
if _, err := f.Write(data); err != nil {
|
||||||
|
return fmt.Errorf("写入 TOML 文件失败: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
absPath, _ := filepath.Abs(filePath)
|
||||||
|
fmt.Printf("默认配置已导出到: %s\n", absPath)
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
exportCmd.Flags().StringP("file", "f", "", "导出文件路径 (默认 ./config.yaml)")
|
||||||
|
exportCmd.Flags().StringP("format", "t", "yaml", "导出格式: yaml, toml, json (默认 yaml)")
|
||||||
|
rootCmd.AddCommand(exportCmd)
|
||||||
|
}
|
||||||
@@ -25,20 +25,14 @@ var staticCmd = &cobra.Command{
|
|||||||
|
|
||||||
// 获取本机网卡
|
// 获取本机网卡
|
||||||
ifaces, _ := stdnet.Interfaces()
|
ifaces, _ := stdnet.Interfaces()
|
||||||
var nics []netmod.NetIfWrap
|
|
||||||
for _, i := range ifaces {
|
|
||||||
nics = append(nics, netmod.NetIfWrap{i})
|
|
||||||
}
|
|
||||||
|
|
||||||
nic, _ := netmod.ChooseNic(nics, constants.DefaultStaticMsg)
|
nic, _ := netmod.ChooseNic(ifaces, constants.DefaultStaticMsg)
|
||||||
|
|
||||||
// 加载配置文件
|
// 加载配置文件
|
||||||
cfg, _ := config.LoadConfig(constants.DefaultConfig)
|
cfg := config.LoadDefaultConfig()
|
||||||
var rooms []model.ClassRoom
|
rooms := cfg.ClassRooms
|
||||||
if cfg != nil && len(cfg.ClassRooms) > 0 {
|
if len(rooms) == 0 {
|
||||||
rooms = cfg.ClassRooms
|
rooms = config.DefaultConfig().ClassRooms
|
||||||
} else {
|
|
||||||
rooms = config.DefaultClassRooms()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 用户选择教室
|
// 用户选择教室
|
||||||
|
|||||||
1
go.mod
1
go.mod
@@ -24,4 +24,5 @@ require (
|
|||||||
golang.org/x/sys v0.29.0 // indirect
|
golang.org/x/sys v0.29.0 // indirect
|
||||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211 // indirect
|
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211 // indirect
|
||||||
golang.org/x/text v0.28.0 // indirect
|
golang.org/x/text v0.28.0 // indirect
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
)
|
)
|
||||||
|
|||||||
1
go.sum
1
go.sum
@@ -80,4 +80,5 @@ golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc
|
|||||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
|||||||
@@ -1,13 +1,9 @@
|
|||||||
package model
|
package model
|
||||||
|
|
||||||
type ClassRoom struct {
|
type ClassRoom struct {
|
||||||
ID string
|
ID string `mapstructure:"id"`
|
||||||
Name string
|
Name string `mapstructure:"name"`
|
||||||
Addr string
|
Addr string `mapstructure:"addr"`
|
||||||
Mask string
|
Mask string `mapstructure:"mask"`
|
||||||
Gateway string
|
Gateway string `mapstructure:"gateway"`
|
||||||
}
|
|
||||||
|
|
||||||
func (c ClassRoom) GetName() string {
|
|
||||||
return c.Name
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +0,0 @@
|
|||||||
package model
|
|
||||||
|
|
||||||
type Named interface {
|
|
||||||
GetName() string
|
|
||||||
}
|
|
||||||
@@ -10,16 +10,8 @@ import (
|
|||||||
"golang.org/x/sys/windows"
|
"golang.org/x/sys/windows"
|
||||||
)
|
)
|
||||||
|
|
||||||
type NetIfWrap struct {
|
|
||||||
net.Interface
|
|
||||||
}
|
|
||||||
|
|
||||||
func (n NetIfWrap) GetName() string {
|
|
||||||
return n.Interface.Name
|
|
||||||
}
|
|
||||||
|
|
||||||
// 选择网卡
|
// 选择网卡
|
||||||
func ChooseNic(nics []NetIfWrap, message string) (NetIfWrap, error) {
|
func ChooseNic(nics []net.Interface, message string) (net.Interface, error) {
|
||||||
var names []string
|
var names []string
|
||||||
for _, nic := range nics {
|
for _, nic := range nics {
|
||||||
names = append(names, nic.Name)
|
names = append(names, nic.Name)
|
||||||
@@ -30,18 +22,18 @@ func ChooseNic(nics []NetIfWrap, message string) (NetIfWrap, error) {
|
|||||||
Options: names,
|
Options: names,
|
||||||
}, &choice)
|
}, &choice)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return NetIfWrap{}, err
|
return net.Interface{}, err
|
||||||
}
|
}
|
||||||
for _, nic := range nics {
|
for _, nic := range nics {
|
||||||
if nic.Name == choice {
|
if nic.Name == choice {
|
||||||
return nic, nil
|
return nic, nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return NetIfWrap{}, fmt.Errorf("未找到网卡: %s", choice)
|
return net.Interface{}, fmt.Errorf("未找到网卡: %s", choice)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 设置静态IP
|
// 设置静态IP
|
||||||
func SetStaticIP(nic NetIfWrap, addr, mask, gateway string) error {
|
func SetStaticIP(nic net.Interface, addr, mask, gateway string) error {
|
||||||
cmd := exec.Command("netsh", "interface", "ip", "set", "address",
|
cmd := exec.Command("netsh", "interface", "ip", "set", "address",
|
||||||
fmt.Sprintf("name=%s", nic.Name),
|
fmt.Sprintf("name=%s", nic.Name),
|
||||||
"static", addr, mask, gateway)
|
"static", addr, mask, gateway)
|
||||||
@@ -53,7 +45,7 @@ func SetStaticIP(nic NetIfWrap, addr, mask, gateway string) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 设置 DHCP
|
// 设置 DHCP
|
||||||
func SetDHCP(nic NetIfWrap) error {
|
func SetDHCP(nic net.Interface) error {
|
||||||
cmd := exec.Command("netsh", "interface", "ip", "set", "address",
|
cmd := exec.Command("netsh", "interface", "ip", "set", "address",
|
||||||
fmt.Sprintf("name=%s", nic.Name), "dhcp")
|
fmt.Sprintf("name=%s", nic.Name), "dhcp")
|
||||||
out, err := cmd.CombinedOutput()
|
out, err := cmd.CombinedOutput()
|
||||||
|
|||||||
Reference in New Issue
Block a user