@nimbus-dev/connectors 0.2.1 → 0.2.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (215) hide show
  1. package/README.md +14 -4
  2. package/connectors/airflow/src/search-filter.ts +2 -15
  3. package/connectors/airflow/src/server.ts +2 -87
  4. package/connectors/airflow/src/tools.ts +74 -0
  5. package/connectors/apple/src/adapters.ts +326 -0
  6. package/connectors/apple/src/apple-mail-core.ts +4 -8
  7. package/connectors/apple/src/calendar-tools.ts +28 -8
  8. package/connectors/apple/src/server.ts +23 -512
  9. package/connectors/apple/src/tools.ts +18 -44
  10. package/connectors/argocd/src/server.ts +25 -38
  11. package/connectors/athena/src/tools.ts +15 -34
  12. package/connectors/aws/src/server.ts +2 -183
  13. package/connectors/aws/src/tools.ts +197 -0
  14. package/connectors/azure/src/server.ts +2 -131
  15. package/connectors/azure/src/tools.ts +133 -0
  16. package/connectors/bigeye/src/search-filter.ts +4 -15
  17. package/connectors/bigeye/src/server.ts +38 -38
  18. package/connectors/bigquery/src/tools.ts +1 -26
  19. package/connectors/bitbucket/src/server.ts +2 -244
  20. package/connectors/bitbucket/src/tools.ts +241 -0
  21. package/connectors/bitrise/src/server.ts +2 -79
  22. package/connectors/bitrise/src/tools.ts +76 -0
  23. package/connectors/canva/src/search-filter.ts +2 -11
  24. package/connectors/canva/src/server.ts +2 -73
  25. package/connectors/canva/src/tools.ts +66 -0
  26. package/connectors/circleci/src/server.ts +2 -176
  27. package/connectors/circleci/src/tools.ts +193 -0
  28. package/connectors/cloud-logging/src/tools.ts +14 -27
  29. package/connectors/cloudwatch/src/tools.ts +15 -34
  30. package/connectors/codemagic/src/server.ts +2 -70
  31. package/connectors/codemagic/src/tools.ts +71 -0
  32. package/connectors/confluence/src/server.ts +2 -301
  33. package/connectors/confluence/src/tools.ts +320 -0
  34. package/connectors/dagster/src/server.ts +2 -214
  35. package/connectors/dagster/src/tools.ts +187 -0
  36. package/connectors/databricks/src/search-filter.ts +7 -7
  37. package/connectors/databricks/src/server.ts +2 -77
  38. package/connectors/databricks/src/tools.ts +66 -0
  39. package/connectors/datadog/src/server.ts +2 -48
  40. package/connectors/datadog/src/tools.ts +48 -0
  41. package/connectors/dataprofile/src/profile.ts +16 -46
  42. package/connectors/dbt/src/server.ts +2 -92
  43. package/connectors/dbt/src/tools.ts +80 -0
  44. package/connectors/dependencytrack/src/server.ts +2 -76
  45. package/connectors/dependencytrack/src/tools.ts +65 -0
  46. package/connectors/discord/src/server.ts +2 -85
  47. package/connectors/discord/src/tools.ts +82 -0
  48. package/connectors/elasticsearch/src/tools.ts +7 -37
  49. package/connectors/fastmail/src/jmap-client.ts +202 -0
  50. package/connectors/fastmail/src/jmap-core.ts +8 -21
  51. package/connectors/fastmail/src/server.ts +2 -187
  52. package/connectors/fastmail/src/tools.ts +4 -23
  53. package/connectors/figma/src/search-filter.ts +2 -11
  54. package/connectors/figma/src/server.ts +2 -113
  55. package/connectors/figma/src/tools.ts +99 -0
  56. package/connectors/firebase/src/server.ts +2 -90
  57. package/connectors/firebase/src/token.ts +2 -5
  58. package/connectors/firebase/src/tools.ts +106 -0
  59. package/connectors/flagsmith/src/server.ts +2 -98
  60. package/connectors/flagsmith/src/tools.ts +95 -0
  61. package/connectors/flux/src/server.ts +52 -53
  62. package/connectors/gcp/src/server.ts +2 -132
  63. package/connectors/gcp/src/tools.ts +123 -0
  64. package/connectors/github/src/server.ts +2 -328
  65. package/connectors/github/src/tools.ts +311 -0
  66. package/connectors/github-actions/src/server.ts +2 -205
  67. package/connectors/github-actions/src/tools.ts +209 -0
  68. package/connectors/gitlab/src/server.ts +2 -384
  69. package/connectors/gitlab/src/tools.ts +392 -0
  70. package/connectors/gmail/src/server.ts +2 -270
  71. package/connectors/gmail/src/tools.ts +209 -0
  72. package/connectors/google-drive/src/server.ts +2 -524
  73. package/connectors/google-drive/src/tools.ts +529 -0
  74. package/connectors/google-meet/src/server.ts +2 -82
  75. package/connectors/google-meet/src/tools.ts +87 -0
  76. package/connectors/google-photos/src/server.ts +2 -132
  77. package/connectors/google-photos/src/tools.ts +138 -0
  78. package/connectors/grafana/src/server.ts +2 -53
  79. package/connectors/grafana/src/tools.ts +53 -0
  80. package/connectors/great-expectations/src/gx-parse.ts +14 -45
  81. package/connectors/greenhouse/src/server.ts +2 -63
  82. package/connectors/greenhouse/src/tools.ts +68 -0
  83. package/connectors/hubspot/src/server.ts +2 -77
  84. package/connectors/hubspot/src/tools.ts +70 -0
  85. package/connectors/iac/src/server.ts +2 -137
  86. package/connectors/iac/src/tools.ts +152 -0
  87. package/connectors/imap/src/imap-core.ts +29 -82
  88. package/connectors/imap/src/server.ts +33 -220
  89. package/connectors/intercom/src/server.ts +2 -61
  90. package/connectors/intercom/src/tools.ts +58 -0
  91. package/connectors/jenkins/src/jenkins-api.ts +2 -6
  92. package/connectors/jenkins/src/server.ts +2 -215
  93. package/connectors/jenkins/src/tools.ts +240 -0
  94. package/connectors/jira/src/server.ts +2 -301
  95. package/connectors/jira/src/tools.ts +325 -0
  96. package/connectors/kubernetes/src/server.ts +2 -155
  97. package/connectors/kubernetes/src/tools.ts +159 -0
  98. package/connectors/launchdarkly/src/server.ts +2 -75
  99. package/connectors/launchdarkly/src/tools.ts +72 -0
  100. package/connectors/lever/src/search-filter.ts +4 -12
  101. package/connectors/lever/src/server.ts +2 -60
  102. package/connectors/lever/src/tools.ts +61 -0
  103. package/connectors/linear/src/server.ts +2 -459
  104. package/connectors/linear/src/tools.ts +482 -0
  105. package/connectors/localdb/src/sql-scan.ts +13 -41
  106. package/connectors/localdb/src/tools.ts +1 -1
  107. package/connectors/looker/src/search-filter.ts +2 -15
  108. package/connectors/looker/src/server.ts +35 -70
  109. package/connectors/mendeley/src/server.ts +2 -55
  110. package/connectors/mendeley/src/tools.ts +54 -0
  111. package/connectors/mercury/src/server.ts +2 -57
  112. package/connectors/mercury/src/tools.ts +50 -0
  113. package/connectors/metabase/src/search-filter.ts +4 -15
  114. package/connectors/metabase/src/server.ts +2 -79
  115. package/connectors/metabase/src/tools.ts +64 -0
  116. package/connectors/miro/src/server.ts +2 -72
  117. package/connectors/miro/src/tools.ts +65 -0
  118. package/connectors/mlflow/src/search-filter.ts +3 -7
  119. package/connectors/mlflow/src/server.ts +20 -40
  120. package/connectors/monte-carlo/src/server.ts +50 -57
  121. package/connectors/netlify/src/search-filter.ts +6 -14
  122. package/connectors/netlify/src/server.ts +2 -60
  123. package/connectors/netlify/src/tools.ts +53 -0
  124. package/connectors/newrelic/src/server.ts +2 -40
  125. package/connectors/newrelic/src/tools.ts +37 -0
  126. package/connectors/notion/src/server.ts +2 -335
  127. package/connectors/notion/src/tools.ts +344 -0
  128. package/connectors/obsidian/src/server.ts +2 -425
  129. package/connectors/obsidian/src/tools.ts +466 -0
  130. package/connectors/onedrive/src/server.ts +2 -249
  131. package/connectors/onedrive/src/tools.ts +253 -0
  132. package/connectors/outlook/src/server.ts +2 -351
  133. package/connectors/outlook/src/tools.ts +357 -0
  134. package/connectors/pagerduty/src/server.ts +2 -132
  135. package/connectors/pagerduty/src/tools.ts +126 -0
  136. package/connectors/pipedrive/src/server.ts +2 -58
  137. package/connectors/pipedrive/src/tools.ts +63 -0
  138. package/connectors/powerbi/src/search-filter.ts +2 -15
  139. package/connectors/powerbi/src/server.ts +43 -68
  140. package/connectors/prefect/src/search-filter.ts +3 -16
  141. package/connectors/prefect/src/server.ts +2 -90
  142. package/connectors/prefect/src/tools.ts +83 -0
  143. package/connectors/protonmail/src/mail-core.ts +16 -75
  144. package/connectors/protonmail/src/server.ts +34 -214
  145. package/connectors/raindrop/src/server.ts +2 -103
  146. package/connectors/raindrop/src/tools.ts +103 -0
  147. package/connectors/ramp/src/server.ts +2 -105
  148. package/connectors/ramp/src/tools.ts +98 -0
  149. package/connectors/readwise/src/server.ts +2 -88
  150. package/connectors/readwise/src/tools.ts +81 -0
  151. package/connectors/sagemaker/src/tools.ts +19 -38
  152. package/connectors/salesforce/src/search-filter.ts +4 -11
  153. package/connectors/salesforce/src/server.ts +2 -90
  154. package/connectors/salesforce/src/tools.ts +75 -0
  155. package/connectors/semgrep/src/search-filter.ts +3 -10
  156. package/connectors/semgrep/src/server.ts +2 -104
  157. package/connectors/semgrep/src/tools.ts +92 -0
  158. package/connectors/sentry/src/server.ts +2 -58
  159. package/connectors/sentry/src/tools.ts +53 -0
  160. package/connectors/slack/src/server.ts +2 -352
  161. package/connectors/slack/src/tools.ts +376 -0
  162. package/connectors/snowflake/src/search-filter.ts +4 -19
  163. package/connectors/snowflake/src/server.ts +6 -25
  164. package/connectors/snyk/src/server.ts +2 -140
  165. package/connectors/snyk/src/tools.ts +136 -0
  166. package/connectors/sonarqube/src/server.ts +2 -111
  167. package/connectors/sonarqube/src/tools.ts +98 -0
  168. package/connectors/stackoverflow/src/server.ts +2 -77
  169. package/connectors/stackoverflow/src/tools.ts +66 -0
  170. package/connectors/storybook/src/storybook-parse.ts +3 -6
  171. package/connectors/stripe/src/server.ts +2 -61
  172. package/connectors/stripe/src/tools.ts +54 -0
  173. package/connectors/superset/src/search-filter.ts +4 -15
  174. package/connectors/superset/src/server.ts +2 -115
  175. package/connectors/superset/src/tools.ts +98 -0
  176. package/connectors/tableau/src/search-filter.ts +2 -15
  177. package/connectors/tableau/src/server.ts +13 -65
  178. package/connectors/teams/src/server.ts +2 -249
  179. package/connectors/teams/src/tools.ts +283 -0
  180. package/connectors/testflight/src/server.ts +2 -70
  181. package/connectors/testflight/src/tools.ts +79 -0
  182. package/connectors/vercel/src/search-filter.ts +2 -10
  183. package/connectors/vercel/src/server.ts +2 -75
  184. package/connectors/vercel/src/tools.ts +73 -0
  185. package/connectors/vertex-ai/src/tools.ts +23 -33
  186. package/connectors/wiz/src/search-filter.ts +6 -30
  187. package/connectors/wiz/src/server.ts +2 -188
  188. package/connectors/wiz/src/tools.ts +168 -0
  189. package/connectors/zendesk/src/server.ts +2 -73
  190. package/connectors/zendesk/src/tools.ts +60 -0
  191. package/connectors/zoom/src/search-filter.ts +1 -7
  192. package/connectors/zoom/src/server.ts +2 -94
  193. package/connectors/zoom/src/tools.ts +93 -0
  194. package/connectors/zotero/src/server.ts +2 -78
  195. package/connectors/zotero/src/tools.ts +67 -0
  196. package/package.json +14 -12
  197. package/shared/access-token-cache.ts +123 -0
  198. package/shared/atlassian-json-fetch.ts +6 -16
  199. package/shared/audit-chain.ts +13 -6
  200. package/shared/cli-json-kit.ts +150 -0
  201. package/shared/collection-tool-kit.ts +143 -0
  202. package/shared/connector-mode.ts +3 -2
  203. package/shared/cursor-list-tool.ts +29 -0
  204. package/shared/env-json-api.ts +209 -0
  205. package/shared/fetch-json-text.ts +27 -0
  206. package/shared/github-rest.ts +62 -0
  207. package/shared/graphql-json.ts +60 -0
  208. package/shared/imap-mail-core.ts +91 -0
  209. package/shared/imap-tool-kit.ts +57 -23
  210. package/shared/imapflow-adapter.ts +417 -0
  211. package/shared/rest-tool-kit.ts +77 -7
  212. package/shared/run-cli-json.ts +38 -16
  213. package/shared/search-filter.ts +58 -18
  214. package/shared/walk-files.ts +118 -0
  215. package/standalone/src/launcher.ts +19 -7
