|
1 | 1 | package controllers |
2 | 2 |
|
3 | 3 | import ( |
| 4 | + "github.com/TimothyYe/godns/internal/settings" |
4 | 5 | "github.com/TimothyYe/godns/internal/utils" |
5 | 6 | "github.com/gofiber/fiber/v2" |
6 | 7 | log "github.com/sirupsen/logrus" |
@@ -54,3 +55,66 @@ func (c *Controller) UpdateProvider(ctx *fiber.Ctx) error { |
54 | 55 |
|
55 | 56 | return ctx.SendStatus(fiber.StatusOK) |
56 | 57 | } |
| 58 | + |
| 59 | +func (c *Controller) GetMultiProviders(ctx *fiber.Ctx) error { |
| 60 | + return ctx.JSON(c.config.Providers) |
| 61 | +} |
| 62 | + |
| 63 | +func (c *Controller) UpdateMultiProviders(ctx *fiber.Ctx) error { |
| 64 | + var providers map[string]*settings.ProviderConfig |
| 65 | + if err := ctx.BodyParser(&providers); err != nil { |
| 66 | + return err |
| 67 | + } |
| 68 | + |
| 69 | + c.config.Providers = providers |
| 70 | + |
| 71 | + if err := c.config.SaveSettings(c.configPath); err != nil { |
| 72 | + log.Errorf("Failed to save settings: %s", err.Error()) |
| 73 | + return ctx.Status(500).SendString("Failed to save settings") |
| 74 | + } |
| 75 | + |
| 76 | + return ctx.SendStatus(fiber.StatusOK) |
| 77 | +} |
| 78 | + |
| 79 | +func (c *Controller) AddProviderConfig(ctx *fiber.Ctx) error { |
| 80 | + providerName := ctx.Params("provider") |
| 81 | + if providerName == "" { |
| 82 | + return ctx.Status(400).SendString("Provider name is required") |
| 83 | + } |
| 84 | + |
| 85 | + var config settings.ProviderConfig |
| 86 | + if err := ctx.BodyParser(&config); err != nil { |
| 87 | + return err |
| 88 | + } |
| 89 | + |
| 90 | + if c.config.Providers == nil { |
| 91 | + c.config.Providers = make(map[string]*settings.ProviderConfig) |
| 92 | + } |
| 93 | + |
| 94 | + c.config.Providers[providerName] = &config |
| 95 | + |
| 96 | + if err := c.config.SaveSettings(c.configPath); err != nil { |
| 97 | + log.Errorf("Failed to save settings: %s", err.Error()) |
| 98 | + return ctx.Status(500).SendString("Failed to save settings") |
| 99 | + } |
| 100 | + |
| 101 | + return ctx.SendStatus(fiber.StatusOK) |
| 102 | +} |
| 103 | + |
| 104 | +func (c *Controller) DeleteProviderConfig(ctx *fiber.Ctx) error { |
| 105 | + providerName := ctx.Params("provider") |
| 106 | + if providerName == "" { |
| 107 | + return ctx.Status(400).SendString("Provider name is required") |
| 108 | + } |
| 109 | + |
| 110 | + if c.config.Providers != nil { |
| 111 | + delete(c.config.Providers, providerName) |
| 112 | + |
| 113 | + if err := c.config.SaveSettings(c.configPath); err != nil { |
| 114 | + log.Errorf("Failed to save settings: %s", err.Error()) |
| 115 | + return ctx.Status(500).SendString("Failed to save settings") |
| 116 | + } |
| 117 | + } |
| 118 | + |
| 119 | + return ctx.SendStatus(fiber.StatusOK) |
| 120 | +} |
0 commit comments