AdMob API MCP 서버 만들기 - Part 3: 애드몹 계정 정보를 도구로 제공하기

// src/tools.ts

enum ToolName {
  GET_ACCOUNT = "get_account",
  LIST_AD_UNITS = "list_ad_units",
  LIST_APPS = "list_apps",
}

// src/tools.ts
import type { Server } from "@modelcontextprotocol/sdk/server/index.js";
import { ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";

// ToolName 선언부 아래에 추가합니다.

export function configureTools(server: Server): void {
  // "사용 가능한 도구 목록 좀 알려줘!" 라는 요청을 처리하는 핸들러
  server.setRequestHandler(ListToolsRequestSchema, async () => {
    return {
      tools: [
        {
          name: ToolName.GET_ACCOUNT,
          description: "Gets information about the AdMob publisher account.",
          inputSchema: {
            type: "object",
            properties: {},
          },
        },
        {
          name: ToolName.LIST_AD_UNITS,
          description: "List the ad units under the AdMob publisher account.",
          inputSchema: {
            type: "object",
            properties: {},
          },
        },
        {
          name: ToolName.LIST_APPS,
          description: "List the apps under the AdMob publisher account.",
          inputSchema: {
            type: "object",
            properties: {},
          },
        },
      ],
    };
  });
}

// src/tools.ts
// CallToolRequestSchema 와 google, admob_v1 등을 추가로 import 합니다.
import type { Server } from "@modelcontextprotocol/sdk/server/index.js";
import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
import { google, admob_v1 } from "googleapis";
import { loadSavedCredentialsIfExist } from "./auth.js";

// ... ToolName enum

export function configureTools(server: Server): void {
  // ... ListToolsRequestSchema 핸들러

  // "이 도구를 실행해줘!" 라는 요청을 처리하는 핸들러
  server.setRequestHandler(CallToolRequestSchema, async request => {
    // 1. 인증 정보 확인
    const auth = await loadSavedCredentialsIfExist();
    if (!auth) {
      return {
        content: [{ type: "text", text: "Not Authenticated. Try running `npm run auth` to authenticate." }],
        isError: true,
      };
    }

    // 2. 게시자 코드 확인
    const publisherCode = process.env.PUBLISHER_CODE;
    if (!publisherCode) {
      return {
        content: [{ type: "text", text: "PUBLISHER_CODE environment variable not set." }],
        isError: true,
      };
    }

    const { name } = request.params;
    const admob = google.admob({ version: "v1", auth });

    // 3. 요청된 도구 이름에 따라 분기 처리
    if (name === ToolName.GET_ACCOUNT) {
      const result = await admob.accounts.get({ name: `accounts/${publisherCode}` });
      if (!result.ok) {
        return {
          content: [{ type: "text", text: `API request failed: ${result.status}` }],
          isError: true,
        };
      }
      const data = result.data;
      return {
        content: [{ type: "text", text: JSON.stringify({
          name: data.name,
          publisherId: data.publisherId,
          reportingTimeZone: data.reportingTimeZone,
          currencyCode: data.currencyCode,
        }, null, 2)}],
      };
    }

    if (name === ToolName.LIST_AD_UNITS) {
      const allAdUnits: admob_v1.Schema$AdUnit[] = [];
      let pageToken: string | undefined;

      try {
        // API가 페이지 단위로 데이터를 주기 때문에, nextPageToken이 없을 때까지 반복 호출합니다.
        do {
          const result = await admob.accounts.adUnits.list({
            parent: `accounts/${publisherCode}`,
            pageToken: pageToken,
          });

          if (!result.ok) {
            return {
              content: [{ type: "text", text: `API request failed with status: ${result.status}` }],
              isError: true,
            };
          }
          const data = result.data;
          if (data.adUnits) {
            allAdUnits.push(...data.adUnits);
          }
          pageToken = data.nextPageToken ?? undefined;
        } while (pageToken);

        return {
          content: [{ type: "text", text: JSON.stringify(allAdUnits, null, 2) }],
        };
      } catch (error) {
        return {
          content: [{ type: "text", text: `An error occurred: ${error}` }],
          isError: true,
        };
      }
    }

    if (name === ToolName.LIST_APPS) {
      const allApps: admob_v1.Schema$App[] = [];
      let pageToken: string | undefined;

      try {
        // 광고 단위와 마찬가지로, 모든 앱 목록을 가져오기 위해 반복 호출합니다.
        do {
          const result = await admob.accounts.apps.list({
            parent: `accounts/${publisherCode}`,
            pageToken: pageToken,
          });

          if (!result.ok) {
            return {
              content: [{ type: "text", text: `API request failed with status: ${result.status}` }],
              isError: true,
            };
          }
          const data = result.data;
          if (data.apps) {
            allApps.push(...data.apps);
          }
          pageToken = data.nextPageToken ?? undefined;
        } while (pageToken);

        return {
          content: [{ type: "text", text: JSON.stringify(allApps, null, 2) }],
        };
      } catch (error) {
        return {
          content: [{ type: "text", text: `An error occurred: ${error}` }],
          isError: true,
        };
      }
    }

    // 정의되지 않은 도구가 호출되면 에러를 발생시킵니다.
    throw new Error(`Unknown tool: ${name}`);
  });
}

// src/index.ts

// .. 다른 import 문은 그대로 둡니다.
import { Server } from "@modelcontextprotocol/sdk/server/index.js";
import { configureTools } from "./tools.js"; // 이 부분을 추가해 주세요.

const server = new Server(...);

// 서버가 도구를 사용할 수 있도록 설정합니다.
configureTools(server);

// ... 그 외 다른 부분은 그대로 둡니다.

// src/index.ts

import * as dotenv from "dotenv";

dotenv.config();

// ... 나머지 코드는 그대로 둡니다.