chore: refactor into getScimBaseUrl

Signed-off-by: Hung-Han (Henry) Chen <chenhungh@gmail.com>

src/resources/(groups)/[groupId]/index.ts

@@ -2,6 +2,7 @@ import { type InferSchema, type ResourceMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { getScimToken } from "../../../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../../../utils/getSCIMBaseUrl";
 
 export const schema = {
   groupId: z.string().describe("The ID of the group"),
@@ -32,7 +33,7 @@ export default async function handler({
 }: InferSchema<typeof schema>) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/resources/(groups)/index.ts

@@ -2,6 +2,7 @@ import { type InferSchema, type ResourceMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { getScimToken } from "../../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../../utils/getSCIMBaseUrl";
 
 export const schema = {
   filter: z
@@ -37,7 +38,7 @@ export default async function handler({
 }: InferSchema<typeof schema>) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/resources/(users)/[userId]/index.ts

@@ -2,6 +2,7 @@ import { type InferSchema, type ResourceMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { getScimToken } from "../../../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../../../utils/getSCIMBaseUrl";
 
 export const schema = {
   userId: z.string().describe("The ID of the user"),
@@ -32,7 +33,7 @@ export default async function handler({
 }: InferSchema<typeof schema>) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/resources/(users)/index.ts

@@ -2,6 +2,7 @@ import { type InferSchema, type ResourceMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { getScimToken } from "../../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../../utils/getSCIMBaseUrl";
 
 export const schema = {
   filter: z
@@ -37,7 +38,7 @@ export default async function handler({
 }: InferSchema<typeof schema>) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/tools/addUserToGroup.ts

@@ -2,6 +2,7 @@ import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { getScimToken } from "../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../utils/getSCIMBaseUrl";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -26,7 +27,7 @@ export default async function addUserToGroup(
 ) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/tools/createGroup.ts

@@ -3,6 +3,7 @@ import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { groupResourceSchema } from "../schemas/groupResourceSchema";
 import { getScimToken } from "../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../utils/getSCIMBaseUrl";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -32,7 +33,7 @@ export default async function createGroup(
   const { groupResource } = params;
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/tools/createUser.ts

@@ -3,6 +3,7 @@ import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { userResourceSchema } from "../schemas/userResourceSchema";
 import { getScimToken } from "../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../utils/getSCIMBaseUrl";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -32,7 +33,7 @@ export default async function createUser(
   const { userResource } = params;
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/tools/deleteGroup.ts

@@ -2,6 +2,7 @@ import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { getScimToken } from "../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../utils/getSCIMBaseUrl";
 
 export const metadata: ToolMetadata = {
   name: "delete-group",
@@ -24,7 +25,7 @@ export default async function deleteGroup(
 ) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/tools/deleteUser.ts

@@ -2,6 +2,7 @@ import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { getScimToken } from "../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../utils/getSCIMBaseUrl";
 
 export const metadata: ToolMetadata = {
   name: "delete-user",
@@ -24,7 +25,7 @@ export default async function deleteUser(
 ) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/tools/patchGroup.ts

@@ -3,6 +3,7 @@ import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { patchOperationSchema } from "../schemas/patchOperationSchema";
 import { getScimToken } from "../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../utils/getSCIMBaseUrl";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -27,7 +28,7 @@ export default async function patchGroup(
 ) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/tools/patchUser.ts

@@ -3,6 +3,7 @@ import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { patchOperationSchema } from "../schemas/patchOperationSchema";
 import { getScimToken } from "../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../utils/getSCIMBaseUrl";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -27,7 +28,7 @@ export default async function patchUser(
 ) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/tools/removeUserFromGroup.ts

@@ -2,6 +2,7 @@ import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { getScimToken } from "../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../utils/getSCIMBaseUrl";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -26,7 +27,7 @@ export default async function removeUserFromGroup(
 ) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/tools/replaceGroup.ts

@@ -3,6 +3,7 @@ import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { groupResourceSchema } from "../schemas/groupResourceSchema";
 import { getScimToken } from "../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../utils/getSCIMBaseUrl";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -27,7 +28,7 @@ export default async function replaceGroup(
 ) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/tools/replaceUser.ts

@@ -3,6 +3,7 @@ import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { userResourceSchema } from "../schemas/userResourceSchema";
 import { getScimToken } from "../utils/getSCIMApiKey";
+import { getScimBaseUrl } from "../utils/getSCIMBaseUrl";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -27,7 +28,7 @@ export default async function replaceUser(
 ) {
   const requestHeaders = headers();
   const apiToken = getScimToken(requestHeaders);
-  const baseUrl = requestHeaders["x-scim-base-url"];
+  const baseUrl = getScimBaseUrl(requestHeaders);
 
   if (!apiToken) {
     throw new Error("Missing required headers: x-scim-api-key");

src/utils/getSCIMBaseUrl.ts

@@ -0,0 +1,15 @@
+type Headers = import("http").IncomingHttpHeaders;
+
+export function getScimBaseUrl(headers: Headers): string {
+  let baseUrl = "";
+
+  if (process.env.SCIM_API_BASE_URL) {
+    baseUrl = process.env.SCIM_API_BASE_URL;
+  }
+  const baseUrlFromHeader = headers["x-scim-base-url"] || headers["X-SCIM-BASE-URL"];
+  if (typeof baseUrlFromHeader === "string") {
+    baseUrl = baseUrlFromHeader.trim();
+  }
+  
+  return baseUrl;
+}