116 lines
2.9 KiB
TypeScript
116 lines
2.9 KiB
TypeScript
import { Server } from '@modelcontextprotocol/sdk/server/index.js'
|
|
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
|
|
import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js'
|
|
import { addToSizeWhitelist, archiveVideo, getVideoMetadata } from '@/utils/archive'
|
|
|
|
const server = new Server(
|
|
{
|
|
name: 'preservetube-mcp',
|
|
version: '1.0.0'
|
|
},
|
|
{
|
|
capabilities: {
|
|
tools: {}
|
|
}
|
|
}
|
|
)
|
|
|
|
server.setRequestHandler(ListToolsRequestSchema, async () => ({
|
|
tools: [
|
|
{
|
|
name: 'add_to_size_whitelist',
|
|
description: 'Add a YouTube video ID to the size whitelist in preservetube-metadata config.json to bypass max video size limits.',
|
|
inputSchema: {
|
|
type: 'object',
|
|
properties: {
|
|
videoId: {
|
|
type: 'string',
|
|
description: 'YouTube video ID (11 characters) or full YouTube URL'
|
|
}
|
|
},
|
|
required: ['videoId']
|
|
}
|
|
},
|
|
{
|
|
name: 'archive_video',
|
|
description: 'Archive a YouTube video by ID or URL on PreserveTube.',
|
|
inputSchema: {
|
|
type: 'object',
|
|
properties: {
|
|
videoId: {
|
|
type: 'string',
|
|
description: 'YouTube video ID (11 characters) or full YouTube URL'
|
|
}
|
|
},
|
|
required: ['videoId']
|
|
}
|
|
},
|
|
{
|
|
name: 'get_video_metadata',
|
|
description: 'Fetch video metadata including title, duration/length, channel, publish date, and preserved database record if archived.',
|
|
inputSchema: {
|
|
type: 'object',
|
|
properties: {
|
|
videoId: {
|
|
type: 'string',
|
|
description: 'YouTube video ID (11 characters) or full YouTube URL'
|
|
}
|
|
},
|
|
required: ['videoId']
|
|
}
|
|
}
|
|
]
|
|
}))
|
|
|
|
server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
const { name, arguments: args } = request.params
|
|
const inputArg = args && typeof args === 'object' && ('videoId' in args || 'url' in args)
|
|
? String((args as Record<string, unknown>).videoId || (args as Record<string, unknown>).url || '')
|
|
: ''
|
|
|
|
if (name === 'add_to_size_whitelist') {
|
|
const result = await addToSizeWhitelist(inputArg)
|
|
return {
|
|
content: [
|
|
{
|
|
type: 'text',
|
|
text: JSON.stringify(result, null, 2)
|
|
}
|
|
]
|
|
}
|
|
}
|
|
|
|
if (name === 'archive_video') {
|
|
const result = await archiveVideo(inputArg)
|
|
return {
|
|
content: [
|
|
{
|
|
type: 'text',
|
|
text: JSON.stringify(result, null, 2)
|
|
}
|
|
]
|
|
}
|
|
}
|
|
|
|
if (name === 'get_video_metadata') {
|
|
const result = await getVideoMetadata(inputArg)
|
|
return {
|
|
content: [
|
|
{
|
|
type: 'text',
|
|
text: JSON.stringify(result, null, 2)
|
|
}
|
|
]
|
|
}
|
|
}
|
|
|
|
throw new Error(`Unknown tool: ${name}`)
|
|
})
|
|
|
|
async function run() {
|
|
const transport = new StdioServerTransport()
|
|
await server.connect(transport)
|
|
}
|
|
|
|
run().catch(console.error)
|