import { AfterContentInit, Component, Input, EventEmitter, Output, SimpleChanges, OnInit, } from '@angular/core'; import { Subscription } from 'rxjs'; import { DiscoverZone, Zone, Host, Port, Service } from '@overflow/commons-typescript/model/discovery'; import { TreeNode } from 'primeng/primeng'; import { ProbeHost } from '@overflow/commons-typescript/model/probe'; import { Anim } from './animation'; import { DiscoveryNotify } from '../subscriber/discovery.subscriber'; @Component({ selector: 'of-discovery-result', templateUrl: './search-result.component.html', animations: Anim }) export class SearchResultComponent implements OnInit { @Input() probeHost: ProbeHost; @Input() filterWord: string; @Input() filterServices: Service[]; @Output() stop = new EventEmitter(); @Input() started: boolean; // Temporary discoverySubscription: Subscription; zoneNode: TreeNode[] = []; hostNode: TreeNode[] = []; selectedItems = []; tempHostId = 0; // Temporary constructor( ) { } ngOnInit(): void { this.zoneNode.push({ label: this.probeHost.probe.cidr, type: 'ZONE', data: { }, children: this.hostNode, expanded: true }); } tempHost() { const idx = Math.floor(Math.random() * (255)); const host: Host = { id: this.tempHostId++, ipv4: '192.168.1.' + idx, }; this.addHost(host); } tempPort() { const idx = Math.floor(Math.random() * (5000)); const hostId = Math.floor(Math.random() * (this.tempHostId - 1)); const port: Port = { id: idx, portNumber: idx, portType: idx % 2 === 0 ? 'TCP' : 'UDP', host: { id: hostId }, }; this.addPort(port); } tempService() { const randStr = Math.random().toString(36).substr(2, 5); const idx = Math.floor(Math.random() * (255)); const hostId = Math.floor(Math.random() * (this.tempHostId - 1)); const service: Service = { id: idx, serviceName: String(randStr), port: { portNumber: idx, portType: idx % 2 === 0 ? 'TCP' : 'UDP', host: { id: hostId } } }; this.addService(service); } addHost(host: Host) { const idx = this.findHostIndex(host); this.hostNode.splice(idx, 0, { type: 'HOST', label: host.ipv4, data: { id: host.id, ip: this.convertIPtoNumber(host.ipv4), openPorts: [] }, expanded: true, children: [] }); } addService(service: Service) { const targetHostNode = this.findHostNodeByService(service); const idx = this.findServiceIndex(targetHostNode.children, service); targetHostNode.children[idx] = { type: 'SERVICE', label: service.serviceName + ' (' + service.port.portNumber + ')', data: { id: service.id, name: service.serviceName, portType: service.port.portType, portNumber: service.port.portNumber, }, }; } 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())) { // index++; // } }); return index; } findHostNodeByService(service: Service) { let targetHost = null; this.hostNode.forEach((value, i) => { if (value.data.id === service.port.host.id) { targetHost = this.hostNode[i]; 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; }); } checkUnhighligt(label: string) { if (this.filterWord && label.indexOf(this.filterWord) === -1) { return true; } return false; } }