Merge pull request #131 from destan0098/patch-1

Create new nGraph.py

lib/lib/nGraph/new nGraph.py

@@ -0,0 +1,51 @@
+import yaml
+import datetime
+import argparse
+import os
+from pyvis.network import Network
+
+def init_argparse():
+    parser = argparse.ArgumentParser(
+        usage="%(prog)s --domain example.com --file vhost.yaml",
+        description="Generate Network Graph For Sudomy."
+    )
+    parser.add_argument('--domain', type=str)
+    parser.add_argument('--file', type=str, help="File in yaml format")
+    return parser
+
+parser = init_argparse()
+args = parser.parse_args()
+
+if args.file is None or args.domain is None:
+    parser.print_help()
+    exit(0)
+
+main_domain = args.domain
+report_path = os.path.dirname(args.file)
+
+try:
+    domain_data = yaml.safe_load(open(args.file))
+except:
+    print("ERROR: File Is Not In Valid Yaml")
+    exit(-1)
+
+net = Network(height="1000px", width="100%", bgcolor="#2d2e2e", font_color="white", notebook=True)
+
+net.add_node(main_domain, color="#162347", label=main_domain, group=main_domain, labelHighlightBold=True)
+
+for ip in domain_data:
+    net.add_node(ip, label=ip, color="#bf281d", group=ip, physics=True, labelHighlightBold=True)
+    net.add_edge(ip, main_domain, title=main_domain, arrowStrikethrough=True, width=3)
+
+    for domain in domain_data.get(ip):
+        net.add_node(domain, label=domain, title=domain, color="#828282", physics=True, group=domain, labelHighlightBold=True)
+        net.add_edge(domain, ip, title=ip)
+
+d = datetime.date.today()
+date_now = '{}-{}-{}'.format(d.strftime('%m'), d.strftime('%d'), d.strftime('%Y'))
+output_directory = "{}/{}-nGraph_{}".format(report_path, main_domain, date_now)
+
+os.makedirs(output_directory, exist_ok=True)
+net.write_html("{}/{}-nGraph_{}.html".format(output_directory, main_domain, date_now))
+
+print(f"Network graph saved to {output_directory}/{main_domain}-nGraph_{date_now}.html")