From afde6eb55f9382311be3157b737d0186a73b3376 Mon Sep 17 00:00:00 2001
From: Alexandre Dulaunoy <a@foo.be>
Date: Tue, 25 May 2021 11:13:48 +0200
Subject: [PATCH] new: [GitHub] import org, user and repos as organisation and
 project in CyCAT.org

---
 crawler/github/README.md          | 18 +++++++
 crawler/github/github-importer.py | 88 +++++++++++++++++++++++++++++++
 crawler/github/lists              |  3 ++
 3 files changed, 109 insertions(+)
 create mode 100644 crawler/github/README.md
 create mode 100644 crawler/github/github-importer.py
 create mode 100644 crawler/github/lists

diff --git a/crawler/github/README.md b/crawler/github/README.md
new file mode 100644
index 0000000..de9c154
--- /dev/null
+++ b/crawler/github/README.md
@@ -0,0 +1,18 @@
+# GitHub organisations, user and repo reference into CyCAT.org
+
+## Usage
+
+~~~
+$ python3 github-importer.py --help
+usage: github-importer.py [-h] [-o ORG] [-f] [-r REPO]
+
+GitHub import for CyCAT
+
+optional arguments:
+  -h, --help            show this help message and exit
+  -o ORG, --org ORG     GitHub organisation (fallback to user if not existing
+                        as org) to import
+  -f, --full            Import all repositories as project in CyCAT
+  -r REPO, --repo REPO  Limit to a single GitHub repository import
+~~~
+
diff --git a/crawler/github/github-importer.py b/crawler/github/github-importer.py
new file mode 100644
index 0000000..889a34e
--- /dev/null
+++ b/crawler/github/github-importer.py
@@ -0,0 +1,88 @@
+import argparse
+import json
+import redis
+import os
+import requests
+import uuid
+parser = argparse.ArgumentParser(description='GitHub import for CyCAT')
+parser.add_argument('-o', '--org', help='GitHub organisation (fallback to user if not existing as org) to import')
+parser.add_argument('-f', '--full', help='Import all repositories as project in CyCAT', default=False, action='store_true')
+parser.add_argument('-r', '--repo', help='Limit to a single GitHub repository import', default=None)
+args = parser.parse_args()
+rdb = redis.Redis(host='127.0.0.1', port='3033')
+
+if not args.org:
+    parser.print_usage()
+    os.sys.exit(1)
+
+r = requests.get("https://api.github.com/orgs/{}".format(args.org))
+urlpath = 'orgs'
+org = r.json()
+if 'node_id' not in org:
+    r = requests.get("https://api.github.com/users/{}".format(args.org))
+    org = r.json()
+    urlpath = 'users'
+
+print(org['node_id'])
+
+# 39d6e10c-dac7-40e2-8e99-1ab1cefea6f4 (UUIDv5)
+
+orguuid = str(uuid.uuid5(uuid.UUID("39d6e10c-dac7-40e2-8e99-1ab1cefea6f4"),
+                     "{}".format(org['node_id'])))
+
+print(orguuid)
+
+rdb.set("u:{}".format(orguuid), 1)
+d = {"{}".format(orguuid): 1}
+k = "t:{}".format(1)
+rdb.zadd(k, d, nx=False)
+orgh = {}
+orgh['github:name'] = org['name']
+if 'description' in org:
+    orgh['github:description'] = org['description']
+if 'bio' in org:
+    orgh['github:bio'] = org['bio']
+orgh['github:html_url'] = org['html_url']
+if org['email'] is not None:
+    orgh['github:email'] = org['email']
+orgh['github:login'] = org['login']
+if org['location'] is not None:
+    orgh['github:location'] = org['location']
+orgh['cycat-oid'] = str(orguuid)
+print(orgh)
+rdb.hmset("{}:{}".format(1, orguuid), orgh)
+
+def cycatoid(node_id=None):
+    if node_id is None:
+        return False
+    _cycatoid = uuid.uuid5(uuid.UUID("39d6e10c-dac7-40e2-8e99-1ab1cefea6f4"), "{}".format(node_id))
+    return str(_cycatoid)
+
+if args.full:
+    r = requests.get("https://api.github.com/{}/{}/repos?per_page=100".format(urlpath, args.org))
+    for repo in r.json():
+        if args.repo is not None:
+            if args.repo != repo['name']:
+                print("Skip {}".format(repo['name']))
+                continue
+        print(repo)
+        projectuuid = cycatoid(node_id=repo['node_id'])
+        rdb.set("u:{}".format(projectuuid), 2)
+        d = {"{}".format(projectuuid): 1}
+        k = "t:{}".format(2)
+        rdb.zadd(k, d, nx=False)
+        print(repo['node_id'])
+        print(repo['name'])
+        print(cycatoid(node_id=repo['node_id']))
+        print(repo['description'])
+        print()
+        projecth = {}
+        projecth['github:name'] = repo['name']
+        if repo['description'] is None:
+            repo['description'] = ""
+        projecth['github:description'] = repo['description']
+        projecth['github:html_url'] = repo['html_url']
+        projecth['cycat-oid'] = str(projectuuid)
+        rdb.hmset("{}:{}".format(2, projectuuid), projecth)
+        rdb.sadd("parent:{}".format(projectuuid), orguuid)
+        rdb.sadd("child:{}".format(orguuid), projectuuid)
diff --git a/crawler/github/lists b/crawler/github/lists
new file mode 100644
index 0000000..54eb939
--- /dev/null
+++ b/crawler/github/lists
@@ -0,0 +1,3 @@
+misp
+mitre,cti
+Neo23x0