From f006f5e6817b3f3135ec7fa12e908185174c87de Mon Sep 17 00:00:00 2001 From: ziporah Date: Wed, 4 Mar 2020 22:29:22 +0100 Subject: [PATCH] Add unifi_port_forward --- .gitignore | 2 + internal/provider/lazy_client.go | 16 ++ internal/provider/portForward.go | 12 ++ internal/provider/provider.go | 6 + internal/provider/resource_port_forward.go | 180 +++++++++++++++++++++ 5 files changed, 216 insertions(+) create mode 100644 internal/provider/portForward.go create mode 100644 internal/provider/resource_port_forward.go diff --git a/.gitignore b/.gitignore index 9df1745..a762c76 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,4 @@ +*.swp + .vscode dist diff --git a/internal/provider/lazy_client.go b/internal/provider/lazy_client.go index 273ad18..0dcd795 100644 --- a/internal/provider/lazy_client.go +++ b/internal/provider/lazy_client.go @@ -169,3 +169,19 @@ func (c *lazyClient) UpdateFirewallGroup(site string, d *unifi.FirewallGroup) (* c.init() return c.inner.UpdateFirewallGroup(site, d) } +func (c *lazyClient) GetPortForward(site, id string) (*unifi.PortForward, error) { + c.init() + return c.inner.GetPortForward(site, id) +} +func (c *lazyClient) DeletePortForward(site, id string) error { + c.init() + return c.inner.DeletePortForward(site, id) +} +func (c *lazyClient) CreatePortForward(site string, d *unifi.PortForward) (*unifi.PortForward, error) { + c.init() + return c.inner.CreatePortForward(site, d) +} +func (c *lazyClient) UpdatePortForward(site string, d *unifi.PortForward) (*unifi.PortForward, error) { + c.init() + return c.inner.UpdatePortForward(site, d) +} diff --git a/internal/provider/portForward.go b/internal/provider/portForward.go new file mode 100644 index 0000000..0c773d9 --- /dev/null +++ b/internal/provider/portForward.go @@ -0,0 +1,12 @@ +package provider + +import ( + "regexp" +) + +var portForwardDstPortRegexp = regexp.MustCompile("(([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5])|([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5])-([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5]))+(,([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5])|,([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5])-([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5])){0,14}") + +// var portForwardDstPortRegexp = regexp.MustCompile("") +var portForwardFwdRegexp = regexp.MustCompile("^(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$") +var portForwardFwdPortRegexp = regexp.MustCompile("(([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5])|([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5])-([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5]))+(,([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5])|,([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5])-([1-9][0-9]{0,3}|[1-5][0-9]{4}|[6][0-4][0-9]{3}|[6][5][0-4][0-9]{2}|[6][5][5][0-2][0-9]|[6][5][5][3][0-5])){0,14}") +var portForwardSrcRegexp = regexp.MustCompile("^(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$|^(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])-(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$|^(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])/([0-9]|[1-2][0-9]|3[0-2])$|^!(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$|^!(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])-(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$|^!(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])/([0-9]|[1-2][0-9]|3[0-2])$|^any$") diff --git a/internal/provider/provider.go b/internal/provider/provider.go index 2afc10a..659fac5 100644 --- a/internal/provider/provider.go +++ b/internal/provider/provider.go @@ -45,6 +45,7 @@ func Provider() terraform.ResourceProvider { "unifi_user_group": resourceUserGroup(), "unifi_user": resourceUser(), "unifi_wlan": resourceWLAN(), + "unifi_port_forward": resourcePortForward(), }, } p.ConfigureFunc = configure(p) @@ -104,6 +105,11 @@ type unifiClient interface { UnblockUserByMAC(site, mac string) error UpdateUser(site string, d *unifi.User) (*unifi.User, error) DeleteUserByMAC(site, mac string) error + + GetPortForward(site, id string) (*unifi.PortForward, error) + DeletePortForward(site, id string) error + CreatePortForward(site string, d *unifi.PortForward) (*unifi.PortForward, error) + UpdatePortForward(site string, d *unifi.PortForward) (*unifi.PortForward, error) } type client struct { diff --git a/internal/provider/resource_port_forward.go b/internal/provider/resource_port_forward.go new file mode 100644 index 0000000..fb28d76 --- /dev/null +++ b/internal/provider/resource_port_forward.go @@ -0,0 +1,180 @@ +package provider + +import ( + "fmt" + + "github.com/hashicorp/terraform-plugin-sdk/helper/schema" + "github.com/hashicorp/terraform-plugin-sdk/helper/validation" + "github.com/paultyng/go-unifi/unifi" +) + +// GetPortForward(site string, id string) (*unifi.PortForward, error) +// DeletePortForward(site string, id string) (*unifi.PortForward, error) +// CreatePortForward(site string, d *unifi.PortForward) (*unifi.PortForward, error) +// UpdatePortForward(site string, d *unifi.PortForward) (*unifi.PortForward, error) + +func resourcePortForward() *schema.Resource { + return &schema.Resource{ + Create: resourcePortForwardCreate, + Read: resourcePortForwardRead, + Update: resourcePortForwardUpdate, + Delete: resourcePortForwardDelete, + Importer: &schema.ResourceImporter{ + State: schema.ImportStatePassthrough, + }, + + Schema: map[string]*schema.Schema{ + "dst_port": { + Type: schema.TypeString, + Required: false, + Optional: true, + ValidateFunc: validation.StringMatch(portForwardDstPortRegexp, "dst_port is invalid"), + }, + "enabled": { + Type: schema.TypeBool, + Optional: true, + }, + "fwd": { + Type: schema.TypeString, + Required: false, + Optional: true, + ValidateFunc: validation.StringMatch(portForwardFwdRegexp, "fwd is invalid"), + }, + "fwd_port": { + Type: schema.TypeString, + Required: false, + Optional: true, + ValidateFunc: validation.StringMatch(portForwardFwdPortRegexp, "fwd_port is invalid"), + }, + "log": { + Type: schema.TypeBool, + Default: false, + Optional: true, + }, + "name": { + Type: schema.TypeString, + Required: false, + Optional: true, + }, + "port_forward_interface": { + Type: schema.TypeString, + Required: false, + Optional: true, + ValidateFunc: validation.StringInSlice([]string{"wan", "wan2", "both", ""}, false), + }, + "proto": { + Type: schema.TypeString, + Optional: true, + Default: "tcp_udp", + ValidateFunc: validation.StringInSlice([]string{"tcp_udp", "tcp", "udp"}, false), + }, + "src": { + Type: schema.TypeString, + Required: false, + Optional: true, + Default: "any", + ValidateFunc: validation.StringMatch(portForwardSrcRegexp, "src is invalid"), + }, + }, + } +} + +func resourcePortForwardCreate(d *schema.ResourceData, meta interface{}) error { + c := meta.(*client) + + req, err := resourcePortForwardGetResourceData(d) + if err != nil { + return err + } + + resp, err := c.c.CreatePortForward(c.site, req) + if err != nil { + apiErr, ok := err.(*unifi.APIError) + if !ok { + + return fmt.Errorf("ResourcePortForwardCreate api error %w. With %w", apiErr, err) + } + + } + + d.SetId(resp.ID) + + return resourcePortForwardSetResourceData(resp, d) +} + +func resourcePortForwardGetResourceData(d *schema.ResourceData) (*unifi.PortForward, error) { + return &unifi.PortForward{ + DstPort: d.Get("dst_port").(string), + Enabled: d.Get("enabled").(bool), + Fwd: d.Get("fwd").(string), + FwdPort: d.Get("fwd_port").(string), + Log: d.Get("log").(bool), + Name: d.Get("name").(string), + PfwdInterface: d.Get("port_forward_interface").(string), + Proto: d.Get("proto").(string), + Src: d.Get("src").(string), + }, nil +} + +func resourcePortForwardSetResourceData(resp *unifi.PortForward, d *schema.ResourceData) error { + log := false + if resp.Log { + log = resp.Log + } + d.Set("dst_port", resp.DstPort) + d.Set("enabled", resp.Enabled) + d.Set("fwd", resp.Fwd) + d.Set("fwd_port", resp.FwdPort) + d.Set("log", log) + d.Set("name", resp.Name) + d.Set("port_forward_interface", resp.PfwdInterface) + d.Set("proto", resp.Proto) + d.Set("src", resp.Src) + + return nil +} + +func resourcePortForwardRead(d *schema.ResourceData, meta interface{}) error { + c := meta.(*client) + + id := d.Id() + + resp, err := c.c.GetPortForward(c.site, id) + if _, ok := err.(*unifi.NotFoundError); ok { + d.SetId("") + return nil + } + if err != nil { + return err + } + + return resourcePortForwardSetResourceData(resp, d) +} + +func resourcePortForwardUpdate(d *schema.ResourceData, meta interface{}) error { + c := meta.(*client) + + req, err := resourcePortForwardGetResourceData(d) + if err != nil { + return err + } + + req.ID = d.Id() + req.SiteID = c.site + + resp, err := c.c.UpdatePortForward(c.site, req) + if err != nil { + return err + } + + return resourcePortForwardSetResourceData(resp, d) +} + +func resourcePortForwardDelete(d *schema.ResourceData, meta interface{}) error { + c := meta.(*client) + + id := d.Id() + + err := c.c.DeletePortForward(c.site, id) + return err +}