d3m0k1d/BanForge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 14 Wiki Activity

feat: update ban/unban command
All checks were successful
build / build (push) Successful in 3m13s
Details

Browse Source
This commit is contained in:
d3m0k1d
2026-01-25 21:13:56 +03:00
parent b0fc0646d2
commit e907fb0b1a
2 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
cmd/banforge/command/fw.go
View File

@@ -12,12 +12,20 @@ import (
)
var (
ip string
ttl_fw string
)
var UnbanCmd = &cobra.Command{
Use: "unban",
Short: "Unban IP",
Run: func(cmd *cobra.Command, args []string) {
if len(args) == 0 {
fmt.Println("IP can't be empty")
os.Exit(1)
}
if ttl_fw == "" {
ttl_fw = "1y"
}
ip := args[0]
db, err := storage.NewBanWriter()
if err != nil {
fmt.Println(err)
@@ -60,6 +68,14 @@ var BanCmd = &cobra.Command{
Use: "ban",
Short: "Ban IP",
Run: func(cmd *cobra.Command, args []string) {
if len(args) == 0 {
fmt.Println("IP can't be empty")
os.Exit(1)
}
if ttl_fw == "" {
ttl_fw = "1y"
}
ip := args[0]
db, err := storage.NewBanWriter()
if err != nil {
fmt.Println(err)
@@ -89,7 +105,7 @@ var BanCmd = &cobra.Command{
fmt.Println(err)
os.Exit(1)
}
err = db.AddBan(ip, "1y")
err = db.AddBan(ip, ttl_fw)
if err != nil {
fmt.Println(err)
os.Exit(1)
@@ -99,6 +115,5 @@ var BanCmd = &cobra.Command{
}
func FwRegister() {
BanCmd.Flags().StringVarP(&ip, "ip", "i", "", "ip to ban")
UnbanCmd.Flags().StringVarP(&ip, "ip", "i", "", "ip to unban")
BanCmd.Flags().StringVarP(&ttl_fw, "ttl", "t", "", "ban time")
}