Add omitempty.

netbox_devices.go

@@ -1,11 +1,13 @@
 package netboxgo
 
 import (
+	"bytes"
 	"crypto/tls"
 	"encoding/json"
 	"io/ioutil"
 	"net/http"
 	"net/url"
+	"strconv"
 	"time"
 
 	"github.com/gorilla/schema"
@@ -129,6 +131,31 @@ type DcimDevicesList struct {
 		//	LocalContextData string   `json:"local_context_data"`
 		Tags         []string `json:"tags"`
 		CustomFields struct {
+			DcimDeviceBelongsToService struct {
+				Value int    `json:"value"`
+				Label string `json:"label"`
+			} `json:"dcim_device_belongs_to_service"`
+			DcimDeviceCendotid          string `json:"dcim_device_cendotid"`
+			DcimDeviceExposedToInternet struct {
+				Value int    `json:"value"`
+				Label string `json:"label"`
+			} `json:"dcim_device_exposed_to_internet"`
+			DcimDeviceImportOsVersion bool `json:"dcim_device_import_os_version"`
+			DcimDevicePod             struct {
+				Value int    `json:"value"`
+				Label string `json:"label"`
+			} `json:"dcim_device_pod"`
+			DcimDeviceOsVersion                 interface{} `json:"dcim_device_os_version"`
+			DcimDevicesImportInterfaces         interface{} `json:"dcim_devices_import_interfaces"`
+			DcimDeviceImportvlan                bool        `json:"dcim_device_importvlan"`
+			DcimDeviceL2Domain                  interface{} `json:"dcim_device_l2domain"`
+			DcimDeviceCountTrunkVlans           interface{} `json:"dcim_device_count_trunk_vlans"`
+			DcimDeviceLogicalportsDeviceLimit   interface{} `json:"dcim_device_logicalports_device_limit"`
+			DcimDeviceLogicalportsActiveNum     interface{} `json:"dcim_device_logicalports_active_num"`
+			DcimDeviceLogicalportsConfiguredNum interface{} `json:"dcim_device_logicalports_configured_num"`
+			DcimDeviceCountTimesLastRun         interface{} `json:"dcim_device_count_times_last_run"`
+			DcimDeviceImportVrfs                interface{} `json:"dcim_device_import_vrfs"`
+			DcimDeviceVrfTag                    interface{} `json:"dcim_device_vrf_tag"`
 		} `json:"custom_fields"`
 		ConfigContext struct {
 			AdditionalProp1 string `json:"additionalProp1"`
@@ -140,12 +167,66 @@ type DcimDevicesList struct {
 	} `json:"results"`
 }
 
+type DcimCreateDevice struct {
+	Name         string `json:"name,omitempty,omitempty"`
+	DeviceType   int    `json:"device_type,omitempty,omitempty"`
+	DeviceRole   int    `json:"device_role,omitempty,omitempty"`
+	Tenant       int    `json:"tenant,omitempty,omitempty"`
+	Platform     int    `json:"platform,omitempty,omitempty"`
+	Serial       string `json:"serial,omitempty"`
+	AssetTag     string `json:"asset_tag,omitempty"`
+	Site         int    `json:"site,omitempty"`
+	Rack         int    `json:"rack,omitempty"`
+	Position     int    `json:"position,omitempty"`
+	Face         string `json:"face,omitempty"`
+	ParentDevice struct {
+		Name string `json:"name,omitempty"`
+	} `json:"parent_device,omitempty"`
+	Status           string   `json:"status,omitempty"`
+	PrimaryIP4       int      `json:"primary_ip4,omitempty"`
+	PrimaryIP6       int      `json:"primary_ip6,omitempty"`
+	Cluster          int      `json:"cluster,omitempty"`
+	VirtualChassis   int      `json:"virtual_chassis,omitempty"`
+	VcPosition       int      `json:"vc_position,omitempty"`
+	VcPriority       int      `json:"vc_priority,omitempty"`
+	Comments         string   `json:"comments,omitempty"`
+	LocalContextData string   `json:"local_context_data,omitempty"`
+	Tags             []string `json:"tags,omitempty"`
+	CustomFields     struct {
+		DcimDeviceBelongsToService struct {
+			Value int    `json:"value,omitempty"`
+			Label string `json:"label,omitempty"`
+		} `json:"dcim_device_belongs_to_service,omitempty"`
+		DcimDeviceCendotid          string `json:"dcim_device_cendotid,omitempty"`
+		DcimDeviceExposedToInternet struct {
+			Value int    `json:"value,omitempty"`
+			Label string `json:"label,omitempty"`
+		} `json:"dcim_device_exposed_to_internet,omitempty"`
+		DcimDeviceImportOsVersion bool `json:"dcim_device_import_os_version,omitempty"`
+		DcimDevicePod             struct {
+			Value int    `json:"value,omitempty"`
+			Label string `json:"label,omitempty"`
+		} `json:"dcim_device_pod,omitempty"`
+		DcimDeviceOsVersion                 interface{} `json:"dcim_device_os_version,omitempty"`
+		DcimDevicesImportInterfaces         interface{} `json:"dcim_devices_import_interfaces,omitempty"`
+		DcimDeviceImportvlan                bool        `json:"dcim_device_importvlan,omitempty"`
+		DcimDeviceL2Domain                  interface{} `json:"dcim_device_l2domain,omitempty"`
+		DcimDeviceCountTrunkVlans           interface{} `json:"dcim_device_count_trunk_vlans,omitempty"`
+		DcimDeviceLogicalportsDeviceLimit   interface{} `json:"dcim_device_logicalports_device_limit,omitempty"`
+		DcimDeviceLogicalportsActiveNum     interface{} `json:"dcim_device_logicalports_active_num,omitempty"`
+		DcimDeviceLogicalportsConfiguredNum interface{} `json:"dcim_device_logicalports_configured_num,omitempty"`
+		DcimDeviceCountTimesLastRun         interface{} `json:"dcim_device_count_times_last_run,omitempty"`
+		DcimDeviceImportVrfs                interface{} `json:"dcim_device_import_vrfs,omitempty"`
+		DcimDeviceVrfTag                    interface{} `json:"dcim_device_vrf_tag,omitempty"`
+	} `json:"custom_fields,omitempty"`
+}
+
 // DeviceFilter is used to filter dcim_device_list query to the Netbox API
 type DeviceFilter struct {
 	Offset int64 `schema:"offset,omitempty"`
 	Limit  int64 `schema:"limit,omitempty"`
 
-	//User specific filters
+	// User specific filters
 	ID                  string `schema:"id,omitempty"`
 	Name                string `schema:"name,omitempty"`
 	AssetTag            string `schema:"asset_tag,omitempty"`
@@ -190,7 +271,7 @@ type DeviceFilter struct {
 
 // ListDevices method returns dcim_device_list from Netbox API
 func (n *NetBox) ListDevices(d *DcimDevicesList, f *DeviceFilter) error {
-	var encoder = schema.NewEncoder()
+	encoder := schema.NewEncoder()
 
 	transport := &http.Transport{
 		TLSClientConfig: &tls.Config{InsecureSkipVerify: n.InsecureSkipVerify},
@@ -243,3 +324,73 @@ func (n *NetBox) ListDevices(d *DcimDevicesList, f *DeviceFilter) error {
 	}
 	return nil
 }
+
+func (n *NetBox) CreateDevice(v *DcimCreateDevice) error {
+	data, err := json.Marshal(v)
+	if err != nil {
+		return err
+	}
+	transport := &http.Transport{
+		TLSClientConfig: &tls.Config{InsecureSkipVerify: n.InsecureSkipVerify},
+	}
+	timeout := time.Duration(60 * time.Second)
+	client := &http.Client{
+		Timeout:   timeout,
+		Transport: transport,
+	}
+	request, err := http.NewRequest("POST", n.RootURL+"/api/dcim/devices/", bytes.NewBuffer(data))
+	if err != nil {
+		return err
+	}
+	request.Header.Add("Accept", "application/json")
+	request.Header.Add("Content-Type", "application/json")
+	request.Header.Add("Authorization", " Token "+n.Token)
+	response, err := client.Do(request)
+	if err != nil {
+		return err
+	}
+	err = response.Body.Close()
+	if err != nil {
+		return err
+	}
+
+	if response.StatusCode == http.StatusCreated {
+		return nil
+	}
+	return errors.Errorf("Error: response was: %d should be %d\n", response.StatusCode, http.StatusCreated)
+}
+
+func (n *NetBox) UpdateDevice(v *DcimCreateDevice, id int) error {
+	data, err := json.Marshal(v)
+	if err != nil {
+		return err
+	}
+	transport := &http.Transport{
+		TLSClientConfig: &tls.Config{InsecureSkipVerify: n.InsecureSkipVerify},
+	}
+	timeout := time.Duration(60 * time.Second)
+	client := &http.Client{
+		Timeout:   timeout,
+		Transport: transport,
+	}
+	request, err := http.NewRequest("PUT", n.RootURL+"/api/dcim/devices/"+strconv.Itoa(id)+"/", bytes.NewBuffer(data))
+	if err != nil {
+		return err
+	}
+	request.Header.Add("Accept", "application/json")
+	request.Header.Add("Content-Type", "application/json")
+	request.Header.Add("Authorization", " Token "+n.Token)
+	response, err := client.Do(request)
+	if err != nil {
+		return err
+	}
+	err = response.Body.Close()
+	if err != nil {
+		return err
+	}
+
+	if response.StatusCode == http.StatusCreated {
+		return nil
+	}
+	return errors.Errorf("Error: response was: %d should be %d\n", response.StatusCode, http.StatusCreated)
+}