@@ -1,218 +1,6 @@
1
- import { z } from "zod";
2
- import { searchToolInputSchema } from "../../../shared/mcp-search-tool.ts";
3
- import { mcpJsonResult as jsonResult } from "../../../shared/mcp-tool-kit.ts";
4
1
  import { runReadOnlyMcpConnector } from "../../../shared/run-read-only-mcp-connector.ts";
5
- import { filterDagsterJobs } from "./search-filter.ts";
6
-
7
- function trimTrailingSlash(s: string): string {
8
- return s.endsWith("/") ? s.slice(0, -1) : s;
9
- }
10
-
11
- function apiBase(): string {
12
- const v = process.env["DAGSTER_BASE_URL"]?.trim();
13
- if (v === undefined || v === "") {
14
- throw new Error("DAGSTER_BASE_URL is not set");
15
- }
16
- return trimTrailingSlash(v);
17
- }
18
-
19
- function apiToken(): string {
20
- // Required for Dagster Cloud; self-hosted OSS may use a placeholder. The
21
- // gateway injects it regardless to keep spawn wiring uniform.
22
- const v = process.env["DAGSTER_API_TOKEN"]?.trim();
23
- if (v === undefined || v === "") {
24
- throw new Error("DAGSTER_API_TOKEN is not set");
25
- }
26
- return v;
27
- }
28
-
29
- const JOBS_QUERY = `
30
- query NimbusJobs {
31
- repositoriesOrError {
32
- __typename
33
- ... on RepositoryConnection {
34
- nodes {
35
- name
36
- location { name }
37
- pipelines { id name description isJob tags { key value } }
38
- }
39
- }
40
- ... on PythonError { message }
41
- }
42
- }
43
- `;
44
-
45
- async function dagsterGraphql<T>(query: string): Promise<T> {
46
- const res = await fetch(`${apiBase()}/graphql`, {
47
- method: "POST",
48
- headers: {
49
- "Dagster-Cloud-Api-Token": apiToken(),
50
- "Content-Type": "application/json",
51
- Accept: "application/json",
52
- },
53
- body: JSON.stringify({ query }),
54
- });
55
- const text = await res.text();
56
- if (!res.ok) {
57
- throw new Error(`Dagster ${String(res.status)}: ${text.slice(0, 400)}`);
58
- }
59
- const parsed = JSON.parse(text) as { data?: T; errors?: unknown };
60
- if (parsed.errors !== undefined) {
61
- throw new Error(`Dagster GraphQL error: ${JSON.stringify(parsed.errors).slice(0, 400)}`);
62
- }
63
- if (parsed.data === undefined) {
64
- throw new Error("Dagster GraphQL: response missing `data` field");
65
- }
66
- return parsed.data;
67
- }
68
-
69
- interface FlatJob {
70
- name: string;
71
- repository: string;
72
- location: string | null;
73
- id: string | null;
74
- description: string | null;
75
- isJob: boolean;
76
- tags: Array<{ key: string; value: string }>;
77
- }
78
-
79
- function asRecord(v: unknown): Record<string, unknown> | undefined {
80
- return v !== null && typeof v === "object" && !Array.isArray(v)
81
- ? (v as Record<string, unknown>)
82
- : undefined;
83
- }
84
-
85
- function str(row: Record<string, unknown>, key: string): string | null {
86
- const v = row[key];
87
- return typeof v === "string" && v !== "" ? v : null;
88
- }
89
-
90
- function tagPairs(raw: unknown): Array<{ key: string; value: string }> {
91
- if (!Array.isArray(raw)) {
92
- return [];
93
- }
94
- const out: Array<{ key: string; value: string }> = [];
95
- for (const t of raw) {
96
- const tr = asRecord(t);
97
- if (tr === undefined) {
98
- continue;
99
- }
100
- const key = str(tr, "key");
101
- if (key === null) {
102
- continue;
103
- }
104
- out.push({ key, value: str(tr, "value") ?? "" });
105
- }
106
- return out;
107
- }
108
-
109
- function pipelinesOf(node: unknown): {
110
- repository: string;
111
- location: string | null;
112
- rows: unknown[];
113
- } {
114
- const repo = asRecord(node);
115
- if (repo === undefined) {
116
- return { repository: "", location: null, rows: [] };
117
- }
118
- const locationRow = asRecord(repo["location"]);
119
- const pipelines = repo["pipelines"];
120
- return {
121
- repository: str(repo, "name") ?? "",
122
- location: locationRow === undefined ? null : str(locationRow, "name"),
123
- rows: Array.isArray(pipelines) ? pipelines : [],
124
- };
125
- }
126
-
127
- function pipelineToJob(p: unknown, repository: string, location: string | null): FlatJob | null {
128
- const pr = asRecord(p);
129
- if (pr === undefined) {
130
- return null;
131
- }
132
- const name = str(pr, "name");
133
- if (name === null) {
134
- return null;
135
- }
136
- return {
137
- name,
138
- repository,
139
- location,
140
- id: str(pr, "id"),
141
- description: str(pr, "description"),
142
- isJob: pr["isJob"] === true,
143
- tags: tagPairs(pr["tags"]),
144
- };
145
- }
146
-
147
- function flattenJobs(data: unknown): FlatJob[] {
148
- const root = asRecord(data) ?? {};
149
- const repos = asRecord(root["repositoriesOrError"]) ?? {};
150
- if (repos["__typename"] === "PythonError") {
151
- const message = str(repos, "message") ?? "unknown error";
152
- throw new Error(`Dagster repositoriesOrError PythonError: ${message}`);
153
- }
154
- const nodes = repos["nodes"];
155
- if (!Array.isArray(nodes)) {
156
- return [];
157
- }
158
- const out: FlatJob[] = [];
159
- for (const node of nodes) {
160
- const { repository, location, rows } = pipelinesOf(node);
161
- for (const p of rows) {
162
- const job = pipelineToJob(p, repository, location);
163
- if (job !== null) {
164
- out.push(job);
165
- }
166
- }
167
- }
168
- return out;
169
- }
170
-
171
- async function listJobs(): Promise<FlatJob[]> {
172
- return flattenJobs(await dagsterGraphql<unknown>(JOBS_QUERY));
173
- }
2
+ import { registerDagsterTools } from "./tools.ts";
174
3
 
