WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

internal/agent/tools/wiki_rename_page.go

@@ -0,0 +1,177 @@
+package tools
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"strings"
+
+	"github.com/Tencent/WeKnora/internal/types"
+	"github.com/Tencent/WeKnora/internal/types/interfaces"
+)
+
+type wikiRenamePageTool struct {
+	BaseTool
+	wikiPageService interfaces.WikiPageService
+	kbIDs           []string
+	routes          *WikiRouteResolver
+}
+
+// NewWikiRenamePageTool creates a new wiki_rename_page tool
+func NewWikiRenamePageTool(
+	wikiPageService interfaces.WikiPageService,
+	kbIDs []string,
+	routes ...*WikiRouteResolver,
+) types.Tool {
+	return &wikiRenamePageTool{
+		BaseTool: NewBaseTool(
+			ToolWikiRenamePage,
+			"Rename a Wiki page's slug. Automatically cascades the new slug to all pages that linked to the old one.",
+			json.RawMessage(`{
+				"type": "object",
+				"properties": {
+					"slug": {
+						"type": "string",
+						"description": "The current slug of the Wiki page"
+					},
+					"new_slug": {
+						"type": "string",
+						"description": "The new slug for the page"
+					}
+				},
+				"required": ["slug", "new_slug"]
+			}`),
+		),
+		wikiPageService: wikiPageService,
+		kbIDs:           kbIDs,
+		routes:          firstWikiRoute(routes),
+	}
+}
+
+func (t *wikiRenamePageTool) Execute(ctx context.Context, args json.RawMessage) (*types.ToolResult, error) {
+	// Attribute every page write performed by this tool to the agent so
+	// revision history distinguishes agent edits from pipeline/user ones.
+	ctx = types.WithWikiEditSource(ctx, types.WikiEditSourceAgent)
+	var params struct {
+		Slug    string `json:"slug"`
+		NewSlug string `json:"new_slug"`
+	}
+
+	if err := json.Unmarshal(args, &params); err != nil {
+		return &types.ToolResult{Success: false, Error: "Failed to parse arguments: " + err.Error()}, nil
+	}
+
+	if len(t.kbIDs) == 0 {
+		return &types.ToolResult{Success: false, Error: "No knowledge bases available for editing"}, nil
+	}
+	if params.NewSlug == "" {
+		return &types.ToolResult{Success: false, Error: "new_slug is required"}, nil
+	}
+	normalizedSlug, slugErr := normalizeAndValidateWikiSlug(params.Slug)
+	if slugErr != nil {
+		return &types.ToolResult{Success: false, Error: slugErr.Error()}, nil
+	}
+	params.Slug = normalizedSlug
+	normalizedNewSlug, slugErr := normalizeAndValidateWikiSlug(params.NewSlug)
+	if slugErr != nil {
+		return &types.ToolResult{Success: false, Error: slugErr.Error()}, nil
+	}
+	params.NewSlug = normalizedNewSlug
+	if params.NewSlug == params.Slug {
+		return &types.ToolResult{Success: false, Error: "new_slug must be different from old slug"}, nil
+	}
+
+	// Get existing page
+	existingPage, kbID, err := resolveUniqueWikiPage(ctx, t.wikiPageService, params.Slug, t.kbIDs, t.routes)
+	if err != nil {
+		return &types.ToolResult{Success: false, Error: "Failed to resolve page to rename: " + err.Error()}, nil
+	}
+
+	inLinks := make([]string, len(existingPage.InLinks))
+	copy(inLinks, existingPage.InLinks)
+
+	// Create new page with new slug but same content
+	newPage := &types.WikiPage{
+		TenantID:        existingPage.TenantID,
+		KnowledgeBaseID: kbID,
+		Slug:            params.NewSlug,
+		Title:           existingPage.Title,
+		Summary:         existingPage.Summary,
+		Content:         existingPage.Content,
+		PageType:        existingPage.PageType,
+		Status:          existingPage.Status,
+		Aliases:         append(types.StringArray(nil), existingPage.Aliases...),
+		ParentSlug:      existingPage.ParentSlug,
+		FolderID:        existingPage.FolderID,
+		SortOrder:       existingPage.SortOrder,
+		SourceRefs:      append(types.StringArray(nil), existingPage.SourceRefs...),
+		ChunkRefs:       append(types.StringArray(nil), existingPage.ChunkRefs...),
+		InLinks:         append(types.StringArray(nil), existingPage.InLinks...),
+		PageMetadata:    append(types.JSON(nil), existingPage.PageMetadata...),
+	}
+	_, err = t.wikiPageService.CreatePage(ctx, newPage)
+	if err != nil {
+		return &types.ToolResult{Success: false, Error: "Failed to create renamed page: " + err.Error()}, nil
+	}
+
+	changes, updatedSlugs, rewriteErr := applyIncomingWikiContentRewrite(
+		ctx, t.wikiPageService, kbID, inLinks,
+		func(content string) (string, bool) {
+			updated := strings.ReplaceAll(
+				content, "[["+params.Slug+"]]", "[["+params.NewSlug+"]]",
+			)
+			updated = strings.ReplaceAll(
+				updated, "[["+params.Slug+"|", "[["+params.NewSlug+"|",
+			)
+			return updated, updated != content
+		},
+	)
+	if rewriteErr != nil {
+		rollbackErr := rollbackWikiContentChanges(ctx, t.wikiPageService, changes)
+		cleanupErr := t.wikiPageService.DeletePage(ctx, kbID, params.NewSlug)
+		return &types.ToolResult{
+			Success: false,
+			Error: "Rename aborted while updating incoming links: " +
+				joinWikiMutationErrors(rewriteErr, rollbackErr, cleanupErr),
+		}, nil
+	}
+	updatedCount := len(updatedSlugs)
+
+	// Delete old page
+	err = t.wikiPageService.DeletePage(ctx, kbID, params.Slug)
+	if err != nil {
+		rollbackErr := rollbackWikiContentChanges(ctx, t.wikiPageService, changes)
+		cleanupErr := t.wikiPageService.DeletePage(ctx, kbID, params.NewSlug)
+		return &types.ToolResult{
+			Success: false,
+			Error: "Rename aborted because the old page could not be deleted: " +
+				joinWikiMutationErrors(err, rollbackErr, cleanupErr),
+		}, nil
+	}
+	t.routes.forget(params.Slug, kbID)
+	t.routes.remember(params.NewSlug, kbID)
+
+	// Inject cross-links so other pages know about this new slug
+	t.wikiPageService.InjectCrossLinks(ctx, kbID, []string{params.NewSlug})
+
+	// Rebuild the index page to reflect the new/updated summary
+	_ = t.wikiPageService.RebuildIndexPage(ctx, kbID)
+
+	outputMsg := fmt.Sprintf("Successfully renamed page [[%s]] → [[%s]] and updated %d incoming links.", params.Slug, params.NewSlug, updatedCount)
+	if updatedCount > 0 {
+		outputMsg += fmt.Sprintf("\n- Affected pages: %s", strings.Join(updatedSlugs, ", "))
+	}
+
+	return &types.ToolResult{
+		Success: true,
+		Output:  outputMsg,
+		Data: map[string]interface{}{
+			"display_type":   "wiki_rename_page",
+			"old_slug":       params.Slug,
+			"new_slug":       params.NewSlug,
+			"title":          existingPage.Title,
+			"updated_count":  updatedCount,
+			"affected_pages": updatedSlugs,
+		},
+	}, nil
+}