Author: Joe

#!/usr/bin/env node

const https = require('https');

const WP_SITE_URL = 'YOUR_SITE_URL';
const CLIENT_ID = 'YOUR_CLIENT_ID';
const CLIENT_SECRET = 'YOUR_CLIENT_SECRET';

const WP_MCP_URL = `${WP_SITE_URL}/wp-json/ai-system-admin-mcp/v1/mcp`;
const TOKEN_URL = `${WP_SITE_URL}/wp-json/ai-system-admin-mcp/v1/oauth/token`;

let accessToken = null;
let tokenExpiry = null;
let buffer = '';

async function getAccessToken() {
  if (accessToken && tokenExpiry && Date.now() < tokenExpiry) {
    return accessToken;
  }

  return new Promise((resolve, reject) => {
    const postData = new URLSearchParams({
      grant_type: 'client_credentials',
      client_id: CLIENT_ID,
      client_secret: CLIENT_SECRET
    }).toString();

    const options = {
      method: 'POST',
      headers: {
        'Content-Type': 'application/x-www-form-urlencoded',
        'Content-Length': Buffer.byteLength(postData)
      }
    };

    const req = https.request(TOKEN_URL, options, (res) => {
      let data = '';

      res.on('data', (chunk) => {
        data += chunk;
      });

      res.on('end', () => {
        try {
          const response = JSON.parse(data);
          if (response.access_token) {
            accessToken = response.access_token;
            tokenExpiry = Date.now() + ((response.expires_in - 300) * 1000);
            resolve(accessToken);
          } else {
            reject(new Error(`OAuth error: ${JSON.stringify(response)}`));
          }
        } catch (e) {
          reject(new Error(`Failed to parse token response: ${e.message}`));
        }
      });
    });

    req.on('error', (e) => {
      reject(new Error(`OAuth request failed: ${e.message}`));
    });

    req.write(postData);
    req.end();
  });
}

process.stdin.setEncoding('utf8');
process.stdin.on('data', (chunk) => {
  buffer += chunk;

  const lines = buffer.split('\n');
  buffer = lines.pop() || '';

  lines.forEach(line => {
    if (line.trim()) {
      try {
        const request = JSON.parse(line);
        handleRequest(request);
      } catch (e) {
        sendError(null, -32700, 'Parse error', e.message);
      }
    }
  });
});

function handleRequest(request) {
  const { id, method, params } = request;

  if (method === 'initialize') {
    sendResponse(id, {
      protocolVersion: '2024-11-05',
      capabilities: {
        tools: {}
      },
      serverInfo: {
        name: 'wordpress-mcp',
        version: '1.0.0'
      }
    });
    return;
  }

  if (method === 'tools/list') {
    forwardToWordPress(request, id);
    return;
  }

  if (method === 'tools/call') {
    forwardToWordPress(request, id);
    return;
  }

  sendError(id, -32601, 'Method not found');
}

async function forwardToWordPress(request, id) {
  try {
    const token = await getAccessToken();
    const postData = JSON.stringify(request);

    const options = {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
        'Content-Length': Buffer.byteLength(postData),
        'Authorization': `Bearer ${token}`
      }
    };

    const req = https.request(WP_MCP_URL, options, (res) => {
      let data = '';

      res.on('data', (chunk) => {
        data += chunk;
      });

      res.on('end', () => {
        try {
          const response = JSON.parse(data);
          sendResponse(id, response.result || response);
        } catch (e) {
          sendError(id, -32603, 'Internal error', `Failed to parse WordPress response: ${e.message}`);
        }
      });
    });

    req.on('error', (e) => {
      sendError(id, -32603, 'Internal error', `Request to WordPress failed: ${e.message}`);
    });

    req.write(postData);
    req.end();
  } catch (e) {
    sendError(id, -32603, 'OAuth error', e.message);
  }
}

function sendResponse(id, result) {
  const response = {
    jsonrpc: '2.0',
    id: id,
    result: result
  };
  process.stdout.write(JSON.stringify(response) + '\n');
}

function sendError(id, code, message, data = null) {
  const response = {
    jsonrpc: '2.0',
    id: id,
    error: {
      code: code,
      message: message,
      ...(data && { data })
    }
  };
  process.stdout.write(JSON.stringify(response) + '\n');
}

console.error('WordPress MCP Bridge started');
console.error('Connecting to:', WP_MCP_URL);
console.error('OAuth endpoint:', TOKEN_URL);