feat: set ip address to agent

Author: mjabascal10

CHANGELOG.md

@@ -1,5 +1,7 @@
-# UTMStack 10.5.21 Release Notes
+# UTMStack 10.5.20 Release Notes
 ## Bug Fixes
+- Fixed the IP location component to accurately determine whether an IP address is public or private.
 
 ## New Features
-- Introduced new standards, sections, dashboards, and visualizations to compliance reports.
+- Introduced new standards, sections, dashboards, and visualizations to compliance reports.
+- Update ip address to agent

frontend/src/app/shared/components/utm/util/utm-agent-detail/utm-agent-detail.component.html

@@ -2,7 +2,16 @@
   <div class="agent-details w-100 d-flex justify-content-start mb-2"
        *ngIf="agent.ip">
     <span class="text-blue-800 font-weight-light has-minimum-width">IP:</span>&nbsp;
-    {{agent.ip}}
+      <ng-select (change)="changeIp($event)"
+                 [(ngModel)]="agentIp"
+                 [loading]="loading"
+                 [clearable]="true"
+                 [items]="ips"
+                 [searchable]="false"
+                 class="has-minimum-width mr-3"
+                 placeholder="Agent ip"
+                 style="width: 40%"
+      ></ng-select>
   </div>
   <div class="agent-details w-100 d-flex justify-content-start mb-2" *ngIf="agent.hostname">
     <span class="text-blue-800 font-weight-light has-minimum-width">Name:</span>&nbsp;

frontend/src/app/shared/components/utm/util/utm-agent-detail/utm-agent-detail.component.ts

@@ -1,5 +1,9 @@
 import {Component, Input, OnInit} from '@angular/core';
+import {EMPTY, of} from 'rxjs';
+import {catchError, map, tap} from 'rxjs/operators';
 import {NetScanType} from '../../../../../assets-discover/shared/types/net-scan.type';
+import {UtmToastService} from '../../../../alert/utm-toast.service';
+import {UtmAgentManagerService} from '../../../../services/agent/utm-agent-manager.service';
 import {AgentStatusEnum, AgentType} from '../../../../types/agent/agent.type';
 
 @Component({
@@ -11,11 +15,19 @@ export class UtmAgentDetailComponent implements OnInit {
   @Input() agent: AgentType;
   @Input() asset: NetScanType;
   agentStatusEnum = AgentStatusEnum;
+  ips: string[];
+  macs: string[];
+  agentIp: string;
+  loading = false;
 
-  constructor() {
+  constructor(private agentManagerService: UtmAgentManagerService,
+              private utmToastService: UtmToastService) {
   }
 
   ngOnInit() {
+    this.agentIp = this.agent.ip;
+    this.ips = this.agent.addresses !== '' ? this.agent.addresses.split(',') : [];
+    this.macs = this.agent.mac !== '' ? this.agent.mac.split(',') : [];
   }
 
   public getAgentIcon(): string {
@@ -36,4 +48,27 @@ export class UtmAgentDetailComponent implements OnInit {
   getAssets() {
 
   }
+
+  changeIp(event: string) {
+    this.loading = true;
+    const agent = {
+      id: this.agent.id,
+      ip: this.agentIp,
+      mac: this.macs.length > 0 ? this.macs[0] : '',
+      agentKey: this.agent.agentKey
+    };
+
+    this.agentManagerService.updateAgent(agent)
+      .pipe(
+        map(response => response.body),
+        tap(response => this.loading = false),
+        catchError(error => {
+          this.utmToastService.showError('Error',
+            'An error occurred while updating the agent ip. Please try again later.');
+          this.loading = false;
+          this.agentIp = this.agent.ip;
+          return EMPTY;
+        }))
+      .subscribe(() => this.agent.ip = this.agentIp);
+  }
 }

frontend/src/app/shared/services/agent/utm-agent-manager.service.ts

@@ -44,5 +44,8 @@ export class UtmAgentManagerService {
     return this.http.get<AgentCommandType[]>(this.resourceUrl + '/agent-commands', {params: options, observe: 'response'});
   }
 
+  updateAgent(agent: Partial<AgentType>): Observable<HttpResponse<any>> {
+    return this.http.post(this.resourceUrl + '/update-agent-attrs', agent, {observe: 'response'});
+  }
 
 }

version.yml

@@ -1 +1 @@
-version: 10.5.21
+version: 10.5.20