member_webapp/@overflow/discovery/component/search-result.component.ts

import {
  AfterContentInit, Component, Input,
  EventEmitter,
  Output,
  SimpleChanges,
  OnInit,
  OnChanges,
} from '@angular/core';

import { Subscription } from 'rxjs';

import { DiscoverZone, Zone, Host, Port, Service } from '@overflow/commons-typescript/model/discovery';
import { TreeNode, Message, Tree } from 'primeng/primeng';

import { ProbeHost } from '@overflow/commons-typescript/model/probe';

import { Anim } from './animation';
import { DiscoveryNotify } from '../subscriber/discovery.subscriber';
import { InfraHost, InfraService } from '@overflow/commons-typescript/model/infra';

@Component({
  selector: 'of-discovery-result',
  templateUrl: './search-result.component.html',
  animations: Anim
})
export class SearchResultComponent implements OnInit, OnChanges {

  @Input() probeHost: ProbeHost;
  @Input() filterWord: string;
  @Input() filterServices: Map<string, boolean>;
  @Input() finished: boolean;

  discoverySubscription: Subscription;
  zoneNode: TreeNode[] = [];
  hostNode: TreeNode[] = [];
  selectedItems: TreeNode[] = [];
  msgs: Message[];

  infraHosts = [];
  infraServices = [];

  constructor(
  ) {
  }

  ngOnInit(): void {
    this.zoneNode.push({
      label: this.probeHost.probe.cidr,
      type: 'ZONE',
      data: {
      },
      children: this.hostNode,
      expanded: true
    });
  }

  ngOnChanges(changes: SimpleChanges): void {
    if (changes['finished'] && changes['finished'].currentValue === true) {
      // this.saveInfras();
      // TODO: Comes after save infra
      this.msgs = [];
      this.msgs.push({
        severity: 'success',
        summary: 'Discovery가 완료되었습니다. 모니터링 대상(들)을 선택 후 저장하세요.',
      });
    }
  }


  addHost(host: Host) {
    const idx = this.findHostIndex(host);
    this.hostNode.splice(idx, 0, {
      type: 'HOST',
      label: host.ipv4,
      data: {
        ip: this.convertIPtoNumber(host.ipv4),
        ipv6: host.ipv6,
        mac: host.mac,
        openPorts: [],
        zone: host.zone,
      },
      expanded: true,
      children: []
    });
  }

  addService(service: Service) {
    const targetHostNode = this.findHostNodeByService(service);
    const idx = this.findServiceIndex(targetHostNode.children, service);
    targetHostNode.children.splice(idx, 0, {
      type: 'SERVICE',
      label: service.serviceName + ' (' + service.port.portNumber + ')',
      data: {
        name: service.serviceName,
        portType: service.port.portType,
        portNumber: service.port.portNumber,
        host: service.port.host
      },
    });
  }
  addPort(port: Port) {
    this.hostNode.forEach(node => {
      if (node.data.id === port.host.id) {
        node.data.openPorts.push(port);
        return;
      }
    });
  }

  onTargetSelect(e, data) {
    if (e.checked) {
      this.selectedItems.push(data);
    } else {
      const index = this.selectedItems.indexOf(data);
      this.selectedItems.splice(index, 1);
    }
  }


  findHostIndex(host: Host): number {
    let index = 0;
    this.hostNode.forEach(node => {
      if (node.data.ip < this.convertIPtoNumber(host.ipv4)) {
        index++;
      }
    });
    return index;
  }

  findServiceIndex(serviceNodes: TreeNode[], service: Service) {
    let index = 0;
    serviceNodes.forEach(node => {
      // if (node.data.portNumber < service.port.portNumber) {
      //   index++;
      // }
      if (node.data.name.toUpperCase().localeCompare(service.serviceName.toUpperCase()) === -1) {
        index++;
      }
    });
    return index;
  }

  findHostNodeByService(service: Service) {
    let targetHost = null;
    this.hostNode.forEach((value, i) => {
      if (value.data.ip === this.convertIPtoNumber(service.port.host.ipv4)) {
        targetHost = value;
        return;
      }
    });
    return targetHost;
  }

  convertIPtoNumber(ip: string) {
    return ip.split('.').map((octet, index, array) => {
      // tslint:disable-next-line:radix
      return parseInt(octet) * Math.pow(256, (array.length - index - 1));
    }).reduce((prev, curr) => {
      return prev + curr;
    });
  }

  checkHighligtHost(label: string) {
    if (!this.filterWord) {
      return true;
    }
    if (this.filterWord &&
      label.toUpperCase().indexOf(this.filterWord.toUpperCase()) > 0) {
      return true;
    }
    return false;
  }

  checkHighligtService(name: string) {
    if (this.filterServices && (this.filterServices[name])) {
      return true;
    }
    return false;
  }

  saveInfras(host: Host, service: Service) {
    //  Host
    const infraHost: InfraHost = {
      probe: this.probeHost.probe,
      infraType: {
        id: 2
      },
      os: {},
      ipv4: host.ipv4,
      ipv6: host.ipv6,
      mac:  host.mac,
    };

    // Service
    const infraService: InfraService = {
      probe: this.probeHost.probe,
      infraType: {
        id: 7
      },
      host: {
      },
      portType: service.port.portType,
      port: service.port.portNumber,
      vendor: {
      }
    };
  }

  saveTargets() {
    console.log(this.selectedItems);
    let node: TreeNode;
    for (node of this.selectedItems) {
      if (node.type === 'HOST') {
        // InfraHost
      } else if (node.type === 'SERVICE') {
        // InfraService
        // InfraOSPort?
      }
    }
  }
}