import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import type { Gist } from "#types";
import { GIST_URI_PREFIX, isContentLoaded, isDailyNoteGist, isPromptGist } from "#utils";
import type { FetchClient } from "./fetch.js";
const GISTS_PER_PAGE = 100;
function filterGists(gists: Gist[]): Gist[] {
return gists.filter((gist: Gist) => {
const files = Object.entries(gist.files);
return files.every(
([_, file]) => file.language === "Markdown" || file.filename.endsWith(".tldraw"),
);
});
}
export abstract class GistStore {
#cache: Gist[] | null = null;
#subscribedGists: Set<string> = new Set();
#server: McpServer;
#triggerNotifications: boolean;
#markdownOnly: boolean;
constructor(
protected fetchClient: FetchClient,
server: McpServer,
triggerNotifications: boolean = true,
markdownOnly: boolean = false,
) {
this.#server = server;
this.#triggerNotifications = triggerNotifications;
this.#markdownOnly = markdownOnly;
}
protected abstract fetchGists(): Promise<Gist[]>;
#notifyResourceListChanged(): void {
if (this.#triggerNotifications) {
this.#server.sendResourceListChanged();
}
}
#notifyResourceChanged(gistId: string): void {
if (this.#triggerNotifications && this.#subscribedGists.has(gistId)) {
// sendResourceUpdated is on the low-level Server, accessed via server.server
this.#server.server.sendResourceUpdated({ uri: `${GIST_URI_PREFIX}${gistId}` });
}
}
#notifyPromptListChanged() {
if (this.#triggerNotifications) {
this.#server.sendPromptListChanged();
}
}
async getAll(forceRefresh: boolean = false): Promise<Gist[]> {
if (this.#cache === null || forceRefresh) {
const gists = await this.fetchGists();
this.#cache = this.#markdownOnly ? filterGists(gists) : gists;
}
return this.#cache;
}
add(gist: Gist): void {
if (this.#cache && !this.#cache.some((g) => g.id === gist.id)) {
this.#cache.push(gist);
if (isPromptGist(gist)) {
this.#notifyPromptListChanged();
} else {
this.#notifyResourceListChanged();
}
}
}
remove(gistId: string) {
if (this.#cache) {
this.#cache = this.#cache.filter((g) => g.id !== gistId);
this.#notifyResourceListChanged();
}
}
update(gist: Gist) {
if (!this.#cache) return;
const index = this.#cache.findIndex((g) => g.id === gist.id);
if (index === -1) return;
const oldGist = this.#cache[index]!;
this.#cache[index] = gist;
// If the gist didn't actually change, we don't need to notify.
if (oldGist.updated_at === gist.updated_at) {
return;
}
if (isPromptGist(gist)) {
this.#notifyPromptListChanged();
} else {
// The resource list only includes the gist ID and
// description, so we only need to notify clients to
// update the list when a gist's description changes.
if (oldGist.description !== gist.description) {
this.#notifyResourceListChanged();
}
this.#notifyResourceChanged(gist.id);
}
}
async refresh() {
await this.getAll(true);
this.#notifyResourceListChanged();
}
subscribe(gistId: string) {
this.#subscribedGists.add(gistId);
}
unsubscribe(gistId: string) {
this.#subscribedGists.delete(gistId);
}
async ensureContentLoaded(gist: Gist): Promise<Gist> {
if (isContentLoaded(gist)) return gist;
const loadedGist = await this.fetchClient.get<Gist>(`/${gist.id}`);
this.update(loadedGist);
return loadedGist;
}
}
export class YourGistStore extends GistStore {
#dailyNotesGistId: string | null = null;
#promptsGistId: string | null = null;
async #getSpecialGist(gistId: string | null): Promise<Gist | null> {
if (!gistId) return null;
const gists = await this.getAll();
const gist = gists.find((g) => g.id === gistId);
return gist ? this.ensureContentLoaded(gist) : null;
}
async getDailyNotes(): Promise<Gist | null> {
return this.#getSpecialGist(this.#dailyNotesGistId);
}
async getPrompts(): Promise<Gist | null> {
return this.#getSpecialGist(this.#promptsGistId);
}
setDailyNotes(gist: Gist) {
this.add(gist);
this.#dailyNotesGistId = gist.id;
}
setPrompts(gist: Gist) {
this.add(gist);
this.#promptsGistId = gist.id;
}
protected async fetchGists(): Promise<Gist[]> {
const allGists: Gist[] = [];
let page = 1;
while (true) {
const gists = await this.fetchClient.get<Gist[]>("", {
params: {
per_page: GISTS_PER_PAGE,
page: page,
},
});
allGists.push(...gists);
if (gists.length < GISTS_PER_PAGE) {
break;
}
page++;
}
const dailyNotesGist = allGists.find(isDailyNoteGist);
if (dailyNotesGist) {
this.#dailyNotesGistId = dailyNotesGist.id;
}
const promptsGist = allGists.find(isPromptGist);
if (promptsGist) {
this.#promptsGistId = promptsGist.id;
}
return allGists;
}
}
export class StarredGistStore extends GistStore {
protected async fetchGists(): Promise<Gist[]> {
return this.fetchClient.get<Gist[]>("/starred");
}
}
