@impart-security/impart-mcp 0.1.2 → 0.1.3

package/dist/index.js

@@ -42,11 +42,14 @@ async function makeImpartRequest(endpoint, queryParams, method = "GET", body) {
         });
     }
     try {
-        const response = await fetch(url.toString(), {
+        const requestOptions = {
             method,
             headers,
-            body: body ? JSON.stringify(body) : "",
-        });
+        };
+        if (body && method !== "GET" && method !== "HEAD") {
+            requestOptions.body = JSON.stringify(body);
+        }
+        const response = await fetch(url.toString(), requestOptions);
         if (!response.ok) {
             throw new Error(`HTTP error! status: ${response.status}`);
         }
@@ -57,21 +60,30 @@ async function makeImpartRequest(endpoint, queryParams, method = "GET", body) {
         return null;
     }
 }
+// Helper function to generate tool names with orgId
+function getToolName(baseName) {
+    return `${baseName}_${config_js_1.config.orgId.slice(-4)}`;
+}
 // Register tools with consistent naming patterns
-server.tool("list_inspectors", "Get a list of inspectors for the organization", {
+server.tool(getToolName("list_inspectors"), "Get a list of inspectors for the organization", {
     page: zod_1.z
         .number()
         .default(1)
+        .optional()
         .describe("The page of results to return (minimum value: 1)"),
     max_results: zod_1.z
         .number()
         .default(100)
+        .optional()
         .describe("The maximum number of results to return"),
 }, async ({ page, max_results }) => {
-    const queryParams = {
-        page: validatePage(page),
-        max_results,
-    };
+    const queryParams = {};
+    if (page !== null && page !== undefined) {
+        queryParams["page"] = validatePage(page);
+    }
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
     const data = await makeImpartRequest("inspectors", queryParams);
     if (!data) {
         return {
@@ -99,42 +111,65 @@ server.tool("list_inspectors", "Get a list of inspectors for the organization",
     };
 });
 // Rules Script Tools
-server.tool("list_rules_scripts", "Get a list of rules scripts for your Impart organization", {
+server.tool(getToolName("list_rules_scripts"), "Get a list of rules scripts for your Impart organization", {
     page: zod_1.z
         .number()
         .default(1)
+        .nullable()
         .describe("The page of results to return (minimum value: 1)"),
     max_results: zod_1.z
         .number()
         .default(100)
+        .nullable()
         .describe("The maximum number of results to return"),
     type: zod_1.z
         .string()
         .optional()
+        .nullable()
         .describe("Filter by type of rule script ('custom' or 'core')"),
     exclude_src: zod_1.z
         .boolean()
         .default(false)
+        .nullable()
         .describe("Whether to exclude the rule script source"),
     exclude_revisions: zod_1.z
         .boolean()
         .default(false)
+        .nullable()
         .describe("Whether to exclude rule script revisions"),
     is_disabled: zod_1.z
         .string()
         .optional()
+        .nullable()
         .describe("Filter by disabled status ('true' or 'false')"),
-    label: zod_1.z.array(zod_1.z.string()).optional().describe("Filter by label slugs"),
+    label: zod_1.z
+        .array(zod_1.z.string())
+        .optional()
+        .nullable()
+        .describe("Filter by label slugs"),
 }, async ({ page, max_results, type, exclude_src, exclude_revisions, is_disabled, label, }) => {
-    const queryParams = {
-        page: validatePage(page),
-        max_results,
-        type,
-        exclude_src,
-        exclude_revisions,
-        is_disabled,
-        label,
-    };
+    const queryParams = {};
+    if (page !== null && page !== undefined) {
+        queryParams["page"] = validatePage(page);
+    }
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
+    if (type !== null && type !== undefined) {
+        queryParams["type"] = type;
+    }
+    if (exclude_src !== null && exclude_src !== undefined) {
+        queryParams["exclude_src"] = exclude_src;
+    }
+    if (exclude_revisions !== null && exclude_revisions !== undefined) {
+        queryParams["exclude_revisions"] = exclude_revisions;
+    }
+    if (is_disabled !== null && is_disabled !== undefined) {
+        queryParams["is_disabled"] = is_disabled;
+    }
+    if (label !== null && label !== undefined) {
+        queryParams["label"] = label;
+    }
     const data = await makeImpartRequest("rules_scripts", queryParams);
     if (!data) {
         return {
@@ -177,11 +212,15 @@ server.tool("list_rules_scripts", "Get a list of rules scripts for your Impart o
         content: [{ type: "text", text: formattedScripts.join("\n") }],
     };
 });
-server.tool("get_rules_script", "Get details of a specific rules script", {
+server.tool(getToolName("get_rules_script"), "Get details of a specific rules script", {
     rules_script_id: zod_1.z
         .string()
         .describe("The unique identifier of the rules script"),
-    revision: zod_1.z.number().optional().describe("The specific revision to fetch"),
+    revision: zod_1.z
+        .number()
+        .optional()
+        .nullable()
+        .describe("The specific revision to fetch"),
 }, async ({ rules_script_id, revision }) => {
     if (!rules_script_id) {
         return {
@@ -190,7 +229,10 @@ server.tool("get_rules_script", "Get details of a specific rules script", {
             ],
         };
     }
-    const queryParams = revision ? { revision } : undefined;
+    const queryParams = {};
+    if (revision !== null && revision !== undefined) {
+        queryParams["revision"] = revision;
+    }
     const data = await makeImpartRequest(`rules_scripts/${rules_script_id}`, queryParams);
     if (!data) {
         return {
@@ -226,7 +268,7 @@ server.tool("get_rules_script", "Get details of a specific rules script", {
         content: [{ type: "text", text: formattedScript.join("\n") }],
     };
 });
-server.tool("create_rules_script", "Creates a new rules script for the organization", {
+server.tool(getToolName("create_rules_script"), "Creates a new rules script for the organization", {
     name: zod_1.z.string().describe("The name of the rules script"),
     src: zod_1.z
         .string()
@@ -234,33 +276,48 @@ server.tool("create_rules_script", "Creates a new rules script for the organizat
     lang: zod_1.z
         .string()
         .default("assemblyscript")
+        .nullable()
         .describe("The language of the script"),
     description: zod_1.z
         .string()
         .optional()
+        .nullable()
         .describe("Optional description of the script"),
     blocking_effect: zod_1.z
         .string()
         .default("block")
+        .nullable()
         .describe("Effect when rule blocks"),
     disabled: zod_1.z
         .boolean()
         .default(false)
+        .nullable()
         .describe("Whether the script is disabled"),
     labels: zod_1.z
         .array(zod_1.z.string())
         .optional()
+        .nullable()
         .describe("Optional list of label slugs"),
 }, async ({ name, src, lang, description, blocking_effect, disabled, labels, }) => {
     const payload = {
         name,
         src,
-        lang,
-        description,
-        blocking_effect,
-        disabled,
-        labels: labels ?? [],
     };
+    if (lang !== null && lang !== undefined) {
+        payload["lang"] = lang;
+    }
+    if (description !== null && description !== undefined) {
+        payload["description"] = description;
+    }
+    if (blocking_effect !== null && blocking_effect !== undefined) {
+        payload["blocking_effect"] = blocking_effect;
+    }
+    if (disabled !== null && disabled !== undefined) {
+        payload["disabled"] = disabled;
+    }
+    if (labels !== null && labels !== undefined) {
+        payload["labels"] = labels;
+    }
     const data = await makeImpartRequest("rules_scripts", undefined, "POST", payload);
     if (!data) {
         return {
@@ -283,7 +340,7 @@ server.tool("create_rules_script", "Creates a new rules script for the organizat
         content: [{ type: "text", text: formattedScript }],
     };
 });
-server.tool("update_rules_script", "Updates an existing rules script", {
+server.tool(getToolName("update_rules_script"), "Updates an existing rules script", {
     rules_script_id: zod_1.z
         .string()
         .describe("The ID of the rules script to update"),
@@ -292,22 +349,27 @@ server.tool("update_rules_script", "Updates an existing rules script", {
     lang: zod_1.z
         .string()
         .default("assemblyscript")
+        .nullable()
         .describe("The language of the script"),
     description: zod_1.z
         .string()
         .optional()
+        .nullable()
         .describe("Optional description of the script"),
     blocking_effect: zod_1.z
         .string()
         .default("block")
+        .nullable()
         .describe("Effect when rule blocks"),
     disabled: zod_1.z
         .boolean()
         .default(false)
+        .nullable()
         .describe("Whether the script is disabled"),
     labels: zod_1.z
         .array(zod_1.z.string())
         .optional()
+        .nullable()
         .describe("Optional list of label slugs"),
 }, async ({ rules_script_id, name, src, lang, description, blocking_effect, disabled, labels, }) => {
     if (!rules_script_id) {
@@ -320,12 +382,22 @@ server.tool("update_rules_script", "Updates an existing rules script", {
     const payload = {
         name,
         src,
-        lang,
-        description,
-        blocking_effect,
-        disabled,
-        labels: labels ?? [],
     };
+    if (lang !== null && lang !== undefined) {
+        payload["lang"] = lang;
+    }
+    if (description !== null && description !== undefined) {
+        payload["description"] = description;
+    }
+    if (blocking_effect !== null && blocking_effect !== undefined) {
+        payload["blocking_effect"] = blocking_effect;
+    }
+    if (disabled !== null && disabled !== undefined) {
+        payload["disabled"] = disabled;
+    }
+    if (labels !== null && labels !== undefined) {
+        payload["labels"] = labels;
+    }
     const data = await makeImpartRequest(`rules_scripts/${rules_script_id}`, undefined, "PUT", payload);
     if (!data) {
         return {
@@ -348,7 +420,7 @@ server.tool("update_rules_script", "Updates an existing rules script", {
         content: [{ type: "text", text: formattedScript }],
     };
 });
-server.tool("write_rule_script", "Generate a rule script using Impart's LLM", {
+server.tool(getToolName("write_rule_script"), "Generate a rule script using Impart's LLM", {
     query: zod_1.z.string().describe("Description of the rule you want to generate"),
 }, async ({ query }) => {
     const payload = {
@@ -379,13 +451,14 @@ server.tool("write_rule_script", "Generate a rule script using Impart's LLM", {
         ],
     };
 });
-server.tool("validate_rules_script", "Validate a rules script", {
+server.tool(getToolName("validate_rules_script"), "Validate a rules script", {
     src: zod_1.z
         .string()
         .describe("The base64-encoded source code of the rules script to validate"),
     lang: zod_1.z
         .string()
         .default("assemblyscript")
+        .nullable()
         .describe("The language of the rules script"),
 }, async ({ src, lang }) => {
     if (!src) {
@@ -400,8 +473,10 @@ server.tool("validate_rules_script", "Validate a rules script", {
     }
     const payload = {
         src,
-        lang,
     };
+    if (lang !== null && lang !== undefined) {
+        payload["lang"] = lang;
+    }
     const data = await makeImpartRequest("rules_script_validate", undefined, "POST", payload);
     if (!data) {
         return {
@@ -430,13 +505,17 @@ server.tool("validate_rules_script", "Validate a rules script", {
         ],
     };
 });
-server.tool("list_endpoints", "Get an inventory of endpoints for the organization", {
+server.tool(getToolName("list_endpoints"), "Get an inventory of endpoints for the organization", {
     max_results: zod_1.z
         .number()
         .default(100)
+        .nullable()
         .describe("The maximum number of results to return"),
 }, async ({ max_results }) => {
-    const queryParams = { max_results };
+    const queryParams = {};
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
     const data = await makeImpartRequest("endpoints", queryParams);
     if (!data) {
         return {
@@ -466,13 +545,17 @@ server.tool("list_endpoints", "Get an inventory of endpoints for the organizatio
         content: [{ type: "text", text: formattedEndpoints.join("\n") }],
     };
 });
-server.tool("list_tags", "Get a list of tags for your Impart organization", {
+server.tool(getToolName("list_tags"), "Get a list of tags for your Impart organization", {
     max_results: zod_1.z
         .number()
         .default(100)
+        .nullable()
         .describe("The maximum number of results to return"),
 }, async ({ max_results }) => {
-    const queryParams = { max_results };
+    const queryParams = {};
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
     const data = await makeImpartRequest("tags", queryParams);
     if (!data) {
         return {
@@ -503,49 +586,73 @@ server.tool("list_tags", "Get a list of tags for your Impart organization", {
         content: [{ type: "text", text: formattedTags.join("\n") }],
     };
 });
-server.tool("get_tag_metrics", "Get timeseries metrics for one or more tags", {
+server.tool(getToolName("get_tag_metrics"), "Get timeseries metrics for one or more tags", {
     metric: zod_1.z
         .array(zod_1.z.string())
         .default(["http-request.count"])
+        .nullable()
         .describe("List of metric names"),
-    from_time: zod_1.z.string().default("-1d").describe("Start time"),
-    until_time: zod_1.z.string().default("-0h").describe("End time"),
+    from_time: zod_1.z.string().default("-1d").nullable().describe("Start time"),
+    until_time: zod_1.z.string().default("-0h").nullable().describe("End time"),
     spec_id: zod_1.z
         .array(zod_1.z.string())
         .optional()
+        .nullable()
         .describe("Optional list of spec IDs"),
     collection_id: zod_1.z
         .array(zod_1.z.string())
         .optional()
+        .nullable()
         .describe("Optional list of collection IDs"),
     endpoint_id: zod_1.z
         .array(zod_1.z.string())
         .optional()
+        .nullable()
         .describe("Optional list of endpoint IDs"),
     rollup: zod_1.z
         .string()
         .optional()
+        .nullable()
         .describe("Optional time window rollup resolution"),
     unlearned: zod_1.z
         .string()
         .default("include")
+        .nullable()
         .describe("How to handle unlearned endpoints"),
     max_results: zod_1.z
         .number()
         .default(100)
+        .nullable()
         .describe("The maximum number of results to return"),
 }, async ({ metric, from_time, until_time, spec_id, collection_id, endpoint_id, rollup, unlearned, max_results, }) => {
-    const queryParams = {
-        metric,
-        from: from_time,
-        until: until_time,
-        spec_id,
-        collection_id,
-        endpoint_id,
-        rollup,
-        unlearned,
-        max_results,
-    };
+    const queryParams = {};
+    if (metric !== null && metric !== undefined) {
+        queryParams["metric"] = metric;
+    }
+    if (from_time !== null && from_time !== undefined) {
+        queryParams["from"] = from_time;
+    }
+    if (until_time !== null && until_time !== undefined) {
+        queryParams["until"] = until_time;
+    }
+    if (spec_id !== null && spec_id !== undefined) {
+        queryParams["spec_id"] = spec_id;
+    }
+    if (collection_id !== null && collection_id !== undefined) {
+        queryParams["collection_id"] = collection_id;
+    }
+    if (endpoint_id !== null && endpoint_id !== undefined) {
+        queryParams["endpoint_id"] = endpoint_id;
+    }
+    if (rollup !== null && rollup !== undefined) {
+        queryParams["rollup"] = rollup;
+    }
+    if (unlearned !== null && unlearned !== undefined) {
+        queryParams["unlearned"] = unlearned;
+    }
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
     const data = await makeImpartRequest("timeseries/tags", queryParams);
     if (!data) {
         return {
@@ -581,20 +688,25 @@ server.tool("get_tag_metrics", "Get timeseries metrics for one or more tags", {
         content: [{ type: "text", text: formattedMetrics.join("\n") }],
     };
 });
-server.tool("list_observed_hosts", "Get a list of observed hosts for your Impart organization", {
+server.tool(getToolName("list_observed_hosts"), "Get a list of observed hosts for your Impart organization", {
     page: zod_1.z
         .number()
         .default(1)
+        .nullable()
         .describe("The page of results to return (minimum value: 1)"),
     max_results: zod_1.z
         .number()
         .default(100)
+        .nullable()
         .describe("The maximum number of results to return"),
 }, async ({ page, max_results }) => {
-    const queryParams = {
-        page: validatePage(page),
-        max_results,
-    };
+    const queryParams = {};
+    if (page !== null && page !== undefined) {
+        queryParams["page"] = validatePage(page);
+    }
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
     const data = await makeImpartRequest("observed_hosts", queryParams);
     if (!data) {
         return {
@@ -628,7 +740,7 @@ server.tool("list_observed_hosts", "Get a list of observed hosts for your Impart
         content: [{ type: "text", text: formattedHosts.join("\n") }],
     };
 });
-server.tool("get_observed_host", "Get details of a specific observed host", {
+server.tool(getToolName("get_observed_host"), "Get details of a specific observed host", {
     observed_host_id: zod_1.z
         .string()
         .describe("The unique identifier of the observed host"),
@@ -671,20 +783,25 @@ server.tool("get_observed_host", "Get details of a specific observed host", {
         content: [{ type: "text", text: formattedHost.join("\n") }],
     };
 });
-server.tool("list_specs", "Get a list of specs for your Impart organization", {
+server.tool(getToolName("list_specs"), "Get a list of specs for your Impart organization", {
     page: zod_1.z
         .number()
         .default(1)
+        .nullable()
         .describe("The page of results to return (minimum value: 1)"),
     max_results: zod_1.z
         .number()
         .default(100)
+        .nullable()
         .describe("The maximum number of results to return"),
 }, async ({ page, max_results }) => {
-    const queryParams = {
-        page: validatePage(page),
-        max_results,
-    };
+    const queryParams = {};
+    if (page !== null && page !== undefined) {
+        queryParams["page"] = validatePage(page);
+    }
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
     const data = await makeImpartRequest("specs", queryParams);
     if (!data) {
         return {
@@ -716,16 +833,23 @@ server.tool("list_specs", "Get a list of specs for your Impart organization", {
         content: [{ type: "text", text: formattedSpecs.join("\n") }],
     };
 });
-server.tool("get_spec", "Get details of a specific spec", {
+server.tool(getToolName("get_spec"), "Get details of a specific spec", {
     spec_id: zod_1.z.string().describe("The unique identifier of the spec"),
-    revision: zod_1.z.number().optional().describe("The specific revision to fetch"),
+    revision: zod_1.z
+        .number()
+        .optional()
+        .nullable()
+        .describe("The specific revision to fetch"),
 }, async ({ spec_id, revision }) => {
     if (!spec_id) {
         return {
             content: [{ type: "text", text: "Error: spec_id is required." }],
         };
     }
-    const queryParams = revision ? { revision } : undefined;
+    const queryParams = {};
+    if (revision !== null && revision !== undefined) {
+        queryParams["revision"] = revision;
+    }
     const data = await makeImpartRequest(`specs/${spec_id}`, queryParams);
     if (!data) {
         return {
@@ -756,20 +880,25 @@ server.tool("get_spec", "Get details of a specific spec", {
         content: [{ type: "text", text: formattedSpec.join("\n") }],
     };
 });
-server.tool("list_api_bindings", "Get a list of API bindings for the organization", {
+server.tool(getToolName("list_api_bindings"), "Get a list of API bindings for the organization", {
     page: zod_1.z
         .number()
         .default(1)
+        .nullable()
         .describe("The page of results to return (minimum value: 1)"),
     max_results: zod_1.z
         .number()
         .default(100)
+        .nullable()
         .describe("The maximum number of results to return"),
 }, async ({ page, max_results }) => {
-    const queryParams = {
-        page: validatePage(page),
-        max_results,
-    };
+    const queryParams = {};
+    if (page !== null && page !== undefined) {
+        queryParams["page"] = validatePage(page);
+    }
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
     const data = await makeImpartRequest("api_bindings", queryParams);
     if (!data) {
         return {
@@ -802,7 +931,7 @@ server.tool("list_api_bindings", "Get a list of API bindings for the organizatio
         content: [{ type: "text", text: formattedBindings.join("\n") }],
     };
 });
-server.tool("create_api_binding", "Create a new API binding for the organization", {
+server.tool(getToolName("create_api_binding"), "Create a new API binding for the organization", {
     name: zod_1.z.string().describe("The name of the API binding"),
     binding_type: zod_1.z
         .string()
@@ -815,26 +944,36 @@ server.tool("create_api_binding", "Create a new API binding for the organization
     description: zod_1.z
         .string()
         .optional()
+        .nullable()
         .describe("Optional description of the binding"),
     patch_operations: zod_1.z
         .array(zod_1.z.record(zod_1.z.any()))
         .optional()
+        .nullable()
         .describe("Optional list of JSON patch operations"),
     enable_learning: zod_1.z
         .boolean()
         .default(true)
+        .nullable()
         .describe("Whether to enable learning for this binding"),
 }, async ({ name, binding_type, host, api_binding_source, binding_value, description, patch_operations, enable_learning, }) => {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     const payload = {
         name,
         type: binding_type,
         host,
         source: api_binding_source,
         value: binding_value,
-        description,
-        patch_operations: patch_operations ?? [],
-        enable_learning,
     };
+    if (description !== null && description !== undefined) {
+        payload["description"] = description;
+    }
+    if (patch_operations !== null && patch_operations !== undefined) {
+        payload["patch_operations"] = patch_operations;
+    }
+    if (enable_learning !== null && enable_learning !== undefined) {
+        payload["enable_learning"] = enable_learning;
+    }
     const data = await makeImpartRequest("api_bindings", undefined, "POST", payload);
     if (!data) {
         return {
@@ -858,24 +997,41 @@ server.tool("create_api_binding", "Create a new API binding for the organization
         content: [{ type: "text", text: formattedBinding }],
     };
 });
-server.tool("list_lists", "Get a list of lists for the organization", {
-    kind: zod_1.z.string().optional().describe("The list kind to query for"),
-    subkind: zod_1.z.string().optional().describe("The list subkind to query for"),
+server.tool(getToolName("list_lists"), "Get a list of lists for the organization", {
+    kind: zod_1.z
+        .string()
+        .optional()
+        .nullable()
+        .describe("The list kind to query for"),
+    subkind: zod_1.z
+        .string()
+        .optional()
+        .nullable()
+        .describe("The list subkind to query for"),
     page: zod_1.z
         .number()
         .default(1)
+        .nullable()
         .describe("The page of results to return (minimum: 1)"),
     max_results: zod_1.z
         .number()
         .default(100)
+        .nullable()
         .describe("The maximum number of results to return"),
 }, async ({ kind, subkind, page, max_results }) => {
-    const queryParams = {
-        kind,
-        subkind,
-        page: validatePage(page),
-        max_results,
-    };
+    const queryParams = {};
+    if (kind !== null && kind !== undefined) {
+        queryParams["kind"] = kind;
+    }
+    if (subkind !== null && subkind !== undefined) {
+        queryParams["subkind"] = subkind;
+    }
+    if (page !== null && page !== undefined) {
+        queryParams["page"] = validatePage(page);
+    }
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
     const data = await makeImpartRequest("lists", queryParams);
     if (!data) {
         return {
@@ -906,7 +1062,7 @@ server.tool("list_lists", "Get a list of lists for the organization", {
         content: [{ type: "text", text: formattedLists.join("\n") }],
     };
 });
-server.tool("get_list_items", "Get items from a specific list", {
+server.tool(getToolName("get_list_items"), "Get items from a specific list", {
     list_id: zod_1.z.string().describe("The ID of the list"),
     page: zod_1.z
         .number()
@@ -945,49 +1101,61 @@ server.tool("get_list_items", "Get items from a specific list", {
         content: [{ type: "text", text: formattedItems.join("\n") }],
     };
 });
-server.tool("create_list", "Creates a new list for the organization", {
-    name: zod_1.z.string().describe("Name of the list"),
-    kind: zod_1.z.string().describe("The kind of list"),
-    subkind: zod_1.z.string().optional().describe("Optional subkind of list"),
-    description: zod_1.z.string().optional().describe("Optional description"),
-    functionality: zod_1.z
-        .string()
-        .default("add/remove")
-        .describe("List functionality"),
+server.tool(getToolName("create_list"), "Create a new list", {
+    kind: zod_1.z.string().describe("The kind of list to create"),
+    subkind: zod_1.z.string().describe("The subkind of list to create"),
+    name: zod_1.z.string().describe("The name of the list"),
+    description: zod_1.z.string().nullable().describe("The description of the list"),
     items: zod_1.z
-        .array(zod_1.z.record(zod_1.z.any()))
-        .optional()
-        .describe("Optional initial list items"),
-}, async ({ name, kind, subkind, description, functionality, items }) => {
-    const payload = {
-        name,
+        .array(zod_1.z.string())
+        .nullable()
+        .describe("The items to add to the list"),
+}, async ({ kind, subkind, name, description, items }) => {
+    if (!kind || !subkind || !name) {
+        return {
+            content: [
+                { type: "text", text: "Error: kind, subkind, and name are required" },
+            ],
+            isError: true,
+        };
+    }
+    const body = {
         kind,
         subkind,
-        description,
-        functionality,
-        items: items ?? [],
+        name,
     };
-    const data = await makeImpartRequest("lists", undefined, "POST", payload);
+    if (description !== null && description !== undefined) {
+        body["description"] = description;
+    }
+    if (items !== null && items !== undefined) {
+        body["items"] = items;
+    }
+    const data = await makeImpartRequest("lists", {}, "POST", body);
     if (!data) {
         return {
             content: [{ type: "text", text: "Failed to create list" }],
+            isError: true,
         };
     }
-    const formattedList = [
-        "List created successfully:",
-        `ID: ${data.id}`,
-        `Name: ${data.name}`,
-        `Kind: ${data.kind}`,
-        `Subkind: ${data.subkind ?? "N/A"}`,
-        `Description: ${data.description ?? "N/A"}`,
-        `Created At: ${data.created_at}`,
-        "---",
-    ].join("\n");
     return {
-        content: [{ type: "text", text: formattedList }],
+        content: [
+            {
+                type: "text",
+                text: [
+                    "List created successfully:",
+                    `ID: ${data.id}`,
+                    `Name: ${name}`,
+                    `Kind: ${kind}`,
+                    `Subkind: ${subkind}`,
+                    `Description: ${description ?? "N/A"}`,
+                    `Created At: ${data.created_at}`,
+                    "---",
+                ].join("\n"),
+            },
+        ],
     };
 });
-server.tool("update_list_items", "Updates items in a list", {
+server.tool(getToolName("update_list_items"), "Updates items in a list", {
     list_id: zod_1.z.string().describe("The ID of the list to update"),
     items: zod_1.z
         .array(zod_1.z.record(zod_1.z.any()))
@@ -1009,7 +1177,7 @@ server.tool("update_list_items", "Updates items in a list", {
         content: [{ type: "text", text: "List items updated successfully" }],
     };
 });
-server.tool("create_label", "Creates a new label for the organization", {
+server.tool(getToolName("create_label"), "Creates a new label for the organization", {
     name: zod_1.z.string().describe("Name of the label"),
     description: zod_1.z.string().optional().describe("Optional description"),
     color: zod_1.z.string().default("gray").describe("Label color"),
@@ -1038,20 +1206,25 @@ server.tool("create_label", "Creates a new label for the organization", {
         content: [{ type: "text", text: formattedLabel }],
     };
 });
-server.tool("list_labels", "Get a list of labels for the organization", {
+server.tool(getToolName("list_labels"), "Get a list of labels for the organization", {
     page: zod_1.z
         .number()
         .default(1)
+        .nullable()
         .describe("The page of results to return (minimum: 1)"),
     max_results: zod_1.z
         .number()
         .default(100)
+        .nullable()
         .describe("The maximum number of results to return"),
 }, async ({ page, max_results }) => {
-    const queryParams = {
-        page: validatePage(page),
-        max_results,
-    };
+    const queryParams = {};
+    if (page !== null && page !== undefined) {
+        queryParams["page"] = validatePage(page);
+    }
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
     const data = await makeImpartRequest("labels", queryParams);
     if (!data) {
         return {
@@ -1080,6 +1253,399 @@ server.tool("list_labels", "Get a list of labels for the organization", {
         content: [{ type: "text", text: formattedLabels.join("\n") }],
     };
 });
+server.tool(getToolName("list_rules_test_cases"), "Get a list of rules test cases for your Impart organization", {
+    page: zod_1.z
+        .number()
+        .default(1)
+        .nullable()
+        .describe("The page of results to return (minimum value: 1)"),
+    max_results: zod_1.z
+        .number()
+        .default(100)
+        .nullable()
+        .describe("The maximum number of results to return"),
+    label: zod_1.z
+        .array(zod_1.z.string())
+        .optional()
+        .nullable()
+        .describe("Optional list of label slugs to filter by"),
+    search: zod_1.z
+        .string()
+        .optional()
+        .nullable()
+        .describe("Optional search string to filter test cases"),
+}, async ({ page, max_results, label, search }) => {
+    const queryParams = {};
+    if (page !== null && page !== undefined) {
+        queryParams["page"] = validatePage(page);
+    }
+    if (max_results !== null && max_results !== undefined) {
+        queryParams["max_results"] = max_results;
+    }
+    if (label !== null && label !== undefined) {
+        queryParams["label"] = label;
+    }
+    if (search !== null && search !== undefined) {
+        queryParams["search"] = search;
+    }
+    const data = await makeImpartRequest("rules_test_cases", queryParams);
+    if (!data) {
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: "Unable to fetch rules test cases: Unknown error",
+                },
+            ],
+        };
+    }
+    const testCases = data.items || [];
+    if (testCases.length === 0) {
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: "No rules test cases found for this organization.",
+                },
+            ],
+        };
+    }
+    const formattedTestCases = testCases.map((testCase) => [
+        `ID: ${testCase.id}`,
+        `Name: ${testCase.name}`,
+        `Description: ${testCase.description ?? "N/A"}`,
+        `Required: ${testCase.required}`,
+        `Labels: ${testCase.labels.join(", ")}`,
+        `Messages Count: ${testCase.messages.length}`,
+        `Assertions Count: ${testCase.assertions.length}`,
+        `Created By: ${testCase.created_by}`,
+        `Created At: ${testCase.created_at}`,
+        `Updated By: ${testCase.updated_by}`,
+        `Updated At: ${testCase.updated_at}`,
+        "---",
+    ].join("\n"));
+    return {
+        content: [{ type: "text", text: formattedTestCases.join("\n") }],
+    };
+});
+server.tool(getToolName("create_rules_test_case"), "Create a new rules test case", {
+    name: zod_1.z.string().describe("The name of the test case"),
+    description: zod_1.z
+        .string()
+        .optional()
+        .nullable()
+        .describe("Optional description of the test case"),
+    required: zod_1.z
+        .boolean()
+        .default(false)
+        .nullable()
+        .describe("Whether the test case is required to pass"),
+    messages: zod_1.z
+        .array(zod_1.z.object({
+        request: zod_1.z
+            .object({
+            method: zod_1.z.string().optional(),
+            url: zod_1.z.string().optional(),
+            headers: zod_1.z.record(zod_1.z.array(zod_1.z.string())).optional(),
+            body: zod_1.z.string().optional(),
+        })
+            .optional(),
+        response: zod_1.z
+            .object({
+            status_code: zod_1.z.number().optional(),
+            headers: zod_1.z.record(zod_1.z.array(zod_1.z.string())).optional(),
+            body: zod_1.z.string().optional(),
+        })
+            .optional(),
+    }))
+        .describe("The test case messages"),
+    assertions: zod_1.z
+        .array(zod_1.z.object({
+        message_index: zod_1.z.number(),
+        message_definition: zod_1.z.object({
+            line: zod_1.z.number(),
+            column: zod_1.z.number(),
+        }),
+        passed: zod_1.z.boolean(),
+    }))
+        .optional()
+        .nullable()
+        .describe("Optional test case assertions"),
+    labels: zod_1.z
+        .array(zod_1.z.string())
+        .optional()
+        .nullable()
+        .describe("Optional list of label slugs"),
+}, async ({ name, description, required, messages, assertions, labels }) => {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const payload = {
+        name,
+        messages,
+    };
+    if (description !== null && description !== undefined) {
+        payload["description"] = description;
+    }
+    if (required !== null && required !== undefined) {
+        payload["required"] = required;
+    }
+    if (assertions !== null && assertions !== undefined) {
+        payload["assertions"] = assertions;
+    }
+    if (labels !== null && labels !== undefined) {
+        payload["labels"] = labels;
+    }
+    const data = await makeImpartRequest("rules_test_cases", undefined, "POST", payload);
+    if (!data) {
+        return {
+            content: [{ type: "text", text: "Failed to create rules test case" }],
+        };
+    }
+    const formattedTestCase = [
+        "Rules test case created successfully:",
+        `ID: ${data.id}`,
+        `Name: ${data.name}`,
+        `Description: ${data.description ?? "N/A"}`,
+        `Required: ${data.required}`,
+        `Labels: ${data.labels.join(", ")}`,
+        `Messages Count: ${data.messages.length}`,
+        `Assertions Count: ${data.assertions.length}`,
+        `Created By: ${data.created_by}`,
+        `Created At: ${data.created_at}`,
+        "---",
+    ].join("\n");
+    return {
+        content: [{ type: "text", text: formattedTestCase }],
+    };
+});
+server.tool(getToolName("get_rules_test_case"), "Get details of a specific rules test case", {
+    rules_test_case_id: zod_1.z
+        .string()
+        .describe("The unique identifier of the rules test case"),
+}, async ({ rules_test_case_id }) => {
+    if (!rules_test_case_id) {
+        return {
+            content: [
+                { type: "text", text: "Error: rules_test_case_id is required." },
+            ],
+        };
+    }
+    const data = await makeImpartRequest(`rules_test_cases/${rules_test_case_id}`);
+    if (!data) {
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: "Unable to fetch rules test case: Unknown error",
+                },
+            ],
+        };
+    }
+    const formattedTestCase = [
+        `ID: ${data.id}`,
+        `Name: ${data.name}`,
+        `Description: ${data.description ?? "N/A"}`,
+        `Required: ${data.required}`,
+        `Labels: ${data.labels.join(", ")}`,
+        "Messages:",
+    ];
+    data.messages.forEach((message, index) => {
+        formattedTestCase.push(`Message ${index + 1}:`);
+        if (message.request) {
+            formattedTestCase.push("  Request:", `    Method: ${message.request.method ?? "N/A"}`, `    URL: ${message.request.url ?? "N/A"}`, "    Headers:");
+            if (message.request.headers) {
+                Object.entries(message.request.headers).forEach(([key, values]) => {
+                    formattedTestCase.push(`      ${key}: ${values.join(", ")}`);
+                });
+            }
+            if (message.request.body) {
+                formattedTestCase.push(`    Body: ${message.request.body}`);
+            }
+        }
+        if (message.response) {
+            formattedTestCase.push("  Response:", `    Status Code: ${message.response.status_code ?? "N/A"}`, "    Headers:");
+            if (message.response.headers) {
+                Object.entries(message.response.headers).forEach(([key, values]) => {
+                    formattedTestCase.push(`      ${key}: ${values.join(", ")}`);
+                });
+            }
+            if (message.response.body) {
+                formattedTestCase.push(`    Body: ${message.response.body}`);
+            }
+        }
+    });
+    formattedTestCase.push("Assertions:", ...data.assertions.map((assertion) => `  Message ${assertion.message_index + 1} at line ${assertion.message_definition.line}, column ${assertion.message_definition.column}: ${assertion.passed ? "Passed" : "Failed"}`), "---");
+    return {
+        content: [{ type: "text", text: formattedTestCase.join("\n") }],
+    };
+});
+server.tool(getToolName("update_rules_test_case"), "Update an existing rules test case", {
+    rules_test_case_id: zod_1.z
+        .string()
+        .describe("The unique identifier of the rules test case"),
+    name: zod_1.z.string().describe("The name of the test case"),
+    description: zod_1.z
+        .string()
+        .optional()
+        .nullable()
+        .describe("Optional description of the test case"),
+    required: zod_1.z
+        .boolean()
+        .default(false)
+        .nullable()
+        .describe("Whether the test case is required to pass"),
+    messages: zod_1.z
+        .array(zod_1.z.object({
+        request: zod_1.z
+            .object({
+            method: zod_1.z.string().optional(),
+            url: zod_1.z.string().optional(),
+            headers: zod_1.z.record(zod_1.z.array(zod_1.z.string())).optional(),
+            body: zod_1.z.string().optional(),
+        })
+            .optional(),
+        response: zod_1.z
+            .object({
+            status_code: zod_1.z.number().optional(),
+            headers: zod_1.z.record(zod_1.z.array(zod_1.z.string())).optional(),
+            body: zod_1.z.string().optional(),
+        })
+            .optional(),
+    }))
+        .describe("The test case messages"),
+    assertions: zod_1.z
+        .array(zod_1.z.object({
+        message_index: zod_1.z.number(),
+        message_definition: zod_1.z.object({
+            line: zod_1.z.number(),
+            column: zod_1.z.number(),
+        }),
+        passed: zod_1.z.boolean(),
+    }))
+        .optional()
+        .nullable()
+        .describe("Optional test case assertions"),
+    labels: zod_1.z
+        .array(zod_1.z.string())
+        .optional()
+        .nullable()
+        .describe("Optional list of label slugs"),
+}, async ({ rules_test_case_id, name, description, required, messages, assertions, labels, }) => {
+    if (!rules_test_case_id) {
+        return {
+            content: [
+                { type: "text", text: "Error: rules_test_case_id is required." },
+            ],
+        };
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const payload = {
+        name,
+        messages,
+    };
+    if (description !== null && description !== undefined) {
+        payload["description"] = description;
+    }
+    if (required !== null && required !== undefined) {
+        payload["required"] = required;
+    }
+    if (assertions !== null && assertions !== undefined) {
+        payload["assertions"] = assertions;
+    }
+    if (labels !== null && labels !== undefined) {
+        payload["labels"] = labels;
+    }
+    const data = await makeImpartRequest(`rules_test_cases/${rules_test_case_id}`, undefined, "PUT", payload);
+    if (!data) {
+        return {
+            content: [{ type: "text", text: "Failed to update rules test case" }],
+        };
+    }
+    const formattedTestCase = [
+        "Rules test case updated successfully:",
+        `ID: ${data.id}`,
+        `Name: ${data.name}`,
+        `Description: ${data.description ?? "N/A"}`,
+        `Required: ${data.required}`,
+        `Labels: ${data.labels.join(", ")}`,
+        `Messages Count: ${data.messages.length}`,
+        `Assertions Count: ${data.assertions.length}`,
+        `Updated By: ${data.updated_by}`,
+        `Updated At: ${data.updated_at}`,
+        "---",
+    ].join("\n");
+    return {
+        content: [{ type: "text", text: formattedTestCase }],
+    };
+});
+server.tool(getToolName("delete_rules_test_case"), "Delete a rules test case", {
+    rules_test_case_id: zod_1.z
+        .string()
+        .describe("The unique identifier of the rules test case"),
+}, async ({ rules_test_case_id }) => {
+    if (!rules_test_case_id) {
+        return {
+            content: [
+                { type: "text", text: "Error: rules_test_case_id is required." },
+            ],
+        };
+    }
+    const data = await makeImpartRequest(`rules_test_cases/${rules_test_case_id}`, undefined, "DELETE");
+    if (data === null) {
+        return {
+            content: [{ type: "text", text: "Failed to delete rules test case" }],
+        };
+    }
+    return {
+        content: [{ type: "text", text: "Rules test case deleted successfully" }],
+    };
+});
+server.tool(getToolName("run_rules_test_case"), "Run a rules test case", {
+    messages: zod_1.z
+        .array(zod_1.z.object({
+        request: zod_1.z
+            .object({
+            method: zod_1.z.string().optional(),
+            url: zod_1.z.string().optional(),
+            headers: zod_1.z.record(zod_1.z.array(zod_1.z.string())).optional(),
+            body: zod_1.z.string().optional(),
+        })
+            .optional(),
+        response: zod_1.z
+            .object({
+            status_code: zod_1.z.number().optional(),
+            headers: zod_1.z.record(zod_1.z.array(zod_1.z.string())).optional(),
+            body: zod_1.z.string().optional(),
+        })
+            .optional(),
+    }))
+        .nullable()
+        .describe("The test case messages to run"),
+}, async ({ messages }) => {
+    if (!messages) {
+        return {
+            content: [{ type: "text", text: "Error: messages is required." }],
+        };
+    }
+    const payload = { messages };
+    const data = await makeImpartRequest("rules_test_case_run", undefined, "POST", payload);
+    if (!data) {
+        return {
+            content: [{ type: "text", text: "Failed to run rules test case" }],
+        };
+    }
+    const formattedResult = [
+        "Rules test case run completed:",
+        `Duration: ${data.duration} microseconds`,
+        "Output:",
+        ...data.output.map((out) => `  ${out.severity.toUpperCase()} at line ${out.line}, column ${out.column}: ${out.message}`),
+        "Assertions:",
+        ...data.assertions.flatMap((assertion) => assertion.results.map((result) => `  Message ${result.message_index + 1} at line ${result.message_definition.line}, column ${result.message_definition.column}: ${result.passed ? "Passed" : "Failed"}`)),
+        "---",
+    ].join("\n");
+    return {
+        content: [{ type: "text", text: formattedResult }],
+    };
+});
 async function main() {
     const transport = new stdio_js_1.StdioServerTransport();
     await server.connect(transport);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@impart-security/impart-mcp",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "Model Context Protocol (MCP) implementation for interacting with Impart's API",
   "author": "Impart Security",
   "license": "Apache-2.0",
@@ -21,9 +21,9 @@
     "node": ">=18.0.0"
   },
   "dependencies": {
-    "@modelcontextprotocol/sdk": "1.9.0",
+    "@modelcontextprotocol/sdk": "1.10.2",
     "axios": "1.8.4",
-    "zod": "3.24.2"
+    "zod": "3.24.3"
   },
   "devDependencies": {
     "@types/node": "22.14.1",