DRAFT: Add enterprise + multi-org collection support

extension/schema.json

@@ -6,6 +6,38 @@
     "namespace": "GH"
   },
   "node_kinds": [
+    {
+      "name": "GH_Enterprise",
+      "display_name": "GitHub Enterprise",
+      "description": "A GitHub Enterprise account that contains organizations, enterprise teams, roles, and managed users",
+      "is_display_kind": true,
+      "icon": "globe",
+      "color": "#6EA8FE"
+    },
+    {
+      "name": "GH_EnterpriseTeam",
+      "display_name": "GitHub Enterprise Team",
+      "description": "A team managed at the GitHub Enterprise level and assignable across organizations",
+      "is_display_kind": true,
+      "icon": "users-between-lines",
+      "color": "#9B8CFF"
+    },
+    {
+      "name": "GH_EnterpriseRole",
+      "display_name": "GitHub Enterprise Role",
+      "description": "The role a user or team has at the GitHub Enterprise level",
+      "is_display_kind": true,
+      "icon": "user-tie",
+      "color": "#B8D7FF"
+    },
+    {
+      "name": "GH_EnterpriseManagedUser",
+      "display_name": "GitHub Enterprise Managed User",
+      "description": "A GitHub Enterprise managed user account linked to an enterprise identity provider",
+      "is_display_kind": true,
+      "icon": "user-lock",
+      "color": "#7DD3FC"
+    },
     {
       "name": "GH_Organization",
       "display_name": "GitHub Organization",
@@ -150,6 +182,30 @@
       "icon": "lock-open",
       "color": "#E89B5C"
     },
+    {
+      "name": "GH_RunnerGroup",
+      "display_name": "GitHub Runner Group",
+      "description": "A GitHub self-hosted runner group that controls runner access and visibility",
+      "is_display_kind": true,
+      "icon": "server",
+      "color": "#94A3B8"
+    },
+    {
+      "name": "GH_OrgRunner",
+      "display_name": "GitHub Org Runner",
+      "description": "An organization-scoped GitHub self-hosted runner available to selected repositories or workflows",
+      "is_display_kind": true,
+      "icon": "microchip",
+      "color": "#22C55E"
+    },
+    {
+      "name": "GH_RepoRunner",
+      "display_name": "GitHub Repo Runner",
+      "description": "A repository-scoped GitHub self-hosted runner available to jobs in a single repository",
+      "is_display_kind": true,
+      "icon": "microchip",
+      "color": "#38BDF8"
+    },
     {
       "name": "GH_EnvironmentVariable",
       "display_name": "GitHub Environment Variable",

justfile

@@ -1,16 +1,16 @@
 set dotenv-load := true
 
-collect +args:
+collect +args='github /tmp/output/raw/':
     @echo "Collecting data"
-    uv run src/main.py collect github {{args}}
+    uv run src/main.py collect {{args}}
 
-preprocess +args:
-    @echo "Collecting data"
-    uv run src/main.py preprocess github {{args}}
+preprocess +args='github /tmp/output/raw/github':
+    @echo "Preprocessing data"
+    uv run openhound preprocess {{args}}
 
-convert +args:
+convert +args='github /tmp/output/raw/github /tmp/output/graph/github':
     @echo "Converting data"
-    uv run src/main.py convert github {{args}}
+    uv run openhound convert {{args}}
 
 sync:
     @echo "Syncing dependencies"

pyproject.toml

@@ -5,7 +5,7 @@ description = "Github collector for OpenHound"
 readme = "README.md"
 requires-python = ">=3.13"
 dependencies = [
-    "cryptography>=46.0.6",
+    "joserfc>=1.6.5",
     "requests>=2.33.0",
 ]
 authors = [
@@ -28,7 +28,7 @@ local_scheme = "no-local-version"
 
 [dependency-groups]
 dev = [
-    "openhound",
+    "openhound==0.1.4",
     "pre-commit>=4.5.1",
     "pytest>=9.0.1",
     "ruff>=0.15.5",