feat: 命令行支持切换监听 IP (#2671)

cmd/server/cmd/listen-ip.go

@@ -0,0 +1,60 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	RootCmd.AddCommand(listenCmd)
+	listenCmd.AddCommand(listenIpv4Cmd)
+	listenCmd.AddCommand(listenIpv6Cmd)
+}
+
+var listenCmd = &cobra.Command{
+	Use:   "listen-ip",
+	Short: "切换监听 IP",
+}
+
+var listenIpv4Cmd = &cobra.Command{
+	Use:   "ipv4",
+	Short: "监听 IPv4",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		return updateBindInfo("ipv4")
+	},
+}
+var listenIpv6Cmd = &cobra.Command{
+	Use:   "ipv6",
+	Short: "监听 IPv6",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		return updateBindInfo("ipv6")
+	},
+}
+
+func updateBindInfo(protocol string) error {
+	if !isRoot() {
+		fmt.Println("请使用 sudo 1pctl listen-ip ipv6 或者切换到 root 用户")
+		return nil
+	}
+	db, err := loadDBConn()
+	if err != nil {
+		return err
+	}
+	ipv6 := "disable"
+	tcp := "tcp4"
+	address := "0.0.0.0"
+	if protocol == "ipv6" {
+		ipv6 = "enable"
+		tcp = "tcp6"
+		address = "::"
+	}
+	if err := setSettingByKey(db, "Ipv6", ipv6); err != nil {
+		return err
+	}
+	if err := setSettingByKey(db, "BindAddress", address); err != nil {
+		return err
+	}
+	fmt.Printf("切换成功！已切换至监听 %s [%s]", tcp, address)
+	return nil
+}