175
4
  await runReadOnlyMcpConnector("nimbus-dagster", (reg) => {
176
- reg(
177
- "dagster_list",
178
- "List Dagster jobs. Walks the repositories catalog via a single GraphQL query (repositoriesOrError → nodes[].pipelines[]) against POST <base_url>/graphql and flattens to jobs, each carrying its repository and code location. Returns an { items: [...] } envelope.",
179
- z.object({
180
- limit: z.number().int().min(1).max(2000).optional(),
181
- }),
182
- async (p) => {
183
- const jobs = await listJobs();
184
- const limited = p.limit === undefined ? jobs : jobs.slice(0, p.limit);
185
- return jsonResult({ items: limited });
186
- },
187
- );
188
-
189
- reg(
190
- "dagster_get",
191
- "Fetch one Dagster job by its stable id. The id is either the full location:repository:jobName triple or just the bare jobName (the first match wins when only a name is given). Throws when no job matches.",
192
- z.object({
193
- id: z.string().min(1),
194
- }),
195
- async (p) => {
196
- const jobs = await listJobs();
197
- const match = jobs.find((j) => {
198
- const triple = `${j.location ?? "_"}:${j.repository}:${j.name}`;
199
- return triple === p.id || j.name === p.id;
200
- });
201
- if (match === undefined) {
202
- throw new Error(`Dagster: job ${p.id} not found`);
203
- }
204
- return jsonResult(match);
205
- },
206
- );
207
-
208
- reg(
209
- "dagster_search",
210
- "Substring search across Dagster jobs. Matches the query (case-insensitive) against the job name, repository, code location, description, and tag keys/values. Returns a { matches: [...] } envelope.",
211
- searchToolInputSchema(2000),
212
- async (p) => {
213
- const jobs = await listJobs();
214
- const matches = filterDagsterJobs(jobs, { query: p.query, limit: p.limit });
215
- return jsonResult({ matches });
216
- },
217
- );
5
+ registerDagsterTools(reg);
218
6
  });
