executor.go 84 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706
  1. package cli
  2. import (
  3. "context"
  4. "fmt"
  5. "net"
  6. "os/exec"
  7. "sort"
  8. "strconv"
  9. "strings"
  10. "time"
  11. "debtik/internal/auth"
  12. "debtik/internal/bridge"
  13. "debtik/internal/bwtest"
  14. "debtik/internal/certs"
  15. "debtik/internal/dhcp"
  16. "debtik/internal/dns"
  17. "debtik/internal/dynamicrouting"
  18. "debtik/internal/firewall"
  19. "debtik/internal/logging"
  20. "debtik/internal/network"
  21. "debtik/internal/reverseproxy"
  22. "debtik/internal/routing"
  23. "debtik/internal/services"
  24. "debtik/internal/system"
  25. "debtik/internal/vpn"
  26. "debtik/internal/vrrp"
  27. )
  28. type Executor struct {
  29. authMgr *auth.Manager
  30. netMgr *network.Manager
  31. routingMgr *routing.RoutingManager
  32. dynRoutingMgr *dynamicrouting.Manager
  33. fwMgr *firewall.FirewallManager
  34. dhcpMgr *dhcp.Manager
  35. dhcpServMgr *dhcp.ServerManager
  36. dnsMgr *dns.Manager
  37. bridgeMgr *bridge.BridgeManager
  38. vpnMgr *vpn.Manager
  39. vrrpMgr *vrrp.Manager
  40. bwtestMgr *bwtest.Manager
  41. servicesMgr *services.Manager
  42. certMgr *certs.CertManager
  43. proxyMgr *reverseproxy.Manager
  44. }
  45. func NewExecutor(
  46. authMgr *auth.Manager,
  47. netMgr *network.Manager,
  48. routingMgr *routing.RoutingManager,
  49. dynRoutingMgr *dynamicrouting.Manager,
  50. fwMgr *firewall.FirewallManager,
  51. dhcpMgr *dhcp.Manager,
  52. dhcpServMgr *dhcp.ServerManager,
  53. dnsMgr *dns.Manager,
  54. bridgeMgr *bridge.BridgeManager,
  55. vpnMgr *vpn.Manager,
  56. vrrpMgr *vrrp.Manager,
  57. bwtestMgr *bwtest.Manager,
  58. servicesMgr *services.Manager,
  59. certMgr *certs.CertManager,
  60. proxyMgr *reverseproxy.Manager,
  61. ) *Executor {
  62. return &Executor{
  63. authMgr: authMgr,
  64. netMgr: netMgr,
  65. routingMgr: routingMgr,
  66. dynRoutingMgr: dynRoutingMgr,
  67. fwMgr: fwMgr,
  68. dhcpMgr: dhcpMgr,
  69. dhcpServMgr: dhcpServMgr,
  70. dnsMgr: dnsMgr,
  71. bridgeMgr: bridgeMgr,
  72. vpnMgr: vpnMgr,
  73. vrrpMgr: vrrpMgr,
  74. bwtestMgr: bwtestMgr,
  75. servicesMgr: servicesMgr,
  76. certMgr: certMgr,
  77. proxyMgr: proxyMgr,
  78. }
  79. }
  80. func (e *Executor) Execute(rawCmd string, user *auth.PublicUser) (output string, shouldExit bool) {
  81. // A malformed command must never take down the router or its tunnels.
  82. defer func() {
  83. if recover() != nil {
  84. logging.Log("cli,error", "Command handler panicked")
  85. output = ColorRed + "Internal command error; the router is still running.\n" + ColorReset
  86. shouldExit = false
  87. }
  88. }()
  89. cmd := strings.TrimSpace(rawCmd)
  90. if cmd == "" {
  91. return "", false
  92. }
  93. cmd = strings.TrimPrefix(cmd, "/")
  94. parts, err := splitCommand(cmd)
  95. if err != nil {
  96. return ColorRed + "Syntax error: " + err.Error() + "\n" + ColorReset, false
  97. }
  98. if len(parts) == 0 {
  99. return "", false
  100. }
  101. mainCmd := strings.ToLower(parts[0])
  102. switch mainCmd {
  103. case "quit", "exit":
  104. return ColorYellow + "Connection closed. Goodbye!\n" + ColorReset, true
  105. case "help", "?":
  106. return e.showHelp(), false
  107. case "clear":
  108. return "\033[2J\033[H", false
  109. case "ping":
  110. return e.handlePing(parts[1:]), false
  111. case "traceroute":
  112. return e.handleTraceroute(parts[1:]), false
  113. case "log":
  114. return e.handleLog(parts[1:]), false
  115. case "system":
  116. return e.handleSystem(parts[1:], user), false
  117. case "interface":
  118. return e.handleInterface(parts[1:], user), false
  119. case "ip":
  120. return e.handleIP(parts[1:], user), false
  121. case "routing":
  122. return e.handleRouting(parts[1:], user), false
  123. case "tool":
  124. if user == nil || user.Role == auth.RoleRead {
  125. return ColorRed + "Permission denied: tools require write privileges\n" + ColorReset, false
  126. }
  127. return e.handleTool(parts[1:]), false
  128. case "user":
  129. return e.handleUser(parts[1:], user), false
  130. case "certificate", "certificates":
  131. return e.handleCertificate(parts[1:], user), false
  132. case "reverse-proxy", "reverseproxy":
  133. return e.handleIPReverseProxy(parts[1:], user), false
  134. default:
  135. return fmt.Sprintf(ColorRed+"Unknown command or syntax error: %q. Type 'help' or '?' for command list.\n"+ColorReset, cmd), false
  136. }
  137. }
  138. func parseArgs(parts []string) ([]string, map[string]string) {
  139. var pos []string
  140. params := make(map[string]string)
  141. for _, p := range parts {
  142. if idx := strings.Index(p, "="); idx != -1 {
  143. k := strings.ToLower(p[:idx])
  144. v := p[idx+1:]
  145. params[k] = v
  146. } else {
  147. pos = append(pos, p)
  148. }
  149. }
  150. return pos, params
  151. }
  152. // ==================== LOGGING ====================
  153. func (e *Executor) handleLog(parts []string) string {
  154. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" || parts[0] == "show" {
  155. limit := 50
  156. filter := ""
  157. args := parts
  158. if len(args) > 0 {
  159. args = args[1:]
  160. }
  161. _, params := parseArgs(args)
  162. if lStr, ok := params["limit"]; ok {
  163. if lVal, err := strconv.Atoi(lStr); err == nil && lVal > 0 {
  164. limit = lVal
  165. }
  166. }
  167. if fStr, ok := params["where"]; ok {
  168. filter = fStr
  169. } else if fStr, ok := params["filter"]; ok {
  170. filter = fStr
  171. }
  172. logger := logging.GetLogger()
  173. entries := logger.List(limit, filter)
  174. if len(entries) == 0 {
  175. return ColorYellow + "No log records found.\n" + ColorReset
  176. }
  177. tbl := NewTable("#", "TIME", "TOPICS", "MESSAGE")
  178. for _, logEntry := range entries {
  179. timeStr := logEntry.Time.Format("Jan/02 15:04:05")
  180. tbl.AddRow(
  181. fmt.Sprintf("%d", logEntry.ID),
  182. timeStr,
  183. logEntry.Topics,
  184. logEntry.Message,
  185. )
  186. }
  187. return tbl.String()
  188. }
  189. return ColorRed + "Usage: /log print [limit=N] [where=filter]\n" + ColorReset
  190. }
  191. // ==================== INTERFACES ====================
  192. func (e *Executor) handleInterface(parts []string, user *auth.PublicUser) string {
  193. if len(parts) == 0 {
  194. return e.printAllInterfaces()
  195. }
  196. sub := strings.ToLower(parts[0])
  197. switch sub {
  198. case "print", "pr", "show", "list":
  199. return e.printAllInterfaces()
  200. case "vlan":
  201. return e.handleInterfaceVLAN(parts[1:], user)
  202. case "bridge":
  203. return e.handleInterfaceBridge(parts[1:], user)
  204. case "wireguard":
  205. return e.handleInterfaceWireGuard(parts[1:], user)
  206. case "aerotun":
  207. return e.handleInterfaceAeroTUN(parts[1:], user)
  208. case "eoip":
  209. return e.handleInterfaceEoIP(parts[1:], user)
  210. case "vrrp":
  211. return e.handleInterfaceVRRP(parts[1:], user)
  212. case "pppoe-client":
  213. return e.handleInterfacePPPoE(parts[1:], user)
  214. case "enable":
  215. if len(parts) < 2 {
  216. return ColorRed + "Usage: /interface enable <name>\n" + ColorReset
  217. }
  218. if user.Role == auth.RoleRead {
  219. return ColorRed + "Permission denied: viewer role cannot modify interfaces\n" + ColorReset
  220. }
  221. name := parts[1]
  222. if err := e.netMgr.SetInterfaceState(name, true); err != nil {
  223. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  224. }
  225. return ColorGreen + fmt.Sprintf("Interface %s enabled successfully.\n", name) + ColorReset
  226. case "disable":
  227. if len(parts) < 2 {
  228. return ColorRed + "Usage: /interface disable <name>\n" + ColorReset
  229. }
  230. if user.Role == auth.RoleRead {
  231. return ColorRed + "Permission denied: viewer role cannot modify interfaces\n" + ColorReset
  232. }
  233. name := parts[1]
  234. if err := e.netMgr.SetInterfaceState(name, false); err != nil {
  235. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  236. }
  237. return ColorGreen + fmt.Sprintf("Interface %s disabled successfully.\n", name) + ColorReset
  238. case "set":
  239. if len(parts) < 2 {
  240. return ColorRed + "Usage: /interface set <name> mtu=<val>\n" + ColorReset
  241. }
  242. if user.Role == auth.RoleRead {
  243. return ColorRed + "Permission denied: viewer role cannot modify interfaces\n" + ColorReset
  244. }
  245. name := parts[1]
  246. _, params := parseArgs(parts[2:])
  247. if mtuStr, ok := params["mtu"]; ok {
  248. mtu, err := strconv.Atoi(mtuStr)
  249. if err != nil || mtu < 68 || mtu > 9000 {
  250. return ColorRed + "Invalid MTU value (68-9000)\n" + ColorReset
  251. }
  252. if err := e.netMgr.SetInterfaceMTU(name, mtu); err != nil {
  253. return ColorRed + "Error setting MTU: " + err.Error() + ColorReset + "\n"
  254. }
  255. return ColorGreen + fmt.Sprintf("Interface %s MTU updated to %d.\n", name, mtu) + ColorReset
  256. }
  257. return ColorRed + "Usage: /interface set <name> mtu=<val>\n" + ColorReset
  258. default:
  259. return ColorRed + "Unknown interface subcommand: " + sub + "\n" + ColorReset
  260. }
  261. }
  262. func (e *Executor) printAllInterfaces() string {
  263. ifaces, err := e.netMgr.ListInterfaces()
  264. if err != nil {
  265. return ColorRed + "Error listing interfaces: " + err.Error() + ColorReset + "\n"
  266. }
  267. tbl := NewTable("FLAGS", "NAME", "TYPE", "STATE", "MTU", "MAC ADDRESS", "IP ADDRESSES")
  268. for _, iface := range ifaces {
  269. isUp := strings.ToUpper(iface.State) == "UP"
  270. flags := ""
  271. if isUp {
  272. flags += "R"
  273. } else {
  274. flags += "X"
  275. }
  276. stateStr := ColorGreen + "UP" + ColorReset
  277. if !isUp {
  278. stateStr = ColorRed + "DOWN" + ColorReset
  279. }
  280. allIPs := append([]string{}, iface.IPv4...)
  281. allIPs = append(allIPs, iface.IPv6...)
  282. ips := strings.Join(allIPs, ", ")
  283. if ips == "" {
  284. ips = "-"
  285. }
  286. tbl.AddRow(flags, iface.Name, string(iface.Type), stateStr, fmt.Sprintf("%d", iface.MTU), iface.MAC, ips)
  287. }
  288. return tbl.String()
  289. }
  290. // --- VLAN ---
  291. func (e *Executor) handleInterfaceVLAN(parts []string, user *auth.PublicUser) string {
  292. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  293. vlans, err := e.netMgr.ListVLANs()
  294. if err != nil {
  295. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  296. }
  297. tbl := NewTable("#", "NAME", "VLAN-ID", "INTERFACE", "MTU", "STATE")
  298. for idx, v := range vlans {
  299. isUp := strings.ToUpper(v.State) == "UP"
  300. st := ColorGreen + "UP" + ColorReset
  301. if !isUp {
  302. st = ColorRed + "DOWN" + ColorReset
  303. }
  304. tbl.AddRow(fmt.Sprintf("%d", idx), v.Name, fmt.Sprintf("%d", v.VlanID), v.ParentName, fmt.Sprintf("%d", v.MTU), st)
  305. }
  306. return tbl.String()
  307. }
  308. sub := strings.ToLower(parts[0])
  309. if user.Role == auth.RoleRead {
  310. return ColorRed + "Permission denied: viewer role cannot modify VLANs\n" + ColorReset
  311. }
  312. switch sub {
  313. case "add":
  314. _, params := parseArgs(parts[1:])
  315. name := params["name"]
  316. parent := params["interface"]
  317. vlanIDStr := params["vlan-id"]
  318. if vlanIDStr == "" {
  319. vlanIDStr = params["vlan_id"]
  320. }
  321. vlanID, err := strconv.Atoi(vlanIDStr)
  322. if err != nil || vlanID < 1 || vlanID > 4094 {
  323. return ColorRed + "Invalid vlan-id (1-4094)\n" + ColorReset
  324. }
  325. if parent == "" {
  326. return ColorRed + "Missing parent interface (interface=...)\n" + ColorReset
  327. }
  328. if name == "" {
  329. name = fmt.Sprintf("%s.%d", parent, vlanID)
  330. }
  331. req := network.CreateVLANRequest{
  332. ParentName: parent,
  333. VlanID: vlanID,
  334. Name: name,
  335. AutoUp: true,
  336. }
  337. if _, err := e.netMgr.CreateVLAN(req); err != nil {
  338. return ColorRed + "Error creating VLAN: " + err.Error() + ColorReset + "\n"
  339. }
  340. return ColorGreen + fmt.Sprintf("VLAN interface %s created.\n", name) + ColorReset
  341. case "remove", "rem", "delete":
  342. pos, _ := parseArgs(parts[1:])
  343. if len(pos) == 0 {
  344. return ColorRed + "Usage: /interface vlan remove <name>\n" + ColorReset
  345. }
  346. name := pos[0]
  347. if err := e.netMgr.DeleteVLAN(name); err != nil {
  348. return ColorRed + "Error deleting VLAN: " + err.Error() + ColorReset + "\n"
  349. }
  350. return ColorGreen + fmt.Sprintf("VLAN %s removed.\n", name) + ColorReset
  351. default:
  352. return ColorRed + "Usage: /interface vlan [print|add|remove]\n" + ColorReset
  353. }
  354. }
  355. // --- Bridge ---
  356. func (e *Executor) handleInterfaceBridge(parts []string, user *auth.PublicUser) string {
  357. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  358. if e.bridgeMgr == nil {
  359. return ColorRed + "Bridge manager not available.\n" + ColorReset
  360. }
  361. bridges := e.bridgeMgr.ListBridges()
  362. tbl := NewTable("FLAGS", "NAME", "MTU", "MAC-ADDRESS", "RUNNING", "PORTS")
  363. for _, b := range bridges {
  364. flags := ""
  365. if b.Disabled {
  366. flags += "X"
  367. } else {
  368. flags += "R"
  369. }
  370. tbl.AddRow(flags, b.Name, fmt.Sprintf("%d", b.MTU), b.MACAddress, fmt.Sprintf("%t", b.Running), fmt.Sprintf("%d", b.PortsCount))
  371. }
  372. return tbl.String()
  373. }
  374. sub := strings.ToLower(parts[0])
  375. if sub == "port" {
  376. return e.handleBridgePort(parts[1:], user)
  377. }
  378. if user.Role == auth.RoleRead {
  379. return ColorRed + "Permission denied: viewer role cannot modify bridge\n" + ColorReset
  380. }
  381. switch sub {
  382. case "add":
  383. _, params := parseArgs(parts[1:])
  384. name := params["name"]
  385. if name == "" {
  386. return ColorRed + "Usage: /interface bridge add name=<name>\n" + ColorReset
  387. }
  388. b := bridge.BridgeInstance{
  389. Name: name,
  390. Disabled: false,
  391. }
  392. if err := e.bridgeMgr.SaveBridge(b); err != nil {
  393. return ColorRed + "Error adding bridge: " + err.Error() + ColorReset + "\n"
  394. }
  395. return ColorGreen + fmt.Sprintf("Bridge %s added.\n", name) + ColorReset
  396. case "remove", "rem", "delete":
  397. pos, _ := parseArgs(parts[1:])
  398. if len(pos) == 0 {
  399. return ColorRed + "Usage: /interface bridge remove <id|name>\n" + ColorReset
  400. }
  401. if err := e.bridgeMgr.DeleteBridge(pos[0]); err != nil {
  402. return ColorRed + "Error deleting bridge: " + err.Error() + ColorReset + "\n"
  403. }
  404. return ColorGreen + fmt.Sprintf("Bridge %s removed.\n", pos[0]) + ColorReset
  405. case "enable":
  406. pos, _ := parseArgs(parts[1:])
  407. if len(pos) == 0 {
  408. return ColorRed + "Usage: /interface bridge enable <id|name>\n" + ColorReset
  409. }
  410. if err := e.bridgeMgr.SetBridgeState(pos[0], false); err != nil {
  411. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  412. }
  413. return ColorGreen + fmt.Sprintf("Bridge %s enabled.\n", pos[0]) + ColorReset
  414. case "disable":
  415. pos, _ := parseArgs(parts[1:])
  416. if len(pos) == 0 {
  417. return ColorRed + "Usage: /interface bridge disable <id|name>\n" + ColorReset
  418. }
  419. if err := e.bridgeMgr.SetBridgeState(pos[0], true); err != nil {
  420. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  421. }
  422. return ColorGreen + fmt.Sprintf("Bridge %s disabled.\n", pos[0]) + ColorReset
  423. default:
  424. return ColorRed + "Usage: /interface bridge [print|add|remove|enable|disable|port]\n" + ColorReset
  425. }
  426. }
  427. func (e *Executor) handleBridgePort(parts []string, user *auth.PublicUser) string {
  428. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  429. ports := e.bridgeMgr.ListPorts()
  430. tbl := NewTable("FLAGS", "INTERFACE", "BRIDGE", "STATUS", "PVID")
  431. for _, p := range ports {
  432. flags := ""
  433. if p.Disabled {
  434. flags += "X"
  435. } else {
  436. flags += "H"
  437. }
  438. tbl.AddRow(flags, p.Interface, p.Bridge, p.Status, fmt.Sprintf("%d", p.PVID))
  439. }
  440. return tbl.String()
  441. }
  442. if user.Role == auth.RoleRead {
  443. return ColorRed + "Permission denied: viewer role cannot modify bridge ports\n" + ColorReset
  444. }
  445. sub := strings.ToLower(parts[0])
  446. switch sub {
  447. case "add":
  448. _, params := parseArgs(parts[1:])
  449. br := params["bridge"]
  450. iface := params["interface"]
  451. if br == "" || iface == "" {
  452. return ColorRed + "Usage: /interface bridge port add bridge=<bridge> interface=<iface>\n" + ColorReset
  453. }
  454. port := bridge.BridgePort{
  455. Bridge: br,
  456. Interface: iface,
  457. Disabled: false,
  458. }
  459. if err := e.bridgeMgr.SavePort(port); err != nil {
  460. return ColorRed + "Error adding bridge port: " + err.Error() + ColorReset + "\n"
  461. }
  462. return ColorGreen + fmt.Sprintf("Port %s attached to bridge %s.\n", iface, br) + ColorReset
  463. case "remove", "rem", "delete":
  464. pos, _ := parseArgs(parts[1:])
  465. if len(pos) == 0 {
  466. return ColorRed + "Usage: /interface bridge port remove <id>\n" + ColorReset
  467. }
  468. if err := e.bridgeMgr.DeletePort(pos[0]); err != nil {
  469. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  470. }
  471. return ColorGreen + fmt.Sprintf("Bridge port %s removed.\n", pos[0]) + ColorReset
  472. default:
  473. return ColorRed + "Usage: /interface bridge port [print|add|remove]\n" + ColorReset
  474. }
  475. }
  476. // --- WireGuard ---
  477. func (e *Executor) handleInterfaceWireGuard(parts []string, user *auth.PublicUser) string {
  478. if e.vpnMgr == nil || e.vpnMgr.WireGuard == nil {
  479. return ColorRed + "WireGuard subsystem not available.\n" + ColorReset
  480. }
  481. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  482. ifaces := e.vpnMgr.WireGuard.ListInterfaces()
  483. tbl := NewTable("FLAGS", "NAME", "LISTEN-PORT", "PUBLIC-KEY", "STATUS")
  484. for _, w := range ifaces {
  485. flags := ""
  486. if w.Disabled {
  487. flags += "X"
  488. } else {
  489. flags += "R"
  490. }
  491. st := "running"
  492. if !w.Running {
  493. st = "stopped"
  494. }
  495. tbl.AddRow(flags, w.Name, fmt.Sprintf("%d", w.ListenPort), w.PublicKey, st)
  496. }
  497. return tbl.String()
  498. }
  499. sub := strings.ToLower(parts[0])
  500. if sub == "peers" || sub == "peer" {
  501. return e.handleWireGuardPeers(parts[1:], user)
  502. }
  503. if user.Role == auth.RoleRead {
  504. return ColorRed + "Permission denied: viewer role cannot modify wireguard\n" + ColorReset
  505. }
  506. switch sub {
  507. case "add":
  508. _, params := parseArgs(parts[1:])
  509. name := params["name"]
  510. portStr := params["listen-port"]
  511. privKey := params["private-key"]
  512. port := 51820
  513. if portStr != "" {
  514. if pVal, err := strconv.Atoi(portStr); err == nil && pVal > 0 {
  515. port = pVal
  516. }
  517. }
  518. if name == "" {
  519. return ColorRed + "Usage: /interface wireguard add name=<name> [listen-port=<port>]\n" + ColorReset
  520. }
  521. req := vpn.SaveWireGuardInterfaceRequest{
  522. WireGuardInterface: vpn.WireGuardInterface{
  523. Name: name,
  524. ListenPort: port,
  525. PrivateKey: privKey,
  526. Disabled: false,
  527. },
  528. }
  529. if err := e.vpnMgr.WireGuard.SaveInterface(req); err != nil {
  530. return ColorRed + "Error adding WireGuard interface: " + err.Error() + ColorReset + "\n"
  531. }
  532. return ColorGreen + fmt.Sprintf("WireGuard interface %s added.\n", name) + ColorReset
  533. case "remove", "rem", "delete":
  534. pos, _ := parseArgs(parts[1:])
  535. if len(pos) == 0 {
  536. return ColorRed + "Usage: /interface wireguard remove <name>\n" + ColorReset
  537. }
  538. if err := e.vpnMgr.WireGuard.DeleteInterface(pos[0]); err != nil {
  539. return ColorRed + "Error deleting WireGuard interface: " + err.Error() + ColorReset + "\n"
  540. }
  541. return ColorGreen + fmt.Sprintf("WireGuard interface %s removed.\n", pos[0]) + ColorReset
  542. case "enable":
  543. pos, _ := parseArgs(parts[1:])
  544. if len(pos) == 0 {
  545. return ColorRed + "Usage: /interface wireguard enable <name>\n" + ColorReset
  546. }
  547. if err := e.vpnMgr.WireGuard.SetInterfaceEnabled(pos[0], true); err != nil {
  548. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  549. }
  550. return ColorGreen + fmt.Sprintf("WireGuard interface %s enabled.\n", pos[0]) + ColorReset
  551. case "disable":
  552. pos, _ := parseArgs(parts[1:])
  553. if len(pos) == 0 {
  554. return ColorRed + "Usage: /interface wireguard disable <name>\n" + ColorReset
  555. }
  556. if err := e.vpnMgr.WireGuard.SetInterfaceEnabled(pos[0], false); err != nil {
  557. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  558. }
  559. return ColorGreen + fmt.Sprintf("WireGuard interface %s disabled.\n", pos[0]) + ColorReset
  560. default:
  561. return ColorRed + "Usage: /interface wireguard [print|add|remove|enable|disable|peers]\n" + ColorReset
  562. }
  563. }
  564. func (e *Executor) handleWireGuardPeers(parts []string, user *auth.PublicUser) string {
  565. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  566. _, params := parseArgs(parts)
  567. peers := e.vpnMgr.WireGuard.ListPeers(params["interface"])
  568. tbl := NewTable("INTERFACE", "PUBLIC-KEY", "ENDPOINT", "ALLOWED-IPS", "LAST-HANDSHAKE")
  569. for _, p := range peers {
  570. hs := "-"
  571. if p.LatestHandshake != "" {
  572. hs = p.LatestHandshake
  573. }
  574. ep := p.EndpointAddress
  575. if p.EndpointPort > 0 {
  576. ep = fmt.Sprintf("%s:%d", p.EndpointAddress, p.EndpointPort)
  577. }
  578. tbl.AddRow(p.Interface, p.PublicKey, ep, strings.Join(p.AllowedIPs, ","), hs)
  579. }
  580. return tbl.String()
  581. }
  582. if user.Role == auth.RoleRead {
  583. return ColorRed + "Permission denied: viewer role cannot modify wireguard peers\n" + ColorReset
  584. }
  585. sub := strings.ToLower(parts[0])
  586. switch sub {
  587. case "add":
  588. _, params := parseArgs(parts[1:])
  589. iface := params["interface"]
  590. pubKey := params["public-key"]
  591. endpoint := params["endpoint"]
  592. allowed := params["allowed-address"]
  593. if allowed == "" {
  594. allowed = params["allowed-ips"]
  595. }
  596. if iface == "" || pubKey == "" {
  597. return ColorRed + "Usage: /interface wireguard peers add interface=<iface> public-key=<key> [endpoint=host:port] [allowed-address=cidr]\n" + ColorReset
  598. }
  599. var allowedList []string
  600. if allowed != "" {
  601. allowedList = strings.Split(allowed, ",")
  602. }
  603. epHost := endpoint
  604. epPort := 0
  605. if strings.Contains(endpoint, ":") {
  606. parts := strings.Split(endpoint, ":")
  607. epHost = parts[0]
  608. if p, err := strconv.Atoi(parts[1]); err == nil {
  609. epPort = p
  610. }
  611. }
  612. req := vpn.SaveWireGuardPeerRequest{
  613. WireGuardPeer: vpn.WireGuardPeer{
  614. Interface: iface,
  615. PublicKey: pubKey,
  616. EndpointAddress: epHost,
  617. EndpointPort: epPort,
  618. AllowedIPs: allowedList,
  619. Disabled: false,
  620. },
  621. }
  622. if err := e.vpnMgr.WireGuard.SavePeer(req); err != nil {
  623. return ColorRed + "Error saving WireGuard peer: " + err.Error() + ColorReset + "\n"
  624. }
  625. return ColorGreen + "WireGuard peer added successfully.\n" + ColorReset
  626. case "remove", "rem", "delete":
  627. _, params := parseArgs(parts[1:])
  628. iface := params["interface"]
  629. pubKey := params["public-key"]
  630. if iface == "" || pubKey == "" {
  631. return ColorRed + "Usage: /interface wireguard peers remove interface=<iface> public-key=<key>\n" + ColorReset
  632. }
  633. if err := e.vpnMgr.WireGuard.DeletePeer(iface, pubKey); err != nil {
  634. return ColorRed + "Error deleting WireGuard peer: " + err.Error() + ColorReset + "\n"
  635. }
  636. return ColorGreen + "WireGuard peer removed successfully.\n" + ColorReset
  637. default:
  638. return ColorRed + "Usage: /interface wireguard peers [print|add|remove]\n" + ColorReset
  639. }
  640. }
  641. // --- EoIP ---
  642. func (e *Executor) handleInterfaceEoIP(parts []string, user *auth.PublicUser) string {
  643. if e.vpnMgr == nil || e.vpnMgr.EoIP == nil {
  644. return ColorRed + "EoIP subsystem not available.\n" + ColorReset
  645. }
  646. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  647. tunnels := e.vpnMgr.EoIP.List()
  648. tbl := NewTable("FLAGS", "NAME", "REMOTE-ADDRESS", "TUNNEL-ID", "STATUS")
  649. for _, t := range tunnels {
  650. flags := ""
  651. if t.Disabled {
  652. flags += "X"
  653. } else {
  654. flags += "R"
  655. }
  656. st := "stopped"
  657. if t.Running {
  658. st = "running"
  659. }
  660. tbl.AddRow(flags, t.Name, t.RemoteAddress, fmt.Sprintf("%d", t.TunnelID), st)
  661. }
  662. return tbl.String()
  663. }
  664. if user.Role == auth.RoleRead {
  665. return ColorRed + "Permission denied: viewer role cannot modify EoIP\n" + ColorReset
  666. }
  667. sub := strings.ToLower(parts[0])
  668. switch sub {
  669. case "add":
  670. _, params := parseArgs(parts[1:])
  671. name := params["name"]
  672. remote := params["remote-address"]
  673. tidStr := params["tunnel-id"]
  674. tid, err := strconv.Atoi(tidStr)
  675. if err != nil || tid < 1 || tid > 65535 {
  676. return ColorRed + "Invalid tunnel-id (1-65535)\n" + ColorReset
  677. }
  678. if name == "" || remote == "" {
  679. return ColorRed + "Usage: /interface eoip add name=<name> remote-address=<ip> tunnel-id=<id>\n" + ColorReset
  680. }
  681. inst := &vpn.EoIPInstance{
  682. Name: name,
  683. RemoteAddress: remote,
  684. TunnelID: tid,
  685. Disabled: false,
  686. }
  687. if err := e.vpnMgr.EoIP.Save(inst); err != nil {
  688. return ColorRed + "Error adding EoIP tunnel: " + err.Error() + ColorReset + "\n"
  689. }
  690. return ColorGreen + fmt.Sprintf("EoIP tunnel %s added.\n", name) + ColorReset
  691. case "remove", "rem", "delete":
  692. pos, _ := parseArgs(parts[1:])
  693. if len(pos) == 0 {
  694. return ColorRed + "Usage: /interface eoip remove <id>\n" + ColorReset
  695. }
  696. if err := e.vpnMgr.EoIP.Delete(pos[0]); err != nil {
  697. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  698. }
  699. return ColorGreen + fmt.Sprintf("EoIP tunnel %s removed.\n", pos[0]) + ColorReset
  700. case "enable":
  701. pos, _ := parseArgs(parts[1:])
  702. if len(pos) == 0 {
  703. return ColorRed + "Usage: /interface eoip enable <id>\n" + ColorReset
  704. }
  705. if err := e.vpnMgr.EoIP.SetState(pos[0], false); err != nil {
  706. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  707. }
  708. return ColorGreen + fmt.Sprintf("EoIP tunnel %s enabled.\n", pos[0]) + ColorReset
  709. case "disable":
  710. pos, _ := parseArgs(parts[1:])
  711. if len(pos) == 0 {
  712. return ColorRed + "Usage: /interface eoip disable <id>\n" + ColorReset
  713. }
  714. if err := e.vpnMgr.EoIP.SetState(pos[0], true); err != nil {
  715. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  716. }
  717. return ColorGreen + fmt.Sprintf("EoIP tunnel %s disabled.\n", pos[0]) + ColorReset
  718. default:
  719. return ColorRed + "Usage: /interface eoip [print|add|remove|enable|disable]\n" + ColorReset
  720. }
  721. }
  722. // --- VRRP ---
  723. func (e *Executor) handleInterfaceVRRP(parts []string, user *auth.PublicUser) string {
  724. if e.vrrpMgr == nil {
  725. return ColorRed + "VRRP subsystem not available.\n" + ColorReset
  726. }
  727. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  728. instances := e.vrrpMgr.List()
  729. tbl := NewTable("FLAGS", "NAME", "INTERFACE", "VRID", "PRIORITY", "STATE", "VIP")
  730. for _, inst := range instances {
  731. flags := ""
  732. if inst.Disabled {
  733. flags += "X"
  734. } else {
  735. flags += "R"
  736. }
  737. tbl.AddRow(flags, inst.Name, inst.Interface, fmt.Sprintf("%d", inst.VRID), fmt.Sprintf("%d", inst.Priority), inst.State, strings.Join(inst.VirtualIPs, ","))
  738. }
  739. return tbl.String()
  740. }
  741. if user.Role == auth.RoleRead {
  742. return ColorRed + "Permission denied: viewer role cannot modify VRRP\n" + ColorReset
  743. }
  744. sub := strings.ToLower(parts[0])
  745. switch sub {
  746. case "add":
  747. _, params := parseArgs(parts[1:])
  748. name := params["name"]
  749. iface := params["interface"]
  750. vridStr := params["vrid"]
  751. prioStr := params["priority"]
  752. vip := params["vip"]
  753. if vip == "" {
  754. vip = params["virtual-ip"]
  755. }
  756. vrid, err := strconv.Atoi(vridStr)
  757. if err != nil || vrid < 1 || vrid > 255 {
  758. return ColorRed + "Invalid vrid (1-255)\n" + ColorReset
  759. }
  760. prio := 100
  761. if prioStr != "" {
  762. if pVal, err := strconv.Atoi(prioStr); err == nil && pVal >= 1 && pVal <= 254 {
  763. prio = pVal
  764. }
  765. }
  766. if name == "" || iface == "" || vip == "" {
  767. return ColorRed + "Usage: /interface vrrp add name=<name> interface=<iface> vrid=<id> vip=<ip/cidr> [priority=100]\n" + ColorReset
  768. }
  769. inst := vrrp.VRRPInstance{
  770. Name: name,
  771. Interface: iface,
  772. VRID: vrid,
  773. Priority: prio,
  774. VirtualIPs: []string{vip},
  775. Disabled: false,
  776. }
  777. if _, err := e.vrrpMgr.Save(inst); err != nil {
  778. return ColorRed + "Error adding VRRP: " + err.Error() + ColorReset + "\n"
  779. }
  780. return ColorGreen + fmt.Sprintf("VRRP instance %s added.\n", name) + ColorReset
  781. case "remove", "rem", "delete":
  782. pos, _ := parseArgs(parts[1:])
  783. if len(pos) == 0 {
  784. return ColorRed + "Usage: /interface vrrp remove <id>\n" + ColorReset
  785. }
  786. if err := e.vrrpMgr.Delete(pos[0]); err != nil {
  787. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  788. }
  789. return ColorGreen + fmt.Sprintf("VRRP instance %s removed.\n", pos[0]) + ColorReset
  790. case "enable":
  791. pos, _ := parseArgs(parts[1:])
  792. if len(pos) == 0 {
  793. return ColorRed + "Usage: /interface vrrp enable <id>\n" + ColorReset
  794. }
  795. if err := e.vrrpMgr.ToggleState(pos[0], false); err != nil {
  796. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  797. }
  798. return ColorGreen + fmt.Sprintf("VRRP instance %s enabled.\n", pos[0]) + ColorReset
  799. case "disable":
  800. pos, _ := parseArgs(parts[1:])
  801. if len(pos) == 0 {
  802. return ColorRed + "Usage: /interface vrrp disable <id>\n" + ColorReset
  803. }
  804. if err := e.vrrpMgr.ToggleState(pos[0], true); err != nil {
  805. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  806. }
  807. return ColorGreen + fmt.Sprintf("VRRP instance %s disabled.\n", pos[0]) + ColorReset
  808. default:
  809. return ColorRed + "Usage: /interface vrrp [print|add|remove|enable|disable]\n" + ColorReset
  810. }
  811. }
  812. // --- PPPoE Client ---
  813. func (e *Executor) handleInterfacePPPoE(parts []string, user *auth.PublicUser) string {
  814. if e.vpnMgr == nil || e.vpnMgr.PPPoE == nil {
  815. return ColorRed + "PPPoE subsystem not available.\n" + ColorReset
  816. }
  817. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  818. clients := e.vpnMgr.PPPoE.List()
  819. tbl := NewTable("FLAGS", "NAME", "INTERFACE", "USER", "STATUS", "IP")
  820. for _, c := range clients {
  821. flags := ""
  822. if c.Disabled {
  823. flags += "X"
  824. } else {
  825. flags += "R"
  826. }
  827. st := "stopped"
  828. if c.Running {
  829. st = "running"
  830. }
  831. tbl.AddRow(flags, c.Name, c.Interface, c.Username, st, c.AssignedIP)
  832. }
  833. return tbl.String()
  834. }
  835. if user.Role == auth.RoleRead {
  836. return ColorRed + "Permission denied: viewer role cannot modify PPPoE\n" + ColorReset
  837. }
  838. sub := strings.ToLower(parts[0])
  839. switch sub {
  840. case "add":
  841. _, params := parseArgs(parts[1:])
  842. name := params["name"]
  843. iface := params["interface"]
  844. usr := params["user"]
  845. pwd := params["password"]
  846. if name == "" || iface == "" || usr == "" {
  847. return ColorRed + "Usage: /interface pppoe-client add name=<name> interface=<iface> user=<usr> password=<pwd>\n" + ColorReset
  848. }
  849. inst := vpn.PPPoEInstance{
  850. Name: name,
  851. Interface: iface,
  852. Username: usr,
  853. Password: pwd,
  854. Disabled: false,
  855. }
  856. if err := e.vpnMgr.PPPoE.Save(inst); err != nil {
  857. return ColorRed + "Error adding PPPoE client: " + err.Error() + ColorReset + "\n"
  858. }
  859. return ColorGreen + fmt.Sprintf("PPPoE client %s added.\n", name) + ColorReset
  860. case "remove", "rem", "delete":
  861. pos, _ := parseArgs(parts[1:])
  862. if len(pos) == 0 {
  863. return ColorRed + "Usage: /interface pppoe-client remove <name>\n" + ColorReset
  864. }
  865. if err := e.vpnMgr.PPPoE.Delete(pos[0]); err != nil {
  866. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  867. }
  868. return ColorGreen + fmt.Sprintf("PPPoE client %s removed.\n", pos[0]) + ColorReset
  869. case "connect":
  870. pos, _ := parseArgs(parts[1:])
  871. if len(pos) == 0 {
  872. return ColorRed + "Usage: /interface pppoe-client connect <name>\n" + ColorReset
  873. }
  874. if err := e.vpnMgr.PPPoE.Connect(pos[0]); err != nil {
  875. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  876. }
  877. return ColorGreen + fmt.Sprintf("Connecting PPPoE client %s...\n", pos[0]) + ColorReset
  878. case "disconnect":
  879. pos, _ := parseArgs(parts[1:])
  880. if len(pos) == 0 {
  881. return ColorRed + "Usage: /interface pppoe-client disconnect <name>\n" + ColorReset
  882. }
  883. if err := e.vpnMgr.PPPoE.Disconnect(pos[0]); err != nil {
  884. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  885. }
  886. return ColorGreen + fmt.Sprintf("Disconnected PPPoE client %s.\n", pos[0]) + ColorReset
  887. default:
  888. return ColorRed + "Usage: /interface pppoe-client [print|add|remove|connect|disconnect]\n" + ColorReset
  889. }
  890. }
  891. // ==================== IP SUBSYSTEM ====================
  892. func (e *Executor) handleIP(parts []string, user *auth.PublicUser) string {
  893. if len(parts) == 0 {
  894. return ColorYellow + "IP subsystem. Available: address, route, dhcp-client, dhcp-server, dns, firewall, service\n" + ColorReset
  895. }
  896. sub := strings.ToLower(parts[0])
  897. switch sub {
  898. case "address", "addr":
  899. return e.handleIPAddress(parts[1:], user)
  900. case "route":
  901. return e.handleIPRoute(parts[1:], user)
  902. case "dhcp-client":
  903. return e.handleIPDHCPClient(parts[1:], user)
  904. case "dhcp-server":
  905. return e.handleIPDHCPServer(parts[1:], user)
  906. case "dns":
  907. return e.handleIPDNS(parts[1:], user)
  908. case "firewall":
  909. return e.handleIPFirewall(parts[1:], user)
  910. case "service", "services":
  911. return e.handleIPService(parts[1:], user)
  912. case "reverse-proxy", "reverseproxy", "proxy":
  913. return e.handleIPReverseProxy(parts[1:], user)
  914. default:
  915. return ColorRed + "Unknown IP command: " + sub + "\n" + ColorReset
  916. }
  917. }
  918. // --- IP Address ---
  919. func (e *Executor) handleIPAddress(parts []string, user *auth.PublicUser) string {
  920. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  921. ifaces, err := e.netMgr.ListInterfaces()
  922. if err != nil {
  923. return ColorRed + "Error listing IP addresses: " + err.Error() + ColorReset + "\n"
  924. }
  925. tbl := NewTable("#", "ADDRESS", "NETWORK", "INTERFACE", "STATUS")
  926. idx := 0
  927. for _, iface := range ifaces {
  928. for _, cidr := range append(iface.IPv4, iface.IPv6...) {
  929. ip, ipNet, parseErr := net.ParseCIDR(cidr)
  930. netStr := "-"
  931. if parseErr == nil && ipNet != nil {
  932. netStr = ipNet.IP.String()
  933. } else if ip != nil {
  934. netStr = ip.String()
  935. }
  936. tbl.AddRow(fmt.Sprintf("%d", idx), cidr, netStr, iface.Name, "enabled")
  937. idx++
  938. }
  939. }
  940. return tbl.String()
  941. }
  942. if user.Role == auth.RoleRead {
  943. return ColorRed + "Permission denied: viewer role cannot modify IP addresses\n" + ColorReset
  944. }
  945. sub := strings.ToLower(parts[0])
  946. switch sub {
  947. case "add":
  948. _, params := parseArgs(parts[1:])
  949. addr := params["address"]
  950. iface := params["interface"]
  951. if addr == "" || iface == "" {
  952. return ColorRed + "Usage: /ip address add address=<ip/mask> interface=<iface>\n" + ColorReset
  953. }
  954. if err := e.netMgr.AddInterfaceIP(iface, addr); err != nil {
  955. return ColorRed + "Error adding IP address: " + err.Error() + ColorReset + "\n"
  956. }
  957. return ColorGreen + fmt.Sprintf("IP address %s added to interface %s.\n", addr, iface) + ColorReset
  958. case "remove", "rem", "delete":
  959. _, params := parseArgs(parts[1:])
  960. addr := params["address"]
  961. iface := params["interface"]
  962. if addr == "" || iface == "" {
  963. return ColorRed + "Usage: /ip address remove address=<ip/mask> interface=<iface>\n" + ColorReset
  964. }
  965. if err := e.netMgr.DeleteInterfaceIP(iface, addr); err != nil {
  966. return ColorRed + "Error removing IP address: " + err.Error() + ColorReset + "\n"
  967. }
  968. return ColorGreen + fmt.Sprintf("IP address %s removed from interface %s.\n", addr, iface) + ColorReset
  969. default:
  970. return ColorRed + "Usage: /ip address [print|add|remove]\n" + ColorReset
  971. }
  972. }
  973. // --- IP Route ---
  974. func (e *Executor) handleIPRoute(parts []string, user *auth.PublicUser) string {
  975. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  976. routes := e.routingMgr.ListRoutes()
  977. tbl := NewTable("FLAGS", "DST-ADDRESS", "GATEWAY", "DISTANCE", "TABLE", "STATUS")
  978. for _, r := range routes {
  979. flags := ""
  980. if r.Disabled {
  981. flags += "X"
  982. } else {
  983. flags += "A"
  984. }
  985. if r.Static {
  986. flags += "S"
  987. } else {
  988. flags += "D"
  989. }
  990. st := "active"
  991. if r.Disabled {
  992. st = "disabled"
  993. }
  994. tbl.AddRow(flags, r.DstAddress, r.Gateway, fmt.Sprintf("%d", r.Distance), r.RoutingTable, st)
  995. }
  996. return tbl.String()
  997. }
  998. sub := strings.ToLower(parts[0])
  999. switch sub {
  1000. case "rule":
  1001. rules := e.routingMgr.ListPolicyRules()
  1002. tbl := NewTable("#", "SRC", "DST", "TABLE", "PRIORITY")
  1003. for idx, r := range rules {
  1004. tbl.AddRow(fmt.Sprintf("%d", idx), r.SrcAddress, r.DstAddress, r.Table, fmt.Sprintf("%d", r.Priority))
  1005. }
  1006. return tbl.String()
  1007. case "table":
  1008. tables := e.routingMgr.ListTables()
  1009. tbl := NewTable("#", "NAME", "TABLE-ID", "COMMENT")
  1010. for idx, t := range tables {
  1011. tbl.AddRow(fmt.Sprintf("%d", idx), t.Name, fmt.Sprintf("%d", t.TableID), t.Comment)
  1012. }
  1013. return tbl.String()
  1014. }
  1015. if user.Role == auth.RoleRead {
  1016. return ColorRed + "Permission denied: viewer role cannot modify routes\n" + ColorReset
  1017. }
  1018. switch sub {
  1019. case "add":
  1020. _, params := parseArgs(parts[1:])
  1021. dst := params["dst-address"]
  1022. if dst == "" {
  1023. dst = params["dst"]
  1024. }
  1025. gw := params["gateway"]
  1026. table := params["routing-table"]
  1027. if table == "" {
  1028. table = "main"
  1029. }
  1030. dist := 1
  1031. if dStr, ok := params["distance"]; ok {
  1032. if dVal, err := strconv.Atoi(dStr); err == nil && dVal >= 1 {
  1033. dist = dVal
  1034. }
  1035. }
  1036. if dst == "" || gw == "" {
  1037. return ColorRed + "Usage: /ip route add dst-address=<cidr> gateway=<ip> [distance=1] [routing-table=main]\n" + ColorReset
  1038. }
  1039. r := routing.RouteItem{
  1040. DstAddress: dst,
  1041. Gateway: gw,
  1042. Distance: dist,
  1043. RoutingTable: table,
  1044. Static: true,
  1045. Disabled: false,
  1046. }
  1047. if err := e.routingMgr.SaveRoute(r); err != nil {
  1048. return ColorRed + "Error adding route: " + err.Error() + ColorReset + "\n"
  1049. }
  1050. return ColorGreen + fmt.Sprintf("Route to %s via %s added.\n", dst, gw) + ColorReset
  1051. case "remove", "rem", "delete":
  1052. pos, _ := parseArgs(parts[1:])
  1053. if len(pos) == 0 {
  1054. return ColorRed + "Usage: /ip route remove <id>\n" + ColorReset
  1055. }
  1056. if err := e.routingMgr.DeleteRoute(pos[0]); err != nil {
  1057. return ColorRed + "Error removing route: " + err.Error() + ColorReset + "\n"
  1058. }
  1059. return ColorGreen + fmt.Sprintf("Route %s removed.\n", pos[0]) + ColorReset
  1060. case "enable":
  1061. pos, _ := parseArgs(parts[1:])
  1062. if len(pos) == 0 {
  1063. return ColorRed + "Usage: /ip route enable <id>\n" + ColorReset
  1064. }
  1065. if err := e.routingMgr.SetRouteState(pos[0], false); err != nil {
  1066. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1067. }
  1068. return ColorGreen + fmt.Sprintf("Route %s enabled.\n", pos[0]) + ColorReset
  1069. case "disable":
  1070. pos, _ := parseArgs(parts[1:])
  1071. if len(pos) == 0 {
  1072. return ColorRed + "Usage: /ip route disable <id>\n" + ColorReset
  1073. }
  1074. if err := e.routingMgr.SetRouteState(pos[0], true); err != nil {
  1075. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1076. }
  1077. return ColorGreen + fmt.Sprintf("Route %s disabled.\n", pos[0]) + ColorReset
  1078. default:
  1079. return ColorRed + "Usage: /ip route [print|add|remove|enable|disable|rule|table]\n" + ColorReset
  1080. }
  1081. }
  1082. // --- IP DHCP Client ---
  1083. func (e *Executor) handleIPDHCPClient(parts []string, user *auth.PublicUser) string {
  1084. if e.dhcpMgr == nil {
  1085. return ColorRed + "DHCP client manager not available.\n" + ColorReset
  1086. }
  1087. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1088. clients := e.dhcpMgr.ListClients()
  1089. tbl := NewTable("FLAGS", "INTERFACE", "USE-PEER-DNS", "STATUS", "ADDRESS", "GATEWAY")
  1090. for _, c := range clients {
  1091. flags := ""
  1092. if !c.Enabled {
  1093. flags += "X"
  1094. } else {
  1095. flags += "R"
  1096. }
  1097. addr := c.IPAddress
  1098. if addr == "" {
  1099. addr = "-"
  1100. }
  1101. gw := c.Gateway
  1102. if gw == "" {
  1103. gw = "-"
  1104. }
  1105. tbl.AddRow(flags, c.Interface, fmt.Sprintf("%t", c.UsePeerDNS), c.Status, addr, gw)
  1106. }
  1107. return tbl.String()
  1108. }
  1109. if user.Role == auth.RoleRead {
  1110. return ColorRed + "Permission denied: viewer role cannot modify DHCP clients\n" + ColorReset
  1111. }
  1112. sub := strings.ToLower(parts[0])
  1113. switch sub {
  1114. case "add":
  1115. _, params := parseArgs(parts[1:])
  1116. iface := params["interface"]
  1117. if iface == "" {
  1118. return ColorRed + "Usage: /ip dhcp-client add interface=<iface> [use-peer-dns=yes]\n" + ColorReset
  1119. }
  1120. peerDNS := true
  1121. if val, ok := params["use-peer-dns"]; ok {
  1122. peerDNS = (strings.ToLower(val) == "yes" || strings.ToLower(val) == "true")
  1123. }
  1124. req := dhcp.CreateDHCPClientRequest{
  1125. Interface: iface,
  1126. UsePeerDNS: peerDNS,
  1127. Enabled: true,
  1128. }
  1129. if _, err := e.dhcpMgr.AddClient(req); err != nil {
  1130. return ColorRed + "Error adding DHCP client: " + err.Error() + ColorReset + "\n"
  1131. }
  1132. return ColorGreen + fmt.Sprintf("DHCP client added on interface %s.\n", iface) + ColorReset
  1133. case "remove", "rem", "delete":
  1134. pos, _ := parseArgs(parts[1:])
  1135. if len(pos) == 0 {
  1136. return ColorRed + "Usage: /ip dhcp-client remove <id>\n" + ColorReset
  1137. }
  1138. if err := e.dhcpMgr.DeleteClient(pos[0]); err != nil {
  1139. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1140. }
  1141. return ColorGreen + fmt.Sprintf("DHCP client %s removed.\n", pos[0]) + ColorReset
  1142. case "renew":
  1143. pos, _ := parseArgs(parts[1:])
  1144. if len(pos) == 0 {
  1145. return ColorRed + "Usage: /ip dhcp-client renew <id>\n" + ColorReset
  1146. }
  1147. if err := e.dhcpMgr.RenewLease(pos[0]); err != nil {
  1148. return ColorRed + "Error renewing lease: " + err.Error() + ColorReset + "\n"
  1149. }
  1150. return ColorGreen + fmt.Sprintf("DHCP lease renewal requested for %s.\n", pos[0]) + ColorReset
  1151. case "release":
  1152. pos, _ := parseArgs(parts[1:])
  1153. if len(pos) == 0 {
  1154. return ColorRed + "Usage: /ip dhcp-client release <id>\n" + ColorReset
  1155. }
  1156. if err := e.dhcpMgr.ReleaseLease(pos[0]); err != nil {
  1157. return ColorRed + "Error releasing lease: " + err.Error() + ColorReset + "\n"
  1158. }
  1159. return ColorGreen + fmt.Sprintf("DHCP lease released for %s.\n", pos[0]) + ColorReset
  1160. default:
  1161. return ColorRed + "Usage: /ip dhcp-client [print|add|remove|renew|release]\n" + ColorReset
  1162. }
  1163. }
  1164. // --- IP DHCP Server ---
  1165. func (e *Executor) handleIPDHCPServer(parts []string, user *auth.PublicUser) string {
  1166. if e.dhcpServMgr == nil {
  1167. return ColorRed + "DHCP server manager not available.\n" + ColorReset
  1168. }
  1169. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1170. servers := e.dhcpServMgr.ListServers()
  1171. tbl := NewTable("FLAGS", "NAME", "INTERFACE", "ADDRESS-POOL", "LEASE-TIME")
  1172. for _, s := range servers {
  1173. flags := ""
  1174. if s.Disabled {
  1175. flags += "X"
  1176. } else {
  1177. flags += "R"
  1178. }
  1179. tbl.AddRow(flags, s.Name, s.Interface, s.AddressPool, s.LeaseTime)
  1180. }
  1181. return tbl.String()
  1182. }
  1183. sub := strings.ToLower(parts[0])
  1184. switch sub {
  1185. case "lease":
  1186. leases := e.dhcpServMgr.ListLeases()
  1187. tbl := NewTable("IP-ADDRESS", "MAC-ADDRESS", "HOSTNAME", "EXPIRES-IN", "STATUS")
  1188. for _, l := range leases {
  1189. tbl.AddRow(l.Address, l.MACAddress, l.HostName, l.ExpiresIn, l.Status)
  1190. }
  1191. return tbl.String()
  1192. case "network":
  1193. networks := e.dhcpServMgr.ListNetworks()
  1194. tbl := NewTable("SUBNET", "GATEWAY", "DNS-SERVERS", "DOMAIN")
  1195. for _, n := range networks {
  1196. tbl.AddRow(n.Address, n.Gateway, strings.Join(n.DNSServers, ","), n.Domain)
  1197. }
  1198. return tbl.String()
  1199. }
  1200. if user.Role == auth.RoleRead {
  1201. return ColorRed + "Permission denied: viewer role cannot modify DHCP server\n" + ColorReset
  1202. }
  1203. switch sub {
  1204. case "add":
  1205. _, params := parseArgs(parts[1:])
  1206. name := params["name"]
  1207. iface := params["interface"]
  1208. pool := params["address-pool"]
  1209. if pool == "" {
  1210. start := params["range-start"]
  1211. end := params["range-end"]
  1212. if start != "" && end != "" {
  1213. pool = fmt.Sprintf("%s-%s", start, end)
  1214. }
  1215. }
  1216. leaseTime := params["lease-time"]
  1217. if leaseTime == "" {
  1218. leaseTime = "12h"
  1219. }
  1220. if name == "" || iface == "" || pool == "" {
  1221. return ColorRed + "Usage: /ip dhcp-server add name=<name> interface=<iface> address-pool=<start-end> [lease-time=12h]\n" + ColorReset
  1222. }
  1223. inst := dhcp.DHCPServerInstance{
  1224. Name: name,
  1225. Interface: iface,
  1226. AddressPool: pool,
  1227. LeaseTime: leaseTime,
  1228. Disabled: false,
  1229. }
  1230. if _, err := e.dhcpServMgr.SaveServer(inst); err != nil {
  1231. return ColorRed + "Error adding DHCP server: " + err.Error() + ColorReset + "\n"
  1232. }
  1233. return ColorGreen + fmt.Sprintf("DHCP server %s added on interface %s.\n", name, iface) + ColorReset
  1234. case "remove", "rem", "delete":
  1235. pos, _ := parseArgs(parts[1:])
  1236. if len(pos) == 0 {
  1237. return ColorRed + "Usage: /ip dhcp-server remove <id>\n" + ColorReset
  1238. }
  1239. if err := e.dhcpServMgr.DeleteServer(pos[0]); err != nil {
  1240. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1241. }
  1242. return ColorGreen + fmt.Sprintf("DHCP server %s removed.\n", pos[0]) + ColorReset
  1243. case "enable":
  1244. pos, _ := parseArgs(parts[1:])
  1245. if len(pos) == 0 {
  1246. return ColorRed + "Usage: /ip dhcp-server enable <id>\n" + ColorReset
  1247. }
  1248. if err := e.dhcpServMgr.ToggleServerState(pos[0], false); err != nil {
  1249. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1250. }
  1251. return ColorGreen + fmt.Sprintf("DHCP server %s enabled.\n", pos[0]) + ColorReset
  1252. case "disable":
  1253. pos, _ := parseArgs(parts[1:])
  1254. if len(pos) == 0 {
  1255. return ColorRed + "Usage: /ip dhcp-server disable <id>\n" + ColorReset
  1256. }
  1257. if err := e.dhcpServMgr.ToggleServerState(pos[0], true); err != nil {
  1258. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1259. }
  1260. return ColorGreen + fmt.Sprintf("DHCP server %s disabled.\n", pos[0]) + ColorReset
  1261. default:
  1262. return ColorRed + "Usage: /ip dhcp-server [print|add|remove|enable|disable|lease|network]\n" + ColorReset
  1263. }
  1264. }
  1265. // --- IP DNS ---
  1266. func (e *Executor) handleIPDNS(parts []string, user *auth.PublicUser) string {
  1267. if e.dnsMgr == nil {
  1268. return ColorRed + "DNS manager not available.\n" + ColorReset
  1269. }
  1270. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1271. settings := e.dnsMgr.GetSettings()
  1272. out := " servers: " + strings.Join(settings.Servers, ", ") + "\n"
  1273. out += " dynamic-servers: " + strings.Join(settings.DynamicServers, ", ") + "\n"
  1274. out += fmt.Sprintf(" allow-remote-requests: %t\n", settings.AllowRemoteRequests)
  1275. out += fmt.Sprintf(" cache-size: %d entries\n", settings.CacheSize)
  1276. out += fmt.Sprintf(" domain: %s\n", settings.Domain)
  1277. return out
  1278. }
  1279. sub := strings.ToLower(parts[0])
  1280. switch sub {
  1281. case "cache":
  1282. if len(parts) > 1 && parts[1] == "flush" {
  1283. if user.Role == auth.RoleRead {
  1284. return ColorRed + "Permission denied\n" + ColorReset
  1285. }
  1286. if err := e.dnsMgr.FlushCache(); err != nil {
  1287. return ColorRed + "Error flushing DNS cache: " + err.Error() + ColorReset + "\n"
  1288. }
  1289. return ColorGreen + "DNS cache flushed.\n" + ColorReset
  1290. }
  1291. return ColorYellow + "Usage: /ip dns cache flush\n" + ColorReset
  1292. case "static":
  1293. records := e.dnsMgr.ListStaticRecords()
  1294. tbl := NewTable("#", "NAME", "ADDRESS", "TTL", "STATUS")
  1295. for idx, r := range records {
  1296. st := "enabled"
  1297. if r.Disabled {
  1298. st = "disabled"
  1299. }
  1300. tbl.AddRow(fmt.Sprintf("%d", idx), r.Name, r.Address, fmt.Sprintf("%d", r.TTL), st)
  1301. }
  1302. return tbl.String()
  1303. case "set":
  1304. if user.Role == auth.RoleRead {
  1305. return ColorRed + "Permission denied\n" + ColorReset
  1306. }
  1307. _, params := parseArgs(parts[1:])
  1308. if sStr, ok := params["servers"]; ok {
  1309. srvs := strings.Split(sStr, ",")
  1310. cfg := e.dnsMgr.GetSettings()
  1311. cfg.Servers = srvs
  1312. if err := e.dnsMgr.SaveSettings(cfg); err != nil {
  1313. return ColorRed + "Error saving DNS settings: " + err.Error() + ColorReset + "\n"
  1314. }
  1315. return ColorGreen + "DNS servers updated.\n" + ColorReset
  1316. }
  1317. return ColorRed + "Usage: /ip dns set servers=<ip1,ip2>\n" + ColorReset
  1318. default:
  1319. return ColorRed + "Usage: /ip dns [print|set|cache flush|static]\n" + ColorReset
  1320. }
  1321. }
  1322. // --- IP Firewall ---
  1323. func (e *Executor) handleIPFirewall(parts []string, user *auth.PublicUser) string {
  1324. if e.fwMgr == nil {
  1325. return ColorRed + "Firewall manager not available.\n" + ColorReset
  1326. }
  1327. if len(parts) == 0 {
  1328. return ColorYellow + "Firewall subsystem. Available: filter, nat, mangle, address-list\n" + ColorReset
  1329. }
  1330. tableType := strings.ToLower(parts[0])
  1331. switch tableType {
  1332. case "filter":
  1333. return e.handleFirewallFilter(parts[1:], user)
  1334. case "nat":
  1335. return e.handleFirewallNAT(parts[1:], user)
  1336. case "mangle":
  1337. return e.handleFirewallMangle(parts[1:], user)
  1338. case "address-list":
  1339. return e.handleFirewallAddressList(parts[1:], user)
  1340. default:
  1341. return ColorRed + "Unknown firewall table: " + tableType + "\n" + ColorReset
  1342. }
  1343. }
  1344. func (e *Executor) handleFirewallFilter(parts []string, user *auth.PublicUser) string {
  1345. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1346. rules := e.fwMgr.ListFilterRules()
  1347. tbl := NewTable("FLAGS", "CHAIN", "ACTION", "PROTO", "SRC-ADDR", "DST-ADDR", "DST-PORT", "COMMENT")
  1348. for _, r := range rules {
  1349. flags := ""
  1350. if r.Disabled {
  1351. flags += "X"
  1352. } else {
  1353. flags += " "
  1354. }
  1355. tbl.AddRow(flags, r.Chain, r.Action, r.Protocol, r.SrcAddress, r.DstAddress, r.DstPort, r.Comment)
  1356. }
  1357. return tbl.String()
  1358. }
  1359. if user.Role == auth.RoleRead {
  1360. return ColorRed + "Permission denied: viewer role cannot modify firewall\n" + ColorReset
  1361. }
  1362. sub := strings.ToLower(parts[0])
  1363. switch sub {
  1364. case "add":
  1365. _, params := parseArgs(parts[1:])
  1366. chain := params["chain"]
  1367. action := params["action"]
  1368. if chain == "" || action == "" {
  1369. return ColorRed + "Usage: /ip firewall filter add chain=<input|forward|output> action=<accept|drop|reject> [protocol=...] [dst-port=...]\n" + ColorReset
  1370. }
  1371. r := firewall.FilterRule{
  1372. Chain: chain,
  1373. Action: action,
  1374. Protocol: params["protocol"],
  1375. SrcAddress: params["src-address"],
  1376. DstAddress: params["dst-address"],
  1377. DstPort: params["dst-port"],
  1378. Comment: params["comment"],
  1379. Disabled: false,
  1380. }
  1381. if err := e.fwMgr.SaveFilterRule(r); err != nil {
  1382. return ColorRed + "Error adding filter rule: " + err.Error() + ColorReset + "\n"
  1383. }
  1384. return ColorGreen + "Firewall filter rule added.\n" + ColorReset
  1385. case "remove", "rem", "delete":
  1386. pos, _ := parseArgs(parts[1:])
  1387. if len(pos) == 0 {
  1388. return ColorRed + "Usage: /ip firewall filter remove <id>\n" + ColorReset
  1389. }
  1390. if err := e.fwMgr.DeleteFilterRule(pos[0]); err != nil {
  1391. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1392. }
  1393. return ColorGreen + fmt.Sprintf("Filter rule %s removed.\n", pos[0]) + ColorReset
  1394. case "enable":
  1395. pos, _ := parseArgs(parts[1:])
  1396. if len(pos) == 0 {
  1397. return ColorRed + "Usage: /ip firewall filter enable <id>\n" + ColorReset
  1398. }
  1399. if err := e.fwMgr.SetFilterRuleState(pos[0], false); err != nil {
  1400. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1401. }
  1402. return ColorGreen + fmt.Sprintf("Filter rule %s enabled.\n", pos[0]) + ColorReset
  1403. case "disable":
  1404. pos, _ := parseArgs(parts[1:])
  1405. if len(pos) == 0 {
  1406. return ColorRed + "Usage: /ip firewall filter disable <id>\n" + ColorReset
  1407. }
  1408. if err := e.fwMgr.SetFilterRuleState(pos[0], true); err != nil {
  1409. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1410. }
  1411. return ColorGreen + fmt.Sprintf("Filter rule %s disabled.\n", pos[0]) + ColorReset
  1412. default:
  1413. return ColorRed + "Usage: /ip firewall filter [print|add|remove|enable|disable]\n" + ColorReset
  1414. }
  1415. }
  1416. func (e *Executor) handleFirewallNAT(parts []string, user *auth.PublicUser) string {
  1417. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1418. rules := e.fwMgr.ListNATRules()
  1419. tbl := NewTable("FLAGS", "CHAIN", "ACTION", "OUT-IFACE", "TO-ADDRESSES", "TO-PORTS", "COMMENT")
  1420. for _, r := range rules {
  1421. flags := ""
  1422. if r.Disabled {
  1423. flags += "X"
  1424. } else {
  1425. flags += " "
  1426. }
  1427. tbl.AddRow(flags, r.Chain, r.Action, r.OutInterface, r.ToAddresses, r.ToPorts, r.Comment)
  1428. }
  1429. return tbl.String()
  1430. }
  1431. if user.Role == auth.RoleRead {
  1432. return ColorRed + "Permission denied: viewer role cannot modify firewall\n" + ColorReset
  1433. }
  1434. sub := strings.ToLower(parts[0])
  1435. switch sub {
  1436. case "add":
  1437. _, params := parseArgs(parts[1:])
  1438. chain := params["chain"]
  1439. action := params["action"]
  1440. if chain == "" || action == "" {
  1441. return ColorRed + "Usage: /ip firewall nat add chain=<srcnat|dstnat> action=<masquerade|dst-nat|src-nat> [out-interface=...] [to-addresses=...]\n" + ColorReset
  1442. }
  1443. r := firewall.NATRule{
  1444. Chain: chain,
  1445. Action: action,
  1446. OutInterface: params["out-interface"],
  1447. ToAddresses: params["to-addresses"],
  1448. ToPorts: params["to-ports"],
  1449. Comment: params["comment"],
  1450. Disabled: false,
  1451. }
  1452. if err := e.fwMgr.SaveNATRule(r); err != nil {
  1453. return ColorRed + "Error adding NAT rule: " + err.Error() + ColorReset + "\n"
  1454. }
  1455. return ColorGreen + "Firewall NAT rule added.\n" + ColorReset
  1456. case "remove", "rem", "delete":
  1457. pos, _ := parseArgs(parts[1:])
  1458. if len(pos) == 0 {
  1459. return ColorRed + "Usage: /ip firewall nat remove <id>\n" + ColorReset
  1460. }
  1461. if err := e.fwMgr.DeleteNATRule(pos[0]); err != nil {
  1462. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1463. }
  1464. return ColorGreen + fmt.Sprintf("NAT rule %s removed.\n", pos[0]) + ColorReset
  1465. case "enable":
  1466. pos, _ := parseArgs(parts[1:])
  1467. if len(pos) == 0 {
  1468. return ColorRed + "Usage: /ip firewall nat enable <id>\n" + ColorReset
  1469. }
  1470. if err := e.fwMgr.SetNATRuleState(pos[0], false); err != nil {
  1471. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1472. }
  1473. return ColorGreen + fmt.Sprintf("NAT rule %s enabled.\n", pos[0]) + ColorReset
  1474. case "disable":
  1475. pos, _ := parseArgs(parts[1:])
  1476. if len(pos) == 0 {
  1477. return ColorRed + "Usage: /ip firewall nat disable <id>\n" + ColorReset
  1478. }
  1479. if err := e.fwMgr.SetNATRuleState(pos[0], true); err != nil {
  1480. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1481. }
  1482. return ColorGreen + fmt.Sprintf("NAT rule %s disabled.\n", pos[0]) + ColorReset
  1483. default:
  1484. return ColorRed + "Usage: /ip firewall nat [print|add|remove|enable|disable]\n" + ColorReset
  1485. }
  1486. }
  1487. func (e *Executor) handleFirewallMangle(parts []string, user *auth.PublicUser) string {
  1488. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1489. rules := e.fwMgr.ListMangleRules()
  1490. tbl := NewTable("FLAGS", "CHAIN", "ACTION", "PASSTHROUGH", "COMMENT")
  1491. for _, r := range rules {
  1492. flags := ""
  1493. if r.Disabled {
  1494. flags += "X"
  1495. } else {
  1496. flags += " "
  1497. }
  1498. tbl.AddRow(flags, r.Chain, r.Action, fmt.Sprintf("%t", r.Passthrough), r.Comment)
  1499. }
  1500. return tbl.String()
  1501. }
  1502. if user.Role == auth.RoleRead {
  1503. return ColorRed + "Permission denied\n" + ColorReset
  1504. }
  1505. sub := strings.ToLower(parts[0])
  1506. switch sub {
  1507. case "add":
  1508. _, params := parseArgs(parts[1:])
  1509. chain := params["chain"]
  1510. action := params["action"]
  1511. if chain == "" || action == "" {
  1512. return ColorRed + "Usage: /ip firewall mangle add chain=... action=...\n" + ColorReset
  1513. }
  1514. r := firewall.MangleRule{
  1515. Chain: chain,
  1516. Action: action,
  1517. Passthrough: true,
  1518. Comment: params["comment"],
  1519. Disabled: false,
  1520. }
  1521. if err := e.fwMgr.SaveMangleRule(r); err != nil {
  1522. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1523. }
  1524. return ColorGreen + "Mangle rule added.\n" + ColorReset
  1525. case "remove", "rem", "delete":
  1526. pos, _ := parseArgs(parts[1:])
  1527. if len(pos) == 0 {
  1528. return ColorRed + "Usage: /ip firewall mangle remove <id>\n" + ColorReset
  1529. }
  1530. if err := e.fwMgr.DeleteMangleRule(pos[0]); err != nil {
  1531. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1532. }
  1533. return ColorGreen + fmt.Sprintf("Mangle rule %s removed.\n", pos[0]) + ColorReset
  1534. default:
  1535. return ColorRed + "Usage: /ip firewall mangle [print|add|remove]\n" + ColorReset
  1536. }
  1537. }
  1538. func (e *Executor) handleFirewallAddressList(parts []string, user *auth.PublicUser) string {
  1539. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1540. entries := e.fwMgr.ListAddressLists()
  1541. tbl := NewTable("FLAGS", "LIST", "ADDRESS", "COMMENT")
  1542. for _, entry := range entries {
  1543. flags := ""
  1544. if entry.Disabled {
  1545. flags += "X"
  1546. } else {
  1547. flags += " "
  1548. }
  1549. tbl.AddRow(flags, entry.Name, entry.Address, entry.Comment)
  1550. }
  1551. return tbl.String()
  1552. }
  1553. if user.Role == auth.RoleRead {
  1554. return ColorRed + "Permission denied\n" + ColorReset
  1555. }
  1556. sub := strings.ToLower(parts[0])
  1557. switch sub {
  1558. case "add":
  1559. _, params := parseArgs(parts[1:])
  1560. list := params["list"]
  1561. addr := params["address"]
  1562. if list == "" || addr == "" {
  1563. return ColorRed + "Usage: /ip firewall address-list add list=<name> address=<cidr> [comment=...]\n" + ColorReset
  1564. }
  1565. entry := firewall.AddressListEntry{
  1566. Name: list,
  1567. Address: addr,
  1568. Comment: params["comment"],
  1569. Disabled: false,
  1570. }
  1571. if err := e.fwMgr.SaveAddressListEntry(entry); err != nil {
  1572. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1573. }
  1574. return ColorGreen + fmt.Sprintf("Added %s to address list %s.\n", addr, list) + ColorReset
  1575. case "remove", "rem", "delete":
  1576. pos, _ := parseArgs(parts[1:])
  1577. if len(pos) == 0 {
  1578. return ColorRed + "Usage: /ip firewall address-list remove <id>\n" + ColorReset
  1579. }
  1580. if err := e.fwMgr.DeleteAddressListEntry(pos[0]); err != nil {
  1581. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1582. }
  1583. return ColorGreen + fmt.Sprintf("Address list entry %s removed.\n", pos[0]) + ColorReset
  1584. default:
  1585. return ColorRed + "Usage: /ip firewall address-list [print|add|remove]\n" + ColorReset
  1586. }
  1587. }
  1588. // --- IP Service ---
  1589. func (e *Executor) handleIPService(parts []string, user *auth.PublicUser) string {
  1590. if e.servicesMgr == nil {
  1591. return ColorRed + "Services manager not available.\n" + ColorReset
  1592. }
  1593. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1594. svcs := e.servicesMgr.ListServices()
  1595. tbl := NewTable("NAME", "PORT", "DISABLED", "CERTIFICATE", "COMMENT")
  1596. for _, s := range svcs {
  1597. disStr := ColorGreen + "no" + ColorReset
  1598. if s.Disabled {
  1599. disStr = ColorRed + "yes" + ColorReset
  1600. }
  1601. certStr := "-"
  1602. if s.Certificate != "" {
  1603. certStr = ColorCyan + s.Certificate + ColorReset
  1604. }
  1605. tbl.AddRow(s.Name, fmt.Sprintf("%d", s.Port), disStr, certStr, s.Comment)
  1606. }
  1607. return tbl.String()
  1608. }
  1609. if user.Role != auth.RoleFull {
  1610. return ColorRed + "Permission denied: admin role required to manage services\n" + ColorReset
  1611. }
  1612. sub := strings.ToLower(parts[0])
  1613. switch sub {
  1614. case "enable":
  1615. pos, _ := parseArgs(parts[1:])
  1616. if len(pos) == 0 {
  1617. return ColorRed + "Usage: /ip service enable <name>\n" + ColorReset
  1618. }
  1619. name := pos[0]
  1620. if err := e.servicesMgr.SetServiceState(name, false); err != nil {
  1621. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1622. }
  1623. return ColorGreen + fmt.Sprintf("Service %s enabled.\n", name) + ColorReset
  1624. case "disable":
  1625. pos, _ := parseArgs(parts[1:])
  1626. if len(pos) == 0 {
  1627. return ColorRed + "Usage: /ip service disable <name>\n" + ColorReset
  1628. }
  1629. name := pos[0]
  1630. if err := e.servicesMgr.SetServiceState(name, true); err != nil {
  1631. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1632. }
  1633. return ColorGreen + fmt.Sprintf("Service %s disabled.\n", name) + ColorReset
  1634. case "set":
  1635. pos, params := parseArgs(parts[1:])
  1636. if len(pos) == 0 {
  1637. return ColorRed + "Usage: /ip service set <name> [port=<port>] [disabled=yes|no] [certificate=<cert_name>]\n" + ColorReset
  1638. }
  1639. name := pos[0]
  1640. svc, ok := e.servicesMgr.GetService(name)
  1641. if !ok {
  1642. return ColorRed + fmt.Sprintf("Service %q not found\n", name) + ColorReset
  1643. }
  1644. port := svc.Port
  1645. disabled := svc.Disabled
  1646. certName := svc.Certificate
  1647. if pStr, ok := params["port"]; ok {
  1648. if pVal, err := strconv.Atoi(pStr); err == nil && pVal > 0 && pVal < 65536 {
  1649. port = pVal
  1650. }
  1651. }
  1652. if dStr, ok := params["disabled"]; ok {
  1653. disabled = (strings.ToLower(dStr) == "yes" || strings.ToLower(dStr) == "true")
  1654. }
  1655. if cStr, ok := params["certificate"]; ok {
  1656. certName = cStr
  1657. }
  1658. if err := e.servicesMgr.SetService(name, port, disabled, certName); err != nil {
  1659. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1660. }
  1661. return ColorGreen + fmt.Sprintf("Service %s updated (port=%d, disabled=%t, certificate=%q).\n", name, port, disabled, certName) + ColorReset
  1662. default:
  1663. return ColorRed + "Usage: /ip service [print|enable|disable|set]\n" + ColorReset
  1664. }
  1665. }
  1666. // ==================== DYNAMIC ROUTING ====================
  1667. func (e *Executor) handleRouting(parts []string, user *auth.PublicUser) string {
  1668. if e.dynRoutingMgr == nil {
  1669. return ColorRed + "Dynamic routing subsystem not available.\n" + ColorReset
  1670. }
  1671. if len(parts) == 0 {
  1672. return ColorYellow + "Routing subsystem. Available: bgp, ospf, filter\n" + ColorReset
  1673. }
  1674. sub := strings.ToLower(parts[0])
  1675. switch sub {
  1676. case "bgp":
  1677. return e.handleRoutingBGP(parts[1:], user)
  1678. case "ospf":
  1679. return e.handleRoutingOSPF(parts[1:], user)
  1680. case "filter":
  1681. return e.handleRoutingFilter(parts[1:], user)
  1682. default:
  1683. return ColorRed + "Unknown routing protocol: " + sub + "\n" + ColorReset
  1684. }
  1685. }
  1686. func (e *Executor) handleRoutingBGP(parts []string, user *auth.PublicUser) string {
  1687. if len(parts) == 0 || parts[0] == "instance" {
  1688. inst := e.dynRoutingMgr.GetBGPInstance()
  1689. out := fmt.Sprintf(" enabled: %t\n", inst.Enabled)
  1690. out += fmt.Sprintf(" as: %d\n", inst.AS)
  1691. out += fmt.Sprintf(" router-id: %s\n", inst.RouterID)
  1692. return out
  1693. }
  1694. sub := strings.ToLower(parts[0])
  1695. switch sub {
  1696. case "peer", "connection", "peers":
  1697. peers := e.dynRoutingMgr.ListBGPPeers()
  1698. tbl := NewTable("FLAGS", "NAME", "REMOTE-AS", "REMOTE-ADDRESS", "STATUS", "UPTIME")
  1699. for _, p := range peers {
  1700. flags := ""
  1701. if p.Disabled {
  1702. flags += "X"
  1703. } else {
  1704. flags += "E"
  1705. }
  1706. st := p.State
  1707. if st == "" {
  1708. st = "Active"
  1709. }
  1710. tbl.AddRow(flags, p.Name, fmt.Sprintf("%d", p.RemoteAS), p.RemoteAddress, st, p.Uptime)
  1711. }
  1712. return tbl.String()
  1713. case "network", "networks":
  1714. nets := e.dynRoutingMgr.ListBGPNetworks()
  1715. tbl := NewTable("NETWORK", "COMMENT")
  1716. for _, n := range nets {
  1717. tbl.AddRow(n.Network, n.Comment)
  1718. }
  1719. return tbl.String()
  1720. case "routes":
  1721. routes := e.dynRoutingMgr.GetBGPRoutes()
  1722. tbl := NewTable("PREFIX", "NEXT-HOP", "PATH", "METRIC", "WEIGHT")
  1723. for _, r := range routes {
  1724. tbl.AddRow(r.Prefix, r.NextHop, r.Path, fmt.Sprintf("%d", r.Metric), fmt.Sprintf("%d", r.Weight))
  1725. }
  1726. return tbl.String()
  1727. default:
  1728. return ColorRed + "Usage: /routing bgp [instance|peer|network|routes]\n" + ColorReset
  1729. }
  1730. }
  1731. func (e *Executor) handleRoutingOSPF(parts []string, user *auth.PublicUser) string {
  1732. if len(parts) == 0 || parts[0] == "instance" {
  1733. instances := e.dynRoutingMgr.ListOSPFInstances()
  1734. tbl := NewTable("NAME", "ROUTER-ID", "VERSION", "STATUS")
  1735. for _, inst := range instances {
  1736. st := "enabled"
  1737. if inst.Disabled {
  1738. st = "disabled"
  1739. }
  1740. tbl.AddRow(inst.Name, inst.RouterID, fmt.Sprintf("%d", inst.Version), st)
  1741. }
  1742. return tbl.String()
  1743. }
  1744. sub := strings.ToLower(parts[0])
  1745. switch sub {
  1746. case "area", "areas":
  1747. areas := e.dynRoutingMgr.ListOSPFAreas()
  1748. tbl := NewTable("NAME", "AREA-ID", "INSTANCE", "TYPE")
  1749. for _, a := range areas {
  1750. tbl.AddRow(a.Name, a.AreaID, a.Instance, a.AreaType)
  1751. }
  1752. return tbl.String()
  1753. case "interface", "interfaces":
  1754. ifaces := e.dynRoutingMgr.ListOSPFInterfaces()
  1755. tbl := NewTable("INTERFACE", "AREA", "COST", "PRIORITY", "NETWORK-TYPE")
  1756. for _, i := range ifaces {
  1757. tbl.AddRow(i.Interface, i.Area, fmt.Sprintf("%d", i.Cost), fmt.Sprintf("%d", i.Priority), i.NetworkType)
  1758. }
  1759. return tbl.String()
  1760. case "neighbor", "neighbors":
  1761. neighbors := e.dynRoutingMgr.GetOSPFNeighbors()
  1762. tbl := NewTable("ROUTER-ID", "ADDRESS", "INTERFACE", "STATE", "PRIORITY")
  1763. for _, n := range neighbors {
  1764. tbl.AddRow(n.NeighborID, n.Address, n.Interface, n.State, fmt.Sprintf("%d", n.Priority))
  1765. }
  1766. return tbl.String()
  1767. default:
  1768. return ColorRed + "Usage: /routing ospf [instance|area|interface|neighbor]\n" + ColorReset
  1769. }
  1770. }
  1771. func (e *Executor) handleRoutingFilter(parts []string, user *auth.PublicUser) string {
  1772. rules := e.dynRoutingMgr.ListRoutingFilters()
  1773. tbl := NewTable("CHAIN", "RULE", "ACTION", "STATUS")
  1774. for _, r := range rules {
  1775. st := "enabled"
  1776. if r.Disabled {
  1777. st = "disabled"
  1778. }
  1779. tbl.AddRow(r.Chain, r.Rule, r.Action, st)
  1780. }
  1781. return tbl.String()
  1782. }
  1783. // ==================== TOOLS ====================
  1784. func (e *Executor) handleTool(parts []string) string {
  1785. if len(parts) == 0 {
  1786. return ColorYellow + "Tools available: ping, traceroute, bandwidth-test\n" + ColorReset
  1787. }
  1788. sub := strings.ToLower(parts[0])
  1789. switch sub {
  1790. case "ping":
  1791. return e.handlePing(parts[1:])
  1792. case "traceroute":
  1793. return e.handleTraceroute(parts[1:])
  1794. case "bandwidth-test", "btest":
  1795. return e.handleBandwidthTest(parts[1:])
  1796. default:
  1797. return ColorRed + "Unknown tool: " + sub + "\n" + ColorReset
  1798. }
  1799. }
  1800. func (e *Executor) handlePing(args []string) string {
  1801. if len(args) == 0 {
  1802. return ColorRed + "Usage: ping <host> [count=4]\n" + ColorReset
  1803. }
  1804. pos, params := parseArgs(args)
  1805. if len(pos) == 0 {
  1806. return ColorRed + "Missing host target for ping\n" + ColorReset
  1807. }
  1808. target := pos[0]
  1809. count := "4"
  1810. if c, ok := params["count"]; ok {
  1811. count = c
  1812. }
  1813. cmd := exec.Command("ping", "-c", count, target)
  1814. out, err := cmd.CombinedOutput()
  1815. if err != nil && len(out) == 0 {
  1816. return ColorRed + "Ping execution failed: " + err.Error() + ColorReset + "\n"
  1817. }
  1818. return string(out)
  1819. }
  1820. func (e *Executor) handleTraceroute(args []string) string {
  1821. if len(args) == 0 {
  1822. return ColorRed + "Usage: traceroute <host>\n" + ColorReset
  1823. }
  1824. pos, _ := parseArgs(args)
  1825. if len(pos) == 0 {
  1826. return ColorRed + "Missing host target for traceroute\n" + ColorReset
  1827. }
  1828. target := pos[0]
  1829. cmd := exec.Command("traceroute", "-n", "-m", "15", target)
  1830. out, err := cmd.CombinedOutput()
  1831. if err != nil {
  1832. cmdFallback := exec.Command("tracepath", "-n", target)
  1833. fbOut, fbErr := cmdFallback.CombinedOutput()
  1834. if fbErr == nil {
  1835. return string(fbOut)
  1836. }
  1837. return ColorRed + "Traceroute failed: " + err.Error() + "\n" + string(out) + ColorReset + "\n"
  1838. }
  1839. return string(out)
  1840. }
  1841. func (e *Executor) handleBandwidthTest(args []string) string {
  1842. if e.bwtestMgr == nil {
  1843. return ColorRed + "Bandwidth test manager not available.\n" + ColorReset
  1844. }
  1845. pos, params := parseArgs(args)
  1846. if len(pos) == 0 {
  1847. return ColorRed + "Usage: /tool bandwidth-test <target-ip> [direction=both|transmit|receive] [protocol=udp|tcp] [duration=5s]\n" + ColorReset
  1848. }
  1849. target := pos[0]
  1850. dir := "both"
  1851. if d, ok := params["direction"]; ok {
  1852. dir = d
  1853. }
  1854. switch dir {
  1855. case "transmit":
  1856. dir = "tx"
  1857. case "receive":
  1858. dir = "rx"
  1859. case "tx", "rx", "both":
  1860. default:
  1861. return cliError(fmt.Errorf("direction must be both, tx/transmit, or rx/receive"))
  1862. }
  1863. proto := "udp"
  1864. if p, ok := params["protocol"]; ok {
  1865. proto = p
  1866. }
  1867. if proto != "udp" && proto != "tcp" {
  1868. return cliError(fmt.Errorf("protocol must be udp or tcp"))
  1869. }
  1870. dur := 5 * time.Second
  1871. if dStr, ok := params["duration"]; ok {
  1872. if dVal, err := time.ParseDuration(dStr); err == nil && dVal >= time.Second {
  1873. dur = dVal
  1874. } else {
  1875. return cliError(fmt.Errorf("duration must be at least 1s"))
  1876. }
  1877. }
  1878. req := bwtest.SessionRequest{
  1879. Target: target,
  1880. Direction: dir,
  1881. Protocol: proto,
  1882. Duration: int(dur.Seconds()),
  1883. }
  1884. ctx, cancel := context.WithTimeout(context.Background(), dur+8*time.Second)
  1885. defer cancel()
  1886. if err := e.bwtestMgr.StartTest(ctx, req); err != nil {
  1887. return ColorRed + "Failed to initiate bandwidth test: " + err.Error() + ColorReset + "\n"
  1888. }
  1889. st := e.bwtestMgr.GetTestStatus()
  1890. ticker := time.NewTicker(100 * time.Millisecond)
  1891. defer ticker.Stop()
  1892. for st.Running {
  1893. select {
  1894. case <-ctx.Done():
  1895. e.bwtestMgr.StopTest()
  1896. return cliError(fmt.Errorf("bandwidth test did not finish before the deadline; stop requested"))
  1897. case <-ticker.C:
  1898. st = e.bwtestMgr.GetTestStatus()
  1899. }
  1900. }
  1901. if st.Error != "" {
  1902. return cliError(fmt.Errorf("bandwidth test failed: %s", st.Error))
  1903. }
  1904. out := fmt.Sprintf("Bandwidth test to %s completed:\n", target)
  1905. if st.Summary != nil {
  1906. out += fmt.Sprintf(" Tx Avg : %.2f Mbps\n", st.Summary.AvgTxMbps)
  1907. out += fmt.Sprintf(" Rx Avg : %.2f Mbps\n", st.Summary.AvgRxMbps)
  1908. out += fmt.Sprintf(" Loss : %.1f%%\n", st.Summary.AvgLossPct)
  1909. out += fmt.Sprintf(" Jitter : %.2f ms\n", st.Summary.AvgJitterMs)
  1910. } else if st.CurrentSample != nil {
  1911. out += fmt.Sprintf(" Tx Rate : %.2f Mbps\n", st.CurrentSample.TxMbps)
  1912. out += fmt.Sprintf(" Rx Rate : %.2f Mbps\n", st.CurrentSample.RxMbps)
  1913. } else {
  1914. out += " Test finished.\n"
  1915. }
  1916. return out
  1917. }
  1918. // ==================== USER MANAGEMENT ====================
  1919. func (e *Executor) handleUser(parts []string, actor *auth.PublicUser) string {
  1920. if e.authMgr == nil {
  1921. return ColorRed + "Auth manager not available.\n" + ColorReset
  1922. }
  1923. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1924. users := e.authMgr.ListUsers()
  1925. tbl := NewTable("FLAGS", "NAME", "ROLE", "LAST-LOGIN", "COMMENT")
  1926. for _, u := range users {
  1927. flags := ""
  1928. if u.Disabled {
  1929. flags += "X"
  1930. } else {
  1931. flags += " "
  1932. }
  1933. lastLogin := "never"
  1934. if !u.LastLoginAt.IsZero() {
  1935. lastLogin = u.LastLoginAt.Format("Jan/02 15:04:05")
  1936. }
  1937. tbl.AddRow(flags, u.Username, string(u.Role), lastLogin, u.Comment)
  1938. }
  1939. return tbl.String()
  1940. }
  1941. if actor.Role != auth.RoleFull {
  1942. return ColorRed + "Permission denied: admin role required to manage users\n" + ColorReset
  1943. }
  1944. sub := strings.ToLower(parts[0])
  1945. switch sub {
  1946. case "add":
  1947. _, params := parseArgs(parts[1:])
  1948. name := params["name"]
  1949. pwd := params["password"]
  1950. roleStr := params["role"]
  1951. if roleStr == "" {
  1952. roleStr = "admin"
  1953. }
  1954. comment := params["comment"]
  1955. if name == "" || pwd == "" {
  1956. return ColorRed + "Usage: /user add name=<username> password=<pwd> [role=admin|operator|viewer] [comment=...]\n" + ColorReset
  1957. }
  1958. req := auth.CreateUserRequest{
  1959. Username: name,
  1960. Password: pwd,
  1961. Role: auth.Role(roleStr),
  1962. Comment: comment,
  1963. }
  1964. if err := e.authMgr.CreateUser(req); err != nil {
  1965. return ColorRed + "Error creating user: " + err.Error() + ColorReset + "\n"
  1966. }
  1967. return ColorGreen + fmt.Sprintf("User %s created.\n", name) + ColorReset
  1968. case "remove", "rem", "delete":
  1969. pos, _ := parseArgs(parts[1:])
  1970. if len(pos) == 0 {
  1971. return ColorRed + "Usage: /user remove <username>\n" + ColorReset
  1972. }
  1973. name := pos[0]
  1974. if err := e.authMgr.DeleteUser(name, actor.Username); err != nil {
  1975. return ColorRed + "Error removing user: " + err.Error() + ColorReset + "\n"
  1976. }
  1977. return ColorGreen + fmt.Sprintf("User %s removed.\n", name) + ColorReset
  1978. case "set":
  1979. pos, params := parseArgs(parts[1:])
  1980. if len(pos) == 0 {
  1981. return ColorRed + "Usage: /user set <username> [password=...] [role=...] [disabled=yes|no]\n" + ColorReset
  1982. }
  1983. name := pos[0]
  1984. pub, err := e.authMgr.GetPublicUser(name)
  1985. if err != nil || pub == nil {
  1986. return ColorRed + "User not found\n" + ColorReset
  1987. }
  1988. role := pub.Role
  1989. if rStr, ok := params["role"]; ok {
  1990. role = auth.Role(rStr)
  1991. }
  1992. dis := pub.Disabled
  1993. if dStr, ok := params["disabled"]; ok {
  1994. dis = (strings.ToLower(dStr) == "yes" || strings.ToLower(dStr) == "true")
  1995. }
  1996. cmt := pub.Comment
  1997. if c, ok := params["comment"]; ok {
  1998. cmt = c
  1999. }
  2000. req := auth.UpdateUserRequest{
  2001. Role: role,
  2002. Disabled: dis,
  2003. Comment: cmt,
  2004. }
  2005. if err := e.authMgr.UpdateUser(name, req); err != nil {
  2006. return ColorRed + "Error updating user: " + err.Error() + ColorReset + "\n"
  2007. }
  2008. if pwd, ok := params["password"]; ok && pwd != "" {
  2009. if err := e.authMgr.ChangePassword(name, "", pwd, true); err != nil {
  2010. return ColorRed + "Error resetting password: " + err.Error() + ColorReset + "\n"
  2011. }
  2012. }
  2013. return ColorGreen + fmt.Sprintf("User %s updated.\n", name) + ColorReset
  2014. default:
  2015. return ColorRed + "Usage: /user [print|add|remove|set]\n" + ColorReset
  2016. }
  2017. }
  2018. // ==================== SYSTEM ====================
  2019. func (e *Executor) handleSystem(parts []string, user *auth.PublicUser) string {
  2020. if len(parts) == 0 {
  2021. return ColorYellow + "System commands: resource print, reboot, identity print, identity set name=...\n" + ColorReset
  2022. }
  2023. sub := strings.ToLower(parts[0])
  2024. switch sub {
  2025. case "resource":
  2026. st, err := system.GetStatus()
  2027. if err != nil {
  2028. return ColorRed + "Error retrieving system resources: " + err.Error() + ColorReset + "\n"
  2029. }
  2030. freeMemMB := uint64(0)
  2031. if st.MemTotalMB > st.MemUsedMB {
  2032. freeMemMB = st.MemTotalMB - st.MemUsedMB
  2033. }
  2034. out := " uptime: " + formatUptime(int64(st.UptimeSeconds)) + "\n"
  2035. out += fmt.Sprintf(" version: DebTik 1.0 (Kernel %s)\n", st.Kernel)
  2036. out += fmt.Sprintf(" free-memory: %d MiB\n", freeMemMB)
  2037. out += fmt.Sprintf(" total-memory: %d MiB\n", st.MemTotalMB)
  2038. out += fmt.Sprintf(" cpu-count: %d\n", st.CPUCores)
  2039. out += fmt.Sprintf(" cpu-load: %.1f%%\n", st.CPUPct)
  2040. out += fmt.Sprintf(" os: %s\n", st.OS)
  2041. return out
  2042. case "identity":
  2043. if len(parts) > 1 && parts[1] == "set" {
  2044. if user.Role != auth.RoleFull {
  2045. return ColorRed + "Permission denied: admin role required\n" + ColorReset
  2046. }
  2047. _, params := parseArgs(parts[2:])
  2048. name := params["name"]
  2049. if name == "" {
  2050. return ColorRed + "Usage: /system identity set name=<new-name>\n" + ColorReset
  2051. }
  2052. if err := exec.Command("hostnamectl", "set-hostname", name).Run(); err != nil {
  2053. return ColorRed + "Error setting identity: " + err.Error() + ColorReset + "\n"
  2054. }
  2055. return ColorGreen + fmt.Sprintf("System identity set to %q.\n", name) + ColorReset
  2056. }
  2057. st, _ := system.GetStatus()
  2058. h := "DebTik"
  2059. if st != nil && st.Hostname != "" {
  2060. h = st.Hostname
  2061. }
  2062. return fmt.Sprintf(" name: %s\n", h)
  2063. case "reboot":
  2064. if user.Role != auth.RoleFull {
  2065. return ColorRed + "Permission denied: admin role required to reboot\n" + ColorReset
  2066. }
  2067. go func() {
  2068. time.Sleep(1 * time.Second)
  2069. _ = exec.Command("systemctl", "reboot").Run()
  2070. }()
  2071. return ColorYellow + "System reboot initiated! Connection will terminate.\n" + ColorReset
  2072. default:
  2073. return ColorRed + "Unknown system subcommand: " + sub + "\n" + ColorReset
  2074. }
  2075. }
  2076. func (e *Executor) showHelp() string {
  2077. b := strings.Builder{}
  2078. b.WriteString(ColorCyan + ColorBold + "DebTik CLI Command Directory:\n" + ColorReset)
  2079. b.WriteString(" /interface - Network interface manager (print, enable, disable, set)\n")
  2080. b.WriteString(" /interface vlan - VLAN virtual interfaces (print, add, remove)\n")
  2081. b.WriteString(" /interface bridge - L2 Bridge and ports (print, add, remove, port)\n")
  2082. b.WriteString(" /interface aerotun - Encrypted tunnels (print detail=yes, add, set, enable, disable, start, stop, remove, genkey)\n")
  2083. b.WriteString(" /interface wireguard - WireGuard tunnels & peers (print, add, peers)\n")
  2084. b.WriteString(" /interface eoip - Ethernet over IP GRE tunnels (print, add, remove)\n")
  2085. b.WriteString(" /interface vrrp - Virtual Router Redundancy Protocol (print, add, remove)\n")
  2086. b.WriteString(" /interface pppoe-client - PPPoE WAN Client sessions (print, add, connect)\n")
  2087. b.WriteString(" /ip\n")
  2088. b.WriteString(" /ip address - IP address assignments (print, add, remove)\n")
  2089. b.WriteString(" /ip route - Static routes & tables (print, add, remove, rule)\n")
  2090. b.WriteString(" /ip dhcp-client - DHCP client on WAN links (print, add, renew, release)\n")
  2091. b.WriteString(" /ip dhcp-server - Local DHCP server & leases (print, add, lease, network)\n")
  2092. b.WriteString(" /ip dns - Router DNS resolver (print, set, cache flush)\n")
  2093. b.WriteString(" /ip firewall - Netfilter / iptables firewall\n")
  2094. b.WriteString(" /ip firewall filter - Traffic filtering rules (print, add, remove)\n")
  2095. b.WriteString(" /ip firewall nat - Source & destination NAT rules (print, add, remove)\n")
  2096. b.WriteString(" /ip firewall mangle - Packet marking and QoS (print, add, remove)\n")
  2097. b.WriteString(" /ip firewall address-list - Dynamic & static address groups (print, add, remove)\n")
  2098. b.WriteString(" /ip service - Management services: Telnet, SSH, Web (print, enable, disable, set)\n")
  2099. b.WriteString(" /routing - Dynamic routing engine\n")
  2100. b.WriteString(" /routing bgp - BGP instance, peers, routes (instance, peer, routes)\n")
  2101. b.WriteString(" /routing ospf - OSPF areas, interfaces, neighbors (instance, area, neighbor)\n")
  2102. b.WriteString(" /routing filter - Route filtering policies (print)\n")
  2103. b.WriteString(" /tool - Diagnostics & performance testing\n")
  2104. b.WriteString(" /tool ping <host> - ICMP ping test\n")
  2105. b.WriteString(" /tool traceroute <host> - Route trace analysis\n")
  2106. b.WriteString(" /tool bandwidth-test <host> - High performance throughput & jitter test\n")
  2107. b.WriteString(" /log - System activity logs (print [limit=N] [where=filter])\n")
  2108. b.WriteString(" /user - Local user accounts (print, add, remove, set)\n")
  2109. b.WriteString(" /system - Router hardware and core controls\n")
  2110. b.WriteString(" /system resource print - CPU, RAM, disk, uptime metrics\n")
  2111. b.WriteString(" /system identity - Hostname identity (print, set name=...)\n")
  2112. b.WriteString(" /system reboot - Safe system restart\n")
  2113. b.WriteString(" quit / exit - Disconnect session\n")
  2114. return b.String()
  2115. }
  2116. // ==================== CERTIFICATES (/certificate) ====================
  2117. func (e *Executor) handleCertificate(parts []string, user *auth.PublicUser) string {
  2118. if e.certMgr == nil {
  2119. return ColorRed + "Certificates manager not available.\n" + ColorReset
  2120. }
  2121. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  2122. list := e.certMgr.ListCertificates()
  2123. tbl := NewTable("#", "NAME", "COMMON-NAME", "TYPE", "ISSUER", "EXPIRES", "SANS")
  2124. for idx, c := range list {
  2125. tStr := ColorGreen + c.Type + ColorReset
  2126. if c.IsCA {
  2127. tStr = ColorCyan + "ca" + ColorReset
  2128. }
  2129. expStr := c.ValidTo.Format("Jan/02/2006")
  2130. if time.Now().After(c.ValidTo) {
  2131. expStr = ColorRed + expStr + " (EXPIRED)" + ColorReset
  2132. }
  2133. sansStr := strings.Join(c.SANs, ",")
  2134. if sansStr == "" {
  2135. sansStr = "-"
  2136. }
  2137. tbl.AddRow(fmt.Sprintf("%d", idx), c.Name, c.CommonName, tStr, c.Issuer, expStr, sansStr)
  2138. }
  2139. return tbl.String()
  2140. }
  2141. if user.Role != auth.RoleFull {
  2142. return ColorRed + "Permission denied: admin role required to manage certificates\n" + ColorReset
  2143. }
  2144. sub := strings.ToLower(parts[0])
  2145. switch sub {
  2146. case "add-ca":
  2147. _, params := parseArgs(parts[1:])
  2148. name := params["name"]
  2149. cn := params["common-name"]
  2150. if name == "" || cn == "" {
  2151. return ColorRed + "Usage: /certificate add-ca name=<name> common-name=<cn> [days=<days>] [key-type=RSA-2048|RSA-4096|ECDSA-P256]\n" + ColorReset
  2152. }
  2153. days := 3650
  2154. if dStr, ok := params["days"]; ok {
  2155. if dVal, err := strconv.Atoi(dStr); err == nil && dVal > 0 {
  2156. days = dVal
  2157. }
  2158. }
  2159. keyType := params["key-type"]
  2160. if keyType == "" {
  2161. keyType = "RSA-2048"
  2162. }
  2163. info, err := e.certMgr.CreateCA(certs.CreateCARequest{
  2164. Name: name,
  2165. CommonName: cn,
  2166. Days: days,
  2167. KeyType: keyType,
  2168. Comment: params["comment"],
  2169. })
  2170. if err != nil {
  2171. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  2172. }
  2173. return ColorGreen + fmt.Sprintf("Certificate Authority %q created successfully.\n", info.Name) + ColorReset
  2174. case "add":
  2175. _, params := parseArgs(parts[1:])
  2176. name := params["name"]
  2177. cn := params["common-name"]
  2178. if name == "" || cn == "" {
  2179. return ColorRed + "Usage: /certificate add name=<name> common-name=<cn> [ca=<ca_name>] [sans=<san1,san2>] [days=<days>] [key-type=...]\n" + ColorReset
  2180. }
  2181. days := 365
  2182. if dStr, ok := params["days"]; ok {
  2183. if dVal, err := strconv.Atoi(dStr); err == nil && dVal > 0 {
  2184. days = dVal
  2185. }
  2186. }
  2187. var sans []string
  2188. if sStr, ok := params["sans"]; ok {
  2189. for _, s := range strings.Split(sStr, ",") {
  2190. if s = strings.TrimSpace(s); s != "" {
  2191. sans = append(sans, s)
  2192. }
  2193. }
  2194. }
  2195. keyType := params["key-type"]
  2196. if keyType == "" {
  2197. keyType = "RSA-2048"
  2198. }
  2199. info, err := e.certMgr.CreateCertificate(certs.CreateCertRequest{
  2200. Name: name,
  2201. CommonName: cn,
  2202. CAName: params["ca"],
  2203. SANs: sans,
  2204. Days: days,
  2205. KeyType: keyType,
  2206. IsServer: true,
  2207. Comment: params["comment"],
  2208. })
  2209. if err != nil {
  2210. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  2211. }
  2212. return ColorGreen + fmt.Sprintf("Certificate %q created successfully (Issuer: %s).\n", info.Name, info.Issuer) + ColorReset
  2213. case "issue-letsencrypt":
  2214. _, params := parseArgs(parts[1:])
  2215. domain := params["domain"]
  2216. email := params["email"]
  2217. if domain == "" || email == "" {
  2218. return ColorRed + "Usage: /certificate issue-letsencrypt domain=<domain> email=<email> [name=<name>] [staging=yes|no]\n" + ColorReset
  2219. }
  2220. name := params["name"]
  2221. if name == "" {
  2222. name = domain
  2223. }
  2224. staging := strings.ToLower(params["staging"]) == "yes" || strings.ToLower(params["staging"]) == "true"
  2225. ctx, cancel := context.WithTimeout(context.Background(), 3*time.Minute)
  2226. defer cancel()
  2227. info, err := e.certMgr.IssueLetsEncrypt(ctx, certs.LetsEncryptRequest{
  2228. Name: name,
  2229. Domain: domain,
  2230. Email: email,
  2231. Staging: staging,
  2232. Comment: params["comment"],
  2233. })
  2234. if err != nil {
  2235. return ColorRed + "Let's Encrypt Error: " + err.Error() + ColorReset + "\n"
  2236. }
  2237. return ColorGreen + fmt.Sprintf("Let's Encrypt certificate %q issued successfully!\n", info.Name) + ColorReset
  2238. case "export":
  2239. pos, _ := parseArgs(parts[1:])
  2240. if len(pos) == 0 {
  2241. return ColorRed + "Usage: /certificate export <name>\n" + ColorReset
  2242. }
  2243. certPEM, keyPEM, err := e.certMgr.ExportPEM(pos[0])
  2244. if err != nil {
  2245. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  2246. }
  2247. return fmt.Sprintf("----- CERTIFICATE -----\n%s\n----- PRIVATE KEY -----\n%s\n", certPEM, keyPEM)
  2248. case "remove", "rem", "delete":
  2249. pos, _ := parseArgs(parts[1:])
  2250. if len(pos) == 0 {
  2251. return ColorRed + "Usage: /certificate remove <name>\n" + ColorReset
  2252. }
  2253. name := pos[0]
  2254. if e.servicesMgr != nil {
  2255. if sslSvc, ok := e.servicesMgr.GetService("www-ssl"); ok {
  2256. if strings.EqualFold(sslSvc.Certificate, name) && !sslSvc.Disabled {
  2257. return ColorRed + "Cannot remove certificate: actively used by enabled www-ssl service\n" + ColorReset
  2258. }
  2259. }
  2260. }
  2261. if err := e.certMgr.DeleteCertificate(name); err != nil {
  2262. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  2263. }
  2264. return ColorGreen + fmt.Sprintf("Certificate %s removed.\n", name) + ColorReset
  2265. default:
  2266. return ColorRed + "Usage: /certificate [print|add-ca|add|issue-letsencrypt|export|remove]\n" + ColorReset
  2267. }
  2268. }
  2269. func formatUptime(seconds int64) string {
  2270. d := seconds / 86400
  2271. h := (seconds % 86400) / 3600
  2272. m := (seconds % 3600) / 60
  2273. s := seconds % 60
  2274. if d > 0 {
  2275. return fmt.Sprintf("%dd %02dh %02dm %02ds", d, h, m, s)
  2276. }
  2277. return fmt.Sprintf("%02dh %02dm %02ds", h, m, s)
  2278. }
  2279. func formatBytes(bytes int64) string {
  2280. const unit = 1024
  2281. if bytes < unit {
  2282. return fmt.Sprintf("%d B", bytes)
  2283. }
  2284. div, exp := int64(unit), 0
  2285. for n := bytes / unit; n >= unit; n /= unit {
  2286. div *= unit
  2287. exp++
  2288. }
  2289. return fmt.Sprintf("%.1f %ciB", float64(bytes)/float64(div), "KMGTPE"[exp])
  2290. }
  2291. func (e *Executor) handleIPReverseProxy(parts []string, user *auth.PublicUser) string {
  2292. if e.proxyMgr == nil {
  2293. return ColorRed + "Reverse proxy service not available\n" + ColorReset
  2294. }
  2295. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  2296. rules := e.proxyMgr.ListRules()
  2297. cfg := e.proxyMgr.GetConfig()
  2298. var out strings.Builder
  2299. stateStr := ColorGreen + "enabled" + ColorReset
  2300. if !cfg.Enabled {
  2301. stateStr = ColorRed + "disabled" + ColorReset
  2302. }
  2303. out.WriteString(fmt.Sprintf("Flags: X - disabled\nReverse Proxy Master: %s | HTTP Port: %d | HTTPS Port: %d\n", stateStr, cfg.HTTPPort, cfg.HTTPSPort))
  2304. tbl := NewTable("Flags", "Name", "Domains", "Path", "Target", "Proto", "SSL Cert", "Requests")
  2305. for _, r := range rules {
  2306. flag := " "
  2307. if !r.Enabled {
  2308. flag = "X"
  2309. }
  2310. doms := strings.Join(r.Domains, ",")
  2311. if doms == "" {
  2312. doms = "*"
  2313. }
  2314. cert := r.Certificate
  2315. if cert == "" {
  2316. cert = "none"
  2317. }
  2318. reqCount := fmt.Sprintf("%d", r.Stats.Requests)
  2319. tbl.AddRow(flag, r.Name, doms, r.Path, r.TargetURL, r.Protocol, cert, reqCount)
  2320. }
  2321. out.WriteString(tbl.String())
  2322. return out.String()
  2323. }
  2324. if parts[0] == "status" {
  2325. cfg := e.proxyMgr.GetConfig()
  2326. var totalReq uint64
  2327. for _, r := range cfg.Rules {
  2328. totalReq += r.Stats.Requests
  2329. }
  2330. return fmt.Sprintf("Reverse Proxy Status:\n Enabled: %v\n HTTP Port: %d\n HTTPS Port: %d\n Rules Count: %d\n Total Requests: %d\n Default Cert: %s\n",
  2331. cfg.Enabled, cfg.HTTPPort, cfg.HTTPSPort, len(cfg.Rules), totalReq, cfg.DefaultCert)
  2332. }
  2333. if user.Role != auth.RoleFull {
  2334. return ColorRed + "Permission denied: admin role required\n" + ColorReset
  2335. }
  2336. sub := strings.ToLower(parts[0])
  2337. switch sub {
  2338. case "add":
  2339. _, params := parseArgs(parts[1:])
  2340. name := params["name"]
  2341. target := params["target"]
  2342. if target == "" {
  2343. target = params["target-url"]
  2344. }
  2345. if name == "" || target == "" {
  2346. return ColorRed + "Usage: /ip reverse-proxy add name=<name> target=<url> [domains=d1,d2] [path=<path>] [protocol=http|https|both] [certificate=<cert>] [strip-path=yes|no] [http-to-https=yes|no]\n" + ColorReset
  2347. }
  2348. var domains []string
  2349. if dStr, ok := params["domains"]; ok {
  2350. for _, d := range strings.Split(dStr, ",") {
  2351. if d = strings.TrimSpace(d); d != "" {
  2352. domains = append(domains, d)
  2353. }
  2354. }
  2355. }
  2356. pfx := params["path"]
  2357. if pfx == "" {
  2358. pfx = "/"
  2359. }
  2360. proto := params["protocol"]
  2361. if proto == "" {
  2362. proto = "both"
  2363. }
  2364. stripPath := (params["strip-path"] == "yes" || params["strip-path"] == "true")
  2365. httpToHttps := (params["http-to-https"] == "yes" || params["http-to-https"] == "true")
  2366. insecureTLS := (params["insecure-tls"] == "yes" || params["insecure-tls"] == "true")
  2367. rule, err := e.proxyMgr.SaveRule(&reverseproxy.Rule{
  2368. Name: name,
  2369. Enabled: true,
  2370. Domains: domains,
  2371. Path: pfx,
  2372. TargetURL: target,
  2373. StripPath: stripPath,
  2374. Protocol: proto,
  2375. Certificate: params["certificate"],
  2376. HTTPToHTTPS: httpToHttps,
  2377. InsecureTLS: insecureTLS,
  2378. Comment: params["comment"],
  2379. })
  2380. if err != nil {
  2381. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  2382. }
  2383. return ColorGreen + fmt.Sprintf("Proxy rule %q (ID: %s) added successfully.\n", rule.Name, rule.ID) + ColorReset
  2384. case "remove", "delete":
  2385. if len(parts) < 2 {
  2386. return ColorRed + "Usage: /ip reverse-proxy remove <id>\n" + ColorReset
  2387. }
  2388. id := parts[1]
  2389. if err := e.proxyMgr.DeleteRule(id); err != nil {
  2390. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  2391. }
  2392. return ColorGreen + fmt.Sprintf("Rule %q removed.\n", id) + ColorReset
  2393. case "enable":
  2394. if len(parts) < 2 {
  2395. return ColorRed + "Usage: /ip reverse-proxy enable <id>\n" + ColorReset
  2396. }
  2397. id := parts[1]
  2398. if err := e.proxyMgr.SetRuleState(id, true); err != nil {
  2399. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  2400. }
  2401. return ColorGreen + fmt.Sprintf("Rule %q enabled.\n", id) + ColorReset
  2402. case "disable":
  2403. if len(parts) < 2 {
  2404. return ColorRed + "Usage: /ip reverse-proxy disable <id>\n" + ColorReset
  2405. }
  2406. id := parts[1]
  2407. if err := e.proxyMgr.SetRuleState(id, false); err != nil {
  2408. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  2409. }
  2410. return ColorGreen + fmt.Sprintf("Rule %q disabled.\n", id) + ColorReset
  2411. default:
  2412. return ColorRed + "Usage: /ip reverse-proxy [print|status|add|set|remove|enable|disable]\n" + ColorReset
  2413. }
  2414. }
  2415. // Helpers for sorting/filtering
  2416. func sortStrings(s []string) {
  2417. sort.Strings(s)
  2418. }