1 files changed, 72 insertions, 1 deletions

diff --git a/clients/netbox/model.go b/clients/netbox/model.go
index 40c4aa6..be4767d 100644
--- a/clients/netbox/model.go
+++ b/clients/netbox/model.go
@@ -1,6 +1,10 @@
 package netbox
 
-import "fmt"
+import (
+        "encoding/json"
+        "fmt"
+        "net"
+)
 
 type ApiError struct {
         Status int
@@ -85,3 +89,70 @@ type Prefix struct {
         Tags         []*Tag                 `json:"tags"`
         CustomFields map[string]interface{} `json:"custom_fields"`
 }
+
+type ServiceList struct {
+        Count    int        `json:"count"`
+        Next     string     `json:"next"`
+        Previous string     `json:"previous"`
+        Results  []*Service `json:"results"`
+}
+
+type ServiceProtocol struct {
+        Value string `json:"value"`
+        Label string `json:"label"`
+}
+
+type DeviceOrVM struct {
+        ID      int    `json:"id"`
+        Url     string `json:"url"`
+        Name    string `json:"name"`
+        Display string `json:"display"`
+}
+
+type ServiceIPAddress struct {
+        ID      int    `json:"id"`
+        Url     string `json:"url"`
+        Display string `json:"display"`
+        Family  int    `json:"family"`
+        Address net.IP
+        Network *net.IPNet
+}
+
+func (a *ServiceIPAddress) UnmarshalJSON(d []byte) error {
+        type Alias ServiceIPAddress
+        v := &struct {
+                *Alias
+                Address string `json:"address"`
+        }{Alias: (*Alias)(a)}
+
+        if err := json.Unmarshal(d, v); err != nil {
+                return err
+        }
+
+        i, n, err := net.ParseCIDR(v.Address)
+        if err != nil {
+                return err
+        }
+
+        a.Address = i
+        a.Network = n
+
+        return nil
+}
+
+type Service struct {
+        ID             int                    `json:"id" mapstructure:"id"`
+        Url            string                 `json:"url" mapstructure:"url"`
+        Name           string                 `json:"name" mapstructure:"name"`
+        Display        string                 `json:"display" mapstructure:"display"`
+        Device         *DeviceOrVM            `json:"device" mapstructure:"device"`
+        VirtualMachine *DeviceOrVM            `json:"virtual_machine" mapstructure:"virtual_machine"`
+        Ports          []int                  `json:"ports" mapstructure:"ports"`
+        Protocol       *ServiceProtocol       `json:"protocol" mapstructure:"protocol"`
+        Addresses      []*ServiceIPAddress    `json:"ipaddresses" mapstructure:"ipaddresses"`
+        Description    string                 `json:"description" mapstructure:"description"`
+        Created        string                 `json:"created" mapstructure:"created"`
+        LastUpdated    string                 `json:"last_updated" mapstructure:"last_updated"`
+        Tags           []*Tag                 `json:"tags" mapstructure:"tags"`
+        CustomFields   map[string]interface{} `json:"custom_fields" mapstructure:"custom_fields"`
+}