chore: refactor into getSCIMToken

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

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

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

src/resources/(groups)/index.ts

@@ -1,6 +1,7 @@
 import { type InferSchema, type ResourceMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
+import { getScimToken } from "../../utils/getSCIMApiKey";
 
 export const schema = {
   filter: z
@@ -35,7 +36,7 @@ export default async function handler({
   count,
 }: InferSchema<typeof schema>) {
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

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

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

src/resources/(users)/index.ts

@@ -1,6 +1,7 @@
 import { type InferSchema, type ResourceMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
+import { getScimToken } from "../../utils/getSCIMApiKey";
 
 export const schema = {
   filter: z
@@ -35,7 +36,7 @@ export default async function handler({
   count,
 }: InferSchema<typeof schema>) {
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/tools/addUserToGroup.ts

@@ -1,6 +1,7 @@
 import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
+import { getScimToken } from "../utils/getSCIMApiKey";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -24,7 +25,7 @@ export default async function addUserToGroup(
   params: InferSchema<typeof schema>
 ) {
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/tools/createGroup.ts

@@ -1,7 +1,8 @@
 import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
-import { groupResourceSchema } from "../schemas/groupResourceSchema";
 import { z } from "zod";
+import { groupResourceSchema } from "../schemas/groupResourceSchema";
+import { getScimToken } from "../utils/getSCIMApiKey";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -30,7 +31,7 @@ export default async function createGroup(
 ) {
   const { groupResource } = params;
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/tools/createUser.ts

@@ -2,6 +2,7 @@ import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
 import { userResourceSchema } from "../schemas/userResourceSchema";
+import { getScimToken } from "../utils/getSCIMApiKey";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -30,7 +31,7 @@ export default async function createUser(
 ) {
   const { userResource } = params;
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/tools/deleteGroup.ts

@@ -1,6 +1,7 @@
 import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
+import { getScimToken } from "../utils/getSCIMApiKey";
 
 export const metadata: ToolMetadata = {
   name: "delete-group",
@@ -22,7 +23,7 @@ export default async function deleteGroup(
   params: InferSchema<typeof schema>
 ) {
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/tools/deleteUser.ts

@@ -1,6 +1,7 @@
 import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
+import { getScimToken } from "../utils/getSCIMApiKey";
 
 export const metadata: ToolMetadata = {
   name: "delete-user",
@@ -22,7 +23,7 @@ export default async function deleteUser(
   params: InferSchema<typeof schema>
 ) {
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/tools/patchGroup.ts

@@ -1,7 +1,8 @@
 import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
-import { patchOperationSchema } from "../schemas/patchOperationSchema";
 import { z } from "zod";
+import { patchOperationSchema } from "../schemas/patchOperationSchema";
+import { getScimToken } from "../utils/getSCIMApiKey";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -25,7 +26,7 @@ export default async function patchGroup(
   params: InferSchema<typeof schema>
 ) {
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/tools/patchUser.ts

@@ -1,7 +1,8 @@
 import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
-import { patchOperationSchema } from "../schemas/patchOperationSchema";
 import { z } from "zod";
+import { patchOperationSchema } from "../schemas/patchOperationSchema";
+import { getScimToken } from "../utils/getSCIMApiKey";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -25,7 +26,7 @@ export default async function patchUser(
   params: InferSchema<typeof schema>
 ) {
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/tools/removeUserFromGroup.ts

@@ -1,6 +1,7 @@
 import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
 import { z } from "zod";
+import { getScimToken } from "../utils/getSCIMApiKey";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -24,7 +25,7 @@ export default async function removeUserFromGroup(
   params: InferSchema<typeof schema>
 ) {
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/tools/replaceGroup.ts

@@ -1,7 +1,8 @@
 import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
-import { groupResourceSchema } from "../schemas/groupResourceSchema";
 import { z } from "zod";
+import { groupResourceSchema } from "../schemas/groupResourceSchema";
+import { getScimToken } from "../utils/getSCIMApiKey";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -25,7 +26,7 @@ export default async function replaceGroup(
   params: InferSchema<typeof schema>
 ) {
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/tools/replaceUser.ts

@@ -1,7 +1,8 @@
 import { type InferSchema, type ToolMetadata } from "xmcp";
 import { headers } from "xmcp/headers";
-import { userResourceSchema } from "../schemas/userResourceSchema";
 import { z } from "zod";
+import { userResourceSchema } from "../schemas/userResourceSchema";
+import { getScimToken } from "../utils/getSCIMApiKey";
 import { readJsonBody } from "../utils/responseBody";
 
 export const metadata: ToolMetadata = {
@@ -25,7 +26,7 @@ export default async function replaceUser(
   params: InferSchema<typeof schema>
 ) {
   const requestHeaders = headers();
-  const apiToken = requestHeaders["x-scim-api-key"];
+  const apiToken = getScimToken(requestHeaders);
   const baseUrl = requestHeaders["x-scim-base-url"];
 
   if (!apiToken) {

src/utils/getSCIMApiKey.ts

@@ -0,0 +1,17 @@
+
+
+type Headers = import("http").IncomingHttpHeaders;
+
+export function getScimToken(headers: Headers): string {
+  let token = "";
+
+  if (process.env.SCIM_API_TOKEN) {
+    token = process.env.SCIM_API_TOKEN;
+  }
+  const tokenFromHeader = headers["x-scim-api-key"] || headers["X-SCIM-API-KEY"];
+  if (typeof tokenFromHeader === "string") {
+    token = tokenFromHeader.trim();
+  }
+  
+  return token;
+}