@@ -0,0 +1,187 @@
1
+ import { z } from "zod";
2
+ import { requiredBaseUrl, requiredEnv } from "../../../shared/env-json-api.ts";
3
+ import { postGraphql } from "../../../shared/graphql-json.ts";
4
+ import { searchToolInputSchema } from "../../../shared/mcp-search-tool.ts";
5
+ import { mcpJsonResult as jsonResult } from "../../../shared/mcp-tool-kit.ts";
6
+ import type { ZodToolRegistrar } from "../../../shared/run-read-only-mcp-connector.ts";
7
+ import { asRecord } from "../../../shared/search-filter.ts";
8
+ import { filterDagsterJobs } from "./search-filter.ts";
9
+
10
+ function apiToken(): string {
11
+ // Required for Dagster Cloud; self-hosted OSS may use a placeholder. The
12
+ // gateway injects it regardless to keep spawn wiring uniform.
13
+ return requiredEnv("DAGSTER_API_TOKEN");
14
+ }
15
+
16
+ const JOBS_QUERY = `
17
+ query NimbusJobs {
18
+ repositoriesOrError {
19
+ __typename
20
+ ... on RepositoryConnection {
21
+ nodes {
22
+ name
23
+ location { name }
24
+ pipelines { id name description isJob tags { key value } }
25
+ }
26
+ }
27
+ ... on PythonError { message }
28
+ }
29
+ }
30
+ `;
31
+
32
+ async function dagsterGraphql<T>(query: string): Promise<T> {
33
+ return postGraphql<T>({
34
+ url: `${requiredBaseUrl("DAGSTER_BASE_URL")}/graphql`,
35
+ label: "Dagster",
36
+ headers: { "Dagster-Cloud-Api-Token": apiToken() },
37
+ query,
38
+ });
39
+ }
40
+
41
+ interface FlatJob {
42
+ name: string;
43
+ repository: string;
44
+ location: string | null;
45
+ id: string | null;
46
+ description: string | null;
47
+ isJob: boolean;
48
+ tags: Array<{ key: string; value: string }>;
49
+ }
50
+
51
+ function str(row: Record<string, unknown>, key: string): string | null {
52
+ const v = row[key];
53
+ return typeof v === "string" && v !== "" ? v : null;
54
+ }
55
+
56
+ function tagPairs(raw: unknown): Array<{ key: string; value: string }> {
57
+ if (!Array.isArray(raw)) {
58
+ return [];
59
+ }
60
+ const out: Array<{ key: string; value: string }> = [];
61
+ for (const t of raw) {
62
+ const tr = asRecord(t);
63
+ if (tr === undefined) {
64
+ continue;
65
+ }
66
+ const key = str(tr, "key");
67
+ if (key === null) {
68
+ continue;
69
+ }
70
+ out.push({ key, value: str(tr, "value") ?? "" });
71
+ }
72
+ return out;
73
+ }
74
+
75
+ function pipelinesOf(node: unknown): {
76
+ repository: string;
77
+ location: string | null;
78
+ rows: unknown[];
79
+ } {
80
+ const repo = asRecord(node);
81
+ if (repo === undefined) {
82
+ return { repository: "", location: null, rows: [] };
83
+ }
84
+ const locationRow = asRecord(repo["location"]);
85
+ const pipelines = repo["pipelines"];
86
+ return {
87
+ repository: str(repo, "name") ?? "",
88
+ location: locationRow === undefined ? null : str(locationRow, "name"),
89
+ rows: Array.isArray(pipelines) ? pipelines : [],
90
+ };
91
+ }
92
+
93
+ function pipelineToJob(p: unknown, repository: string, location: string | null): FlatJob | null {
94
+ const pr = asRecord(p);
95
+ if (pr === undefined) {
96
+ return null;
97
+ }
98
+ const name = str(pr, "name");
99
+ if (name === null) {
100
+ return null;
101
+ }
102
+ return {
103
+ name,
104
+ repository,
105
+ location,
106
+ id: str(pr, "id"),
107
+ description: str(pr, "description"),
108
+ isJob: pr["isJob"] === true,
109
+ tags: tagPairs(pr["tags"]),
110
+ };
111
+ }
112
+
113
+ function flattenJobs(data: unknown): FlatJob[] {
114
+ const root = asRecord(data) ?? {};
115
+ const repos = asRecord(root["repositoriesOrError"]) ?? {};
116
+ if (repos["__typename"] === "PythonError") {
117
+ const message = str(repos, "message") ?? "unknown error";
118
+ throw new Error(`Dagster repositoriesOrError PythonError: ${message}`);
119
+ }
120
+ const nodes = repos["nodes"];
121
+ if (!Array.isArray(nodes)) {
122
+ return [];
123
+ }
124
+ const out: FlatJob[] = [];
125
+ for (const node of nodes) {
126
+ const { repository, location, rows } = pipelinesOf(node);
127
+ for (const p of rows) {
128
+ const job = pipelineToJob(p, repository, location);
129
+ if (job !== null) {
130
+ out.push(job);
131
+ }
132
+ }
133
+ }
134
+ return out;
135
+ }
136
+
137
+ async function listJobs(): Promise<FlatJob[]> {
138
+ return flattenJobs(await dagsterGraphql<unknown>(JOBS_QUERY));
139
+ }
140
+
141
+ /** Tool names exposed by this connector — for contract/introspection tests. */
142
+ export const DAGSTER_TOOL_NAMES = ["dagster_get", "dagster_list", "dagster_search"] as const;
143
+
144
+ export function registerDagsterTools(reg: ZodToolRegistrar): void {
145
+ reg(
146
+ "dagster_list",
147
+ "List Dagster jobs. Walks the repositories catalog via a single GraphQL query (repositoriesOrError → nodes[].pipelines[]) against POST <base_url>/graphql and flattens to jobs, each carrying its repository and code location. Returns an { items: [...] } envelope.",
148
+ z.object({
149
+ limit: z.number().int().min(1).max(2000).optional(),
150
+ }),
151
+ async (p) => {
152
+ const jobs = await listJobs();
153
+ const limited = p.limit === undefined ? jobs : jobs.slice(0, p.limit);
154
+ return jsonResult({ items: limited });
155
+ },
156
+ );
157
+
158
+ reg(
159
+ "dagster_get",
160
+ "Fetch one Dagster job by its stable id. The id is either the full location:repository:jobName triple or just the bare jobName (the first match wins when only a name is given). Throws when no job matches.",
161
+ z.object({
162
+ id: z.string().min(1),
163
+ }),
164
+ async (p) => {
165
+ const jobs = await listJobs();
166
+ const match = jobs.find((j) => {
167
+ const triple = `${j.location ?? "_"}:${j.repository}:${j.name}`;
168
+ return triple === p.id || j.name === p.id;
169
+ });
170
+ if (match === undefined) {
171
+ throw new Error(`Dagster: job ${p.id} not found`);
172
+ }
173
+ return jsonResult(match);
174
+ },
175
+ );
176
+
177
+ reg(
178
+ "dagster_search",
179
+ "Substring search across Dagster jobs. Matches the query (case-insensitive) against the job name, repository, code location, description, and tag keys/values. Returns a { matches: [...] } envelope.",
180
+ searchToolInputSchema(2000),
181
+ async (p) => {
182
+ const jobs = await listJobs();
183
+ const matches = filterDagsterJobs(jobs, { query: p.query, limit: p.limit });
184
+ return jsonResult({ matches });
185
+ },
186
+ );
187
+ }
@@ -1,16 +1,13 @@
1
1
  import {
2
2
  asRecord,
3
3
  makeQueryFilter,
4
+ nestedString,
4
5
  type SearchMatchOptions,
6
+ stringField,
5
7
  } from "../../../shared/search-filter.ts";
