got them all

Author: kentcdodds

exercises/99.finished/01.solution.finished/src/prompts.ts

@@ -4,7 +4,7 @@ import { z } from 'zod'
 import { type EpicMeMCP } from './index.ts'
 
 export async function initializePrompts(agent: EpicMeMCP) {
-	agent.server.registerPrompt(
+	const suggestTagsPrompt = agent.server.registerPrompt(
 		'suggest_tags',
 		{
 			title: 'Suggest Tags',
@@ -73,4 +73,15 @@ For each tag I approve, if it does not yet exist, create it with the EpicMe "cre
 			}
 		},
 	)
+
+	async function updatePrompts() {
+		const entries = await agent.db.getEntries()
+		if (entries.length > 0) {
+			if (!suggestTagsPrompt.enabled) suggestTagsPrompt.enable()
+		} else {
+			if (suggestTagsPrompt.enabled) suggestTagsPrompt.disable()
+		}
+	}
+	agent.db.subscribe(updatePrompts)
+	await updatePrompts()
 }

exercises/99.finished/01.solution.finished/src/tools.ts

@@ -57,7 +57,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 		},
 	)
 
-	agent.server.registerTool(
+	const getEntryTool = agent.server.registerTool(
 		'get_entry',
 		{
 			title: 'Get Entry',
@@ -82,7 +82,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 		},
 	)
 
-	agent.server.registerTool(
+	const listEntriesTool = agent.server.registerTool(
 		'list_entries',
 		{
 			title: 'List Entries',
@@ -104,7 +104,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 		},
 	)
 
-	agent.server.registerTool(
+	const updateEntryTool = agent.server.registerTool(
 		'update_entry',
 		{
 			title: 'Update Entry',
@@ -134,7 +134,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 		},
 	)
 
-	agent.server.registerTool(
+	const deleteEntryTool = agent.server.registerTool(
 		'delete_entry',
 		{
 			title: 'Delete Entry',
@@ -211,7 +211,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 		},
 	)
 
-	agent.server.registerTool(
+	const getTagTool = agent.server.registerTool(
 		'get_tag',
 		{
 			title: 'Get Tag',
@@ -236,7 +236,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 		},
 	)
 
-	agent.server.registerTool(
+	const listTagsTool = agent.server.registerTool(
 		'list_tags',
 		{
 			title: 'List Tags',
@@ -255,7 +255,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 		},
 	)
 
-	agent.server.registerTool(
+	const updateTagTool = agent.server.registerTool(
 		'update_tag',
 		{
 			title: 'Update Tag',
@@ -282,7 +282,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 		},
 	)
 
-	agent.server.registerTool(
+	const deleteTagTool = agent.server.registerTool(
 		'delete_tag',
 		{
 			title: 'Delete Tag',
@@ -333,7 +333,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 		},
 	)
 
-	agent.server.registerTool(
+	const addTagToEntryTool = agent.server.registerTool(
 		'add_tag_to_entry',
 		{
 			title: 'Add Tag to Entry',
@@ -371,7 +371,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 		},
 	)
 
-	agent.server.registerTool(
+	const createWrappedVideoTool = agent.server.registerTool(
 		'create_wrapped_video',
 		{
 			title: 'Create Wrapped Video',
@@ -437,6 +437,44 @@ export async function initializeTools(agent: EpicMeMCP) {
 			}
 		},
 	)
+
+	async function updateTools() {
+		const entries = await agent.db.getEntries()
+		if (entries.length > 0) {
+			if (!deleteEntryTool.enabled) deleteEntryTool.enable()
+			if (!updateEntryTool.enabled) updateEntryTool.enable()
+			if (!listEntriesTool.enabled) listEntriesTool.enable()
+			if (!getEntryTool.enabled) getEntryTool.enable()
+			if (!createWrappedVideoTool.enabled) createWrappedVideoTool.enable()
+		} else {
+			if (deleteEntryTool.enabled) deleteEntryTool.disable()
+			if (updateEntryTool.enabled) updateEntryTool.disable()
+			if (listEntriesTool.enabled) listEntriesTool.disable()
+			if (getEntryTool.enabled) getEntryTool.disable()
+			if (createWrappedVideoTool.enabled) createWrappedVideoTool.disable()
+		}
+
+		const tags = await agent.db.getTags()
+		if (tags.length > 0) {
+			if (!deleteTagTool.enabled) deleteTagTool.enable()
+			if (!updateTagTool.enabled) updateTagTool.enable()
+			if (!listTagsTool.enabled) listTagsTool.enable()
+			if (!getTagTool.enabled) getTagTool.enable()
+		} else {
+			if (deleteTagTool.enabled) deleteTagTool.disable()
+			if (updateTagTool.enabled) updateTagTool.disable()
+			if (listTagsTool.enabled) listTagsTool.disable()
+			if (getTagTool.enabled) getTagTool.disable()
+		}
+
+		if (entries.length > 0 && tags.length > 0) {
+			if (!addTagToEntryTool.enabled) addTagToEntryTool.enable()
+		} else {
+			if (addTagToEntryTool.enabled) addTagToEntryTool.disable()
+		}
+	}
+	agent.db.subscribe(updateTools)
+	await updateTools()
 }
 
 function createTextContent(text: unknown): CallToolResult['content'][number] {