feat: 项目初始化

cmd/dhcp.go

@@ -0,0 +1,53 @@
+package cmd
+
+import (
+	"fmt"
+	stdnet "net"
+
+	"netcli/internal/constants"
+	"netcli/internal/net"
+
+	"github.com/spf13/cobra"
+)
+
+// dhcpCmd represents the dhcp command
+var dhcpCmd = &cobra.Command{
+	Use:   "dhcp",
+	Short: "将指定网卡改为 DHCP 自动获取 IP",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		// 检查管理员权限
+		if !net.IsAdmin() {
+			fmt.Println("程序需要管理员权限，尝试提升...")
+			return net.RunAsAdmin()
+		}
+
+		// 获取所有网卡
+		ifaces, err := stdnet.Interfaces()
+		if err != nil {
+			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)
+		if err != nil {
+			return err
+		}
+
+		// 执行 netsh 命令设置 DHCP
+		if err := net.SetDHCP(nic); err != nil {
+			return err
+		}
+
+		fmt.Printf("网卡 %s 已设置为 DHCP 自动获取 IP\n", nic.Name)
+		return nil
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(dhcpCmd)
+}

cmd/root.go

@@ -0,0 +1,46 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+
+	netmod "netcli/internal/net"
+
+	"github.com/spf13/cobra"
+)
+
+// rootCmd represents the base command when called without any subcommands
+var rootCmd = &cobra.Command{
+	Use:   "netctl",
+	Short: "网络切换工具",
+	Long:  "netctl 是一个用于切换网卡静态IP/DHCP的 CLI 工具",
+}
+
+// Execute adds all child commands to the root command and sets flags appropriately.
+// This is called by main.main().
+func Execute() {
+	if err := rootCmd.Execute(); err != nil {
+		os.Exit(1)
+	}
+}
+
+// 通用命令运行函数
+func run(args ...string) {
+	cmd := exec.Command(args[0], args[1:]...)
+	out, err := cmd.CombinedOutput()
+	fmt.Println(string(out))
+	if err != nil {
+		panic(err)
+	}
+}
+
+// 判断当前程序是否管理员
+func IsAdmin() bool {
+	return netmod.IsAdmin()
+}
+
+// 提升管理员权限
+func RunAsAdmin() error {
+	return netmod.RunAsAdmin()
+}

cmd/static.go

@@ -0,0 +1,70 @@
+package cmd
+
+import (
+	"fmt"
+	stdnet "net"
+
+	"netcli/config"
+	"netcli/internal/constants"
+	"netcli/internal/model"
+	netmod "netcli/internal/net"
+
+	"github.com/AlecAivazis/survey/v2"
+	"github.com/spf13/cobra"
+)
+
+var staticCmd = &cobra.Command{
+	Use:   "static",
+	Short: "设置静态IP",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		// 管理员权限检查
+		if !netmod.IsAdmin() {
+			fmt.Println("程序需要管理员权限，尝试提升...")
+			return netmod.RunAsAdmin()
+		}
+
+		// 获取本机网卡
+		ifaces, _ := stdnet.Interfaces()
+		var nics []netmod.NetIfWrap
+		for _, i := range ifaces {
+			nics = append(nics, netmod.NetIfWrap{i})
+		}
+
+		nic, _ := netmod.ChooseNic(nics, constants.DefaultStaticMsg)
+
+		// 加载配置文件
+		cfg, _ := config.LoadConfig(constants.DefaultConfig)
+		var rooms []model.ClassRoom
+		if cfg != nil && len(cfg.ClassRooms) > 0 {
+			rooms = cfg.ClassRooms
+		} else {
+			rooms = config.DefaultClassRooms()
+		}
+
+		// 用户选择教室
+		var roomNames []string
+		for _, r := range rooms {
+			roomNames = append(roomNames, r.Name)
+		}
+
+		var choice string
+		survey.AskOne(&survey.Select{Message: "请选择教室:", Options: roomNames}, &choice)
+
+		var room model.ClassRoom
+		for _, r := range rooms {
+			if r.Name == choice {
+				room = r
+				break
+			}
+		}
+
+		fmt.Printf("正在设置静态IP...\n网卡: %s\nIP: %s\nMask: %s\nGateway: %s\n",
+			nic.Name, room.Addr, room.Mask, room.Gateway)
+
+		return netmod.SetStaticIP(nic, room.Addr, room.Mask, room.Gateway)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(staticCmd)
+}

cmd/version.go

@@ -0,0 +1,28 @@
+package cmd
+
+import (
+	"fmt"
+	"github.com/spf13/cobra"
+)
+
+// 编译时注入变量
+var (
+	Version   = "dev" // 默认版本
+	Commit    = "none"
+	BuildTime = "unknown"
+)
+
+var versionCmd = &cobra.Command{
+	Use:   "version",
+	Short: "显示软件版本信息",
+	Run: func(cmd *cobra.Command, args []string) {
+		fmt.Println("netctl - 网络切换工具")
+		fmt.Printf("Version: %s\n", Version)
+		fmt.Printf("Commit: %s\n", Commit)
+		fmt.Printf("BuildTime: %s\n", BuildTime)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(versionCmd)
+}