6
8
 
7
9
  export type DatabricksSearchMatchOptions = SearchMatchOptions;
8
10
 
9
- function stringAt(row: Record<string, unknown>, key: string): string {
10
- const v = row[key];
11
- return typeof v === "string" ? v : "";
12
- }
13
-
14
11
  function numberAt(row: Record<string, unknown>, key: string): string {
15
12
  const v = row[key];
16
13
  return typeof v === "number" && Number.isFinite(v) ? String(v) : "";
@@ -21,8 +18,11 @@ function fieldsOf(item: unknown): readonly string[] | null {
21
18
  if (row === undefined) {
22
19
  return null;
23
20
  }
24
- const settings = asRecord(row["settings"]) ?? {};
25
- return [stringAt(settings, "name"), stringAt(row, "creator_user_name"), numberAt(row, "job_id")];
21
+ return [
22
+ nestedString(row, ["settings", "name"]),
23
+ stringField(row, "creator_user_name"),
24
+ numberAt(row, "job_id"),
25
+ ];
26
26
  }
27
27
 
28
28
  export const filterDatabricksJobs = makeQueryFilter(fieldsOf);
@@ -1,81 +1,6 @@
1
- import { z } from "zod";
2
- import { searchToolInputSchema } from "../../../shared/mcp-search-tool.ts";
3
- import { mcpJsonResult as jsonResult } from "../../../shared/mcp-tool-kit.ts";
4
1
  import { runReadOnlyMcpConnector } from "../../../shared/run-read-only-mcp-connector.ts";
5
- import { filterDatabricksJobs } from "./search-filter.ts";
6
-
7
- function trimTrailingSlash(s: string): string {
8
- return s.endsWith("/") ? s.slice(0, -1) : s;
9
- }
10
-
11
- function apiBase(): string {
12
- const v = process.env["DATABRICKS_HOST"]?.trim();
13
- if (v === undefined || v === "") {
14
- throw new Error("DATABRICKS_HOST is not set");
15
- }
16
- return trimTrailingSlash(v);
17
- }
18
-
19
- function authHeader(): Record<string, string> {
20
- const k = process.env["DATABRICKS_TOKEN"]?.trim();
21
- if (k === undefined || k === "") {
22
- throw new Error("DATABRICKS_TOKEN is not set");
23
- }
24
- return { Authorization: `Bearer ${k}`, Accept: "application/json" };
25
- }
26
-
27
- async function dbGet(path: string): Promise<unknown> {
28
- const res = await fetch(`${apiBase()}${path}`, { headers: authHeader() });
29
- const text = await res.text();
30
- if (!res.ok) {
31
- throw new Error(`Databricks ${String(res.status)}: ${text.slice(0, 400)}`);
32
- }
33
- return JSON.parse(text) as unknown;
34
- }
35
-
36
- function jobsFrom(root: unknown): unknown[] {
37
- const jobs = (root as { jobs?: unknown } | null)?.jobs;
38
- return Array.isArray(jobs) ? jobs : [];
39
- }
2
+ import { registerDatabricksTools } from "./tools.ts";
40
3
 
41
4
  await runReadOnlyMcpConnector("nimbus-databricks", (reg) => {
42
- reg(
43
- "databricks_list",
44
- "List Databricks jobs (`GET /api/2.1/jobs/list`). Returns a single page; `limit` (1..100, default 25) caps the page size. Returns the raw `{ jobs, has_more, next_page_token }` envelope.",
45
- z.object({
46
- limit: z.number().int().min(1).max(100).optional(),
47
- }),
48
- async (p) => {
49
- const limit = p.limit ?? 25;
50
- const params = new URLSearchParams({ limit: String(limit) });
51
- return jsonResult(await dbGet(`/api/2.1/jobs/list?${params.toString()}`));
52
- },
53
- );
54
-
55
- reg(
56
- "databricks_get",
57
- "Fetch one Databricks job by numeric `jobId` (`GET /api/2.1/jobs/get?job_id={jobId}`). Throws when no job with that id exists.",
58
- z.object({
59
- jobId: z.number().int(),
60
- }),
61
- async (p) => {
62
- const params = new URLSearchParams({ job_id: String(p.jobId) });
63
- return jsonResult(await dbGet(`/api/2.1/jobs/get?${params.toString()}`));
64
- },
65
- );
66
-
67
- reg(
68
- "databricks_search",
69
- "Substring search across Databricks jobs. Matches the query (case-insensitive) against the job's `settings.name`, `creator_user_name`, and `job_id`. Returns a `{ matches: [...] }` envelope.",
70
- searchToolInputSchema(100),
71
- async (p) => {
72
- const params = new URLSearchParams({ limit: "100" });
73
- const root = await dbGet(`/api/2.1/jobs/list?${params.toString()}`);
74
- const matches = filterDatabricksJobs(jobsFrom(root), {
75
- query: p.query,
76
- limit: p.limit,
77
- });
78
- return jsonResult({ matches });
79
- },
80
- );
5
+ registerDatabricksTools(reg);
81
6
  });
@@ -0,0 +1,66 @@
1
+ import { z } from "zod";
2
+ import { createJsonGetter, envAuthHeaders, requiredBaseUrl } from "../../../shared/env-json-api.ts";
3
+ import { searchToolInputSchema } from "../../../shared/mcp-search-tool.ts";
4
+ import { mcpJsonResult as jsonResult } from "../../../shared/mcp-tool-kit.ts";
5
+ import type { ZodToolRegistrar } from "../../../shared/run-read-only-mcp-connector.ts";
6
+ import { filterDatabricksJobs } from "./search-filter.ts";
7
+
8
+ const dbGet = createJsonGetter({
9
+ base: () => requiredBaseUrl("DATABRICKS_HOST"),
10
+ label: "Databricks",
11
+ headers: envAuthHeaders({ env: "DATABRICKS_TOKEN" }),
12
+ });
13
+
14
+ function jobsFrom(root: unknown): unknown[] {
15
+ const jobs = (root as { jobs?: unknown } | null)?.jobs;
16
+ return Array.isArray(jobs) ? jobs : [];
17
+ }
18
+
19
+ /** Tool names exposed by this connector — for contract/introspection tests. */
20
+ export const DATABRICKS_TOOL_NAMES = [
21
+ "databricks_list",
22
+ "databricks_get",
23
+ "databricks_search",
24
+ ] as const;
25
+
26
+ export function registerDatabricksTools(reg: ZodToolRegistrar): void {
27
+ reg(
28
+ "databricks_list",
29
+ "List Databricks jobs (`GET /api/2.1/jobs/list`). Returns a single page; `limit` (1..100, default 25) caps the page size. Returns the raw `{ jobs, has_more, next_page_token }` envelope.",
30
+ z.object({
31
+ limit: z.number().int().min(1).max(100).optional(),
32
+ }),
33
+ async (p) => {
34
+ const limit = p.limit ?? 25;
35
+ const params = new URLSearchParams({ limit: String(limit) });
36
+ return jsonResult(await dbGet(`/api/2.1/jobs/list?${params.toString()}`));
37
+ },
38
+ );
39
+
40
+ reg(
41
+ "databricks_get",
42
+ "Fetch one Databricks job by numeric `jobId` (`GET /api/2.1/jobs/get?job_id={jobId}`). Throws when no job with that id exists.",
43
+ z.object({
44
+ jobId: z.number().int(),
45
+ }),
46
+ async (p) => {
47
+ const params = new URLSearchParams({ job_id: String(p.jobId) });
48
+ return jsonResult(await dbGet(`/api/2.1/jobs/get?${params.toString()}`));
49
+ },
50
+ );
51
+
52
+ reg(
53
+ "databricks_search",
54
+ "Substring search across Databricks jobs. Matches the query (case-insensitive) against the job's `settings.name`, `creator_user_name`, and `job_id`. Returns a `{ matches: [...] }` envelope.",
55
+ searchToolInputSchema(100),
56
+ async (p) => {
57
+ const params = new URLSearchParams({ limit: "100" });
58
+ const root = await dbGet(`/api/2.1/jobs/list?${params.toString()}`);
59
+ const matches = filterDatabricksJobs(jobsFrom(root), {
60
+ query: p.query,
61
+ limit: p.limit,
62
+ });
63
+ return jsonResult({ matches });
64
+ },
65
+ );
66
+ }
@@ -1,56 +1,10 @@
1
1
  import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
2
2
  import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
3
- import { z } from "zod";
4
-
5
- import {
6
- createRegisterSimpleTool,
7
- createZodToolRegistrar,
8
- mcpJsonResult as jsonResult,
9
- } from "../../../shared/mcp-tool-kit.ts";
10
-
11
- function siteHost(): string {
12
- const s = process.env["DD_SITE"]?.trim() || "datadoghq.com";
13
- return `api.${s}`;
14
- }
15
-
16
- function headers(): Record<string, string> {
17
- const ak = process.env["DD_API_KEY"]?.trim();
18
- const app = process.env["DD_APP_KEY"]?.trim();
19
- if (ak === undefined || ak === "" || app === undefined || app === "") {
20
- throw new Error("DD_API_KEY and DD_APP_KEY must be set");
21
- }
22
- return {
23
- "DD-API-KEY": ak,
24
- "DD-APPLICATION-KEY": app,
25
- Accept: "application/json",
26
- };
27
- }
28
-
29
- async function ddGet(path: string): Promise<unknown> {
30
- const res = await fetch(`https://${siteHost()}${path}`, { headers: headers() });
31
- const text = await res.text();
32
- if (!res.ok) {
33
- throw new Error(`Datadog ${String(res.status)}: ${text.slice(0, 400)}`);
34
- }
35
- return JSON.parse(text) as unknown;
36
- }
3
+ import { registerDatadogTools } from "./tools.ts";
37
4
 
38
5
  const mcp = new McpServer({ name: "nimbus-datadog", version: "0.1.0" });
39
- const reg = createZodToolRegistrar(createRegisterSimpleTool(mcp));
40
-
41
- reg("datadog_monitor_list", "List monitors.", z.object({}), async () =>
42
- jsonResult(await ddGet("/api/v1/monitor")),
43
- );
44
6
 
45
- reg(
46
- "datadog_incident_list",
47
- "List incidents (v2).",
48
- z.object({ limit: z.number().int().min(1).max(50).optional() }),
49
- async (p) => {
50
- const lim = p.limit ?? 10;
51
- return jsonResult(await ddGet(`/api/v2/incidents?page[size]=${String(lim)}`));
52
- },
53
- );
7
+ registerDatadogTools(mcp);
54
8
 
55
9
  const transport = new StdioServerTransport();
56
10
  await mcp.connect(transport);
@@ -0,0 +1,48 @@
1
+ import { z } from "zod";
2
+ import { createJsonGetter, optionalEnv } from "../../../shared/env-json-api.ts";
3
+ import {
4
+ createRegisterSimpleTool,
5
+ createZodToolRegistrar,
6
+ mcpJsonResult as jsonResult,
7
+ } from "../../../shared/mcp-tool-kit.ts";
8
+
9
+ /** Both keys or neither: one alone authenticates nothing, so the refusal names the pair. */
10
+ function headers(): Record<string, string> {
11
+ const ak = process.env["DD_API_KEY"]?.trim();
12
+ const app = process.env["DD_APP_KEY"]?.trim();
13
+ if (ak === undefined || ak === "" || app === undefined || app === "") {
14
+ throw new Error("DD_API_KEY and DD_APP_KEY must be set");
15
+ }
16
+ return {
17
+ "DD-API-KEY": ak,
18
+ "DD-APPLICATION-KEY": app,
19
+ Accept: "application/json",
20
+ };
21
+ }
22
+
23
+ const ddGet = createJsonGetter({
24
+ base: () => `https://api.${optionalEnv("DD_SITE", "datadoghq.com")}`,
25
+ label: "Datadog",
26
+ headers,
27
+ });
28
+
29
+ /** Tool names exposed by this connector — for contract/introspection tests. */
30
+ export const DATADOG_TOOL_NAMES = ["datadog_incident_list", "datadog_monitor_list"] as const;
31
+
32
+ export function registerDatadogTools(server: { tool: (...args: never) => unknown }): void {
33
+ const reg = createZodToolRegistrar(createRegisterSimpleTool(server));
34
+
35
+ reg("datadog_monitor_list", "List monitors.", z.object({}), async () =>
36
+ jsonResult(await ddGet("/api/v1/monitor")),
37
+ );
38
+
39
+ reg(
40
+ "datadog_incident_list",
41
+ "List incidents (v2).",
42
+ z.object({ limit: z.number().int().min(1).max(50).optional() }),
43
+ async (p) => {
44
+ const lim = p.limit ?? 10;
45
+ return jsonResult(await ddGet(`/api/v2/incidents?page[size]=${String(lim)}`));
46
+ },
47
+ );
48
+ }