Use Domain ID when fetching the device list from netshot (#10)

Co-authored-by: Michaël Arnauts <[email protected]>

README.md

@@ -67,7 +67,8 @@ OPTIONS:
             The Netshot token [env: NETSHOT_TOKEN]
 
         --netshot-url <netshot-url>
-            The Netshot API URL [env: NETSHOT_URL=]```
+            The Netshot API URL [env: NETSHOT_URL=]
+```
 
 The query-string format need to be like this (url query string without the `?`):
 

src/main.rs

@@ -119,7 +119,7 @@ fn main() -> Result<(), Error> {
     netshot_client.ping()?;
 
     log::info!("Getting devices list from Netshot");
-    let netshot_devices = netshot_client.get_devices()?;
+    let netshot_devices = netshot_client.get_devices(opt.netshot_domain_id)?;
 
     let netshot_disabled_devices: Vec<&netshot::Device> = netshot_devices
         .iter()

src/rest/netbox.rs

@@ -35,7 +35,7 @@ pub struct Device {
     pub primary_ip4: Option<PrimaryIP>,
 }
 
-/// Represent the API response from /api/devim/devices call
+/// Represent the API response from /api/dcim/devices call
 #[derive(Debug, Serialize, Deserialize)]
 pub struct NetboxDCIMDeviceList {
     count: u32,

src/rest/netshot.rs

@@ -121,8 +121,10 @@ impl NetshotClient {
     }
 
     /// Get devices registered in Netshot
-    pub fn get_devices(&self) -> Result<Vec<Device>, Error> {
-        let url = format!("{}{}", self.url, PATH_DEVICES);
+    pub fn get_devices(&self,
+                       domain_id: u32,
+    ) -> Result<Vec<Device>, Error> {
+        let url = format!("{}{}?group={}", self.url, PATH_DEVICES, domain_id);
         let devices: Vec<Device> = self.client.get(url).send()?.json()?;
 
         log::debug!("Got {} devices from Netshot", devices.len());
@@ -291,7 +293,7 @@ mod tests {
             .create();
 
         let client = NetshotClient::new(url.clone(), String::new(), None, None, None).unwrap();
-        let devices = client.get_devices().unwrap();
+        let devices = client.get_devices(1).unwrap();
 
         assert_eq!(devices.len(), 1);