mirror of
https://github.com/ayflying/p2p.git
synced 2026-03-04 17:29:22 +00:00
111 lines
3.2 KiB
Go
111 lines
3.2 KiB
Go
package cmd
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/ayflying/p2p/internal/service"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/net/ghttp"
|
||
"github.com/gogf/gf/v2/os/gcmd"
|
||
)
|
||
|
||
// p2pHelpDescription 定义P2P命令的详细帮助信息
|
||
const p2pHelpDescription = `
|
||
P2P连接工具使用帮助:
|
||
|
||
模式1: 网关服务器
|
||
功能: 拥有外网IP，接收客户端连接，协助P2P打洞
|
||
命令: p2p -a gateway
|
||
|
||
模式2: 客户端
|
||
功能: 连接到网关，通过打洞实现与其他客户端的长连接通讯
|
||
命令: p2p -a client --gateway 网关ID
|
||
|
||
高级功能:
|
||
客户端间连接: p2p --mode client --gateway 网关ID --action connect --target 目标客户端ID
|
||
发送消息: p2p -mode client --gateway 网关ID --action send --target 目标客户端ID --message "消息内容"
|
||
`
|
||
|
||
var (
|
||
// P2p 命令定义了P2P连接工具的入口命令
|
||
// 遵循GoFrame的Command对象定义规范，包含名称、用法、简短描述和执行函数
|
||
P2p = gcmd.Command{
|
||
// Name 为命令名称
|
||
Name: "p2p",
|
||
// Usage 描述命令的基本用法
|
||
Usage: "p2p [options]",
|
||
// Brief 提供命令的简短功能描述
|
||
Brief: "P2P连接工具，支持网关和客户端模式，实现NAT穿透和点对点通信",
|
||
// Description 提供命令的详细描述和使用帮助
|
||
Description: p2pHelpDescription,
|
||
// Func 为命令的执行函数，接收上下文和参数解析器
|
||
Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
|
||
s := g.Server()
|
||
|
||
// 配置日志输出
|
||
g.Log().SetConfigWithMap(g.Map{
|
||
"level": "all",
|
||
"stdout": true,
|
||
})
|
||
|
||
parser, err = gcmd.Parse(g.MapStrBool{
|
||
"m,mode": true,
|
||
"g,gateway": true,
|
||
"a,action": true,
|
||
"t,target": true,
|
||
})
|
||
|
||
// 获取运行模式参数
|
||
mode := parser.GetOpt("mode").String()
|
||
|
||
// 根据不同模式调用服务层对应的方法
|
||
switch mode {
|
||
case "gateway":
|
||
// 启动网关服务器模式
|
||
g.Log().Debug(ctx, "开始执行gatway")
|
||
s.Group("/ws", func(group *ghttp.RouterGroup) {
|
||
group.Middleware(ghttp.MiddlewareHandlerResponse)
|
||
service.P2P().GatewayStart(ctx, group)
|
||
})
|
||
case "client":
|
||
// 获取客户端模式所需的参数
|
||
g.Log().Debug(ctx, "开始执行client")
|
||
//addrs := []string{"/ip4/127.0.0.1/tcp/51888", "/ip4/192.168.50.173/tcp/51888"}
|
||
//addr := "/ip4/192.168.50.173/tcp/51888/p2p/12D3KooWJKBB9bF9MjqgsFYUUsPBG249FDq7a3ZdaYc9iw8G78JQ"
|
||
//addrs := "WyIvaXA0LzEyNy4wLjAuMS90Y3AvNTE4ODgiLCIvaXA0LzE5Mi4xNjguNTAuMTczL3RjcC81MTg4OCJd"
|
||
wsStr := "ws://192.168.50.173:51888/ws"
|
||
err = service.P2P().Start(ctx, wsStr)
|
||
case "dht":
|
||
h, _ := service.P2P().CreateLibp2pHost(ctx, 0)
|
||
err := service.P2P().DHTStart(ctx, h)
|
||
if err != nil {
|
||
g.Log().Error(ctx, err)
|
||
}
|
||
|
||
publicIp, err := service.P2P().GetIPv4PublicIP()
|
||
err = service.P2P().StoreAddrToDHT(ctx, "ip", publicIp)
|
||
|
||
case "dht2":
|
||
h, _ := service.P2P().CreateLibp2pHost(ctx, 0)
|
||
err := service.P2P().DHTStart(ctx, h)
|
||
if err != nil {
|
||
g.Log().Error(ctx, err)
|
||
}
|
||
get, _ := service.P2P().FindAddrFromDHT(ctx, "ip")
|
||
g.Dump(get)
|
||
s.SetPort(0)
|
||
default:
|
||
// 显示帮助信息
|
||
g.Log().Info(ctx, p2pHelpDescription)
|
||
}
|
||
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
s.Run()
|
||
return
|
||
},
|
||
}
|
||
)
|