import { Effect, Schema } from "effect"

import { HttpClient } from "effect/unstable/http"

import * as Tool from "./tool"

import * as McpWebSearch from "./mcp-websearch"

import DESCRIPTION from "./codesearch.txt"

export const Parameters = Schema.Struct({

query: Schema.String.annotate({

description:

"Search query to find relevant context for APIs, Libraries, and SDKs. For example, 'React useState hook examples', 'Python pandas dataframe filtering', 'Express.js middleware', 'Next js partial prerendering configuration'",

}),

tokensNum: Schema.Number.check(Schema.isGreaterThanOrEqualTo(1000))

.check(Schema.isLessThanOrEqualTo(50000))

.pipe(Schema.optional, Schema.withDecodingDefault(Effect.succeed(5000)))

.annotate({

description:

"Number of tokens to return (1000-50000). Default is 5000 tokens. Adjust this value based on how much context you need - use lower values for focused queries and higher values for comprehensive documentation.",

}),

})

export const CodeSearchTool = Tool.define(

"codesearch",

Effect.gen(function* () {

const http = yield* HttpClient.HttpClient

return {

description: DESCRIPTION,

parameters: Parameters,

execute: (params: { query: string; tokensNum: number }, ctx: Tool.Context) =>

Effect.gen(function* () {

yield* ctx.ask({

permission: "codesearch",

patterns: [params.query],

always: ["*"],

metadata: {

query: params.query,

tokensNum: params.tokensNum,

},

})

const result = yield* McpWebSearch.call(

http,

McpWebSearch.EXA_URL,

"get_code_context_exa",

McpWebSearch.CodeArgs,

{

query: params.query,

tokensNum: params.tokensNum,

},

"30 seconds",

)

return {

output:

result ??

"No code snippets or documentation found. Please try a different query, be more specific about the library or programming concept, or check the spelling of framework names.",

title: `Code search: ${params.query}`,

metadata: {},

}

}).pipe(Effect.orDie),

}

}),

)