Initial commit: Open sourcing all of the Maple Open Technologies code.

cloud/maplepress-backend/pkg/search/index.go

@@ -0,0 +1,216 @@
+// File Path: monorepo/cloud/maplepress-backend/pkg/search/index.go
+package search
+
+import (
+	"fmt"
+
+	"github.com/meilisearch/meilisearch-go"
+)
+
+// PageDocument represents a document in the search index
+type PageDocument struct {
+	ID          string `json:"id"`        // page_id
+	SiteID      string `json:"site_id"`   // for filtering (though each site has its own index)
+	TenantID    string `json:"tenant_id"` // for additional isolation
+	Title       string `json:"title"`
+	Content     string `json:"content"` // HTML stripped
+	Excerpt     string `json:"excerpt"`
+	URL         string `json:"url"`
+	Status      string `json:"status"`    // publish, draft, trash
+	PostType    string `json:"post_type"` // page, post
+	Author      string `json:"author"`
+	PublishedAt int64  `json:"published_at"` // Unix timestamp for sorting
+	ModifiedAt  int64  `json:"modified_at"`  // Unix timestamp for sorting
+}
+
+// CreateIndex creates a new index for a site
+func (c *Client) CreateIndex(siteID string) error {
+	indexName := c.GetIndexName(siteID)
+
+	// Create index with site_id as primary key
+	_, err := c.client.CreateIndex(&meilisearch.IndexConfig{
+		Uid:        indexName,
+		PrimaryKey: "id", // page_id is the primary key
+	})
+
+	if err != nil {
+		return fmt.Errorf("failed to create index %s: %w", indexName, err)
+	}
+
+	// Configure index settings
+	return c.ConfigureIndex(siteID)
+}
+
+// ConfigureIndex configures the index settings
+func (c *Client) ConfigureIndex(siteID string) error {
+	indexName := c.GetIndexName(siteID)
+	index := c.client.Index(indexName)
+
+	// Set searchable attributes (in order of priority)
+	searchableAttributes := []string{
+		"title",
+		"excerpt",
+		"content",
+	}
+
+	_, err := index.UpdateSearchableAttributes(&searchableAttributes)
+	if err != nil {
+		return fmt.Errorf("failed to set searchable attributes: %w", err)
+	}
+
+	// Set filterable attributes
+	filterableAttributes := []interface{}{
+		"status",
+		"post_type",
+		"author",
+		"published_at",
+	}
+
+	_, err = index.UpdateFilterableAttributes(&filterableAttributes)
+	if err != nil {
+		return fmt.Errorf("failed to set filterable attributes: %w", err)
+	}
+
+	// Set ranking rules
+	rankingRules := []string{
+		"words",
+		"typo",
+		"proximity",
+		"attribute",
+		"sort",
+		"exactness",
+	}
+
+	_, err = index.UpdateRankingRules(&rankingRules)
+	if err != nil {
+		return fmt.Errorf("failed to set ranking rules: %w", err)
+	}
+
+	// Set displayed attributes (don't return full content in search results)
+	displayedAttributes := []string{
+		"id",
+		"title",
+		"excerpt",
+		"url",
+		"status",
+		"post_type",
+		"author",
+		"published_at",
+		"modified_at",
+	}
+
+	_, err = index.UpdateDisplayedAttributes(&displayedAttributes)
+	if err != nil {
+		return fmt.Errorf("failed to set displayed attributes: %w", err)
+	}
+
+	return nil
+}
+
+// IndexExists checks if an index exists
+func (c *Client) IndexExists(siteID string) (bool, error) {
+	indexName := c.GetIndexName(siteID)
+
+	_, err := c.client.GetIndex(indexName)
+	if err != nil {
+		// Check if error is "index not found" (status code 404)
+		if meiliErr, ok := err.(*meilisearch.Error); ok {
+			if meiliErr.StatusCode == 404 {
+				return false, nil
+			}
+		}
+		return false, fmt.Errorf("failed to check index existence: %w", err)
+	}
+
+	return true, nil
+}
+
+// DeleteIndex deletes an index for a site
+func (c *Client) DeleteIndex(siteID string) error {
+	indexName := c.GetIndexName(siteID)
+
+	_, err := c.client.DeleteIndex(indexName)
+	if err != nil {
+		return fmt.Errorf("failed to delete index %s: %w", indexName, err)
+	}
+
+	return nil
+}
+
+// AddDocuments adds or updates documents in the index
+func (c *Client) AddDocuments(siteID string, documents []PageDocument) (*meilisearch.TaskInfo, error) {
+	indexName := c.GetIndexName(siteID)
+	index := c.client.Index(indexName)
+
+	taskInfo, err := index.AddDocuments(documents, nil)
+	if err != nil {
+		return nil, fmt.Errorf("failed to add documents to index %s: %w", indexName, err)
+	}
+
+	return taskInfo, nil
+}
+
+// UpdateDocuments updates documents in the index
+func (c *Client) UpdateDocuments(siteID string, documents []PageDocument) (*meilisearch.TaskInfo, error) {
+	indexName := c.GetIndexName(siteID)
+	index := c.client.Index(indexName)
+
+	taskInfo, err := index.UpdateDocuments(documents, nil)
+	if err != nil {
+		return nil, fmt.Errorf("failed to update documents in index %s: %w", indexName, err)
+	}
+
+	return taskInfo, nil
+}
+
+// DeleteDocument deletes a single document from the index
+func (c *Client) DeleteDocument(siteID string, documentID string) (*meilisearch.TaskInfo, error) {
+	indexName := c.GetIndexName(siteID)
+	index := c.client.Index(indexName)
+
+	taskInfo, err := index.DeleteDocument(documentID)
+	if err != nil {
+		return nil, fmt.Errorf("failed to delete document %s from index %s: %w", documentID, indexName, err)
+	}
+
+	return taskInfo, nil
+}
+
+// DeleteDocuments deletes multiple documents from the index
+func (c *Client) DeleteDocuments(siteID string, documentIDs []string) (*meilisearch.TaskInfo, error) {
+	indexName := c.GetIndexName(siteID)
+	index := c.client.Index(indexName)
+
+	taskInfo, err := index.DeleteDocuments(documentIDs)
+	if err != nil {
+		return nil, fmt.Errorf("failed to delete documents from index %s: %w", indexName, err)
+	}
+
+	return taskInfo, nil
+}
+
+// DeleteAllDocuments deletes all documents from the index
+func (c *Client) DeleteAllDocuments(siteID string) (*meilisearch.TaskInfo, error) {
+	indexName := c.GetIndexName(siteID)
+	index := c.client.Index(indexName)
+
+	taskInfo, err := index.DeleteAllDocuments()
+	if err != nil {
+		return nil, fmt.Errorf("failed to delete all documents from index %s: %w", indexName, err)
+	}
+
+	return taskInfo, nil
+}
+
+// GetStats returns statistics about an index
+func (c *Client) GetStats(siteID string) (*meilisearch.StatsIndex, error) {
+	indexName := c.GetIndexName(siteID)
+	index := c.client.Index(indexName)
+
+	stats, err := index.GetStats()
+	if err != nil {
+		return nil, fmt.Errorf("failed to get stats for index %s: %w", indexName, err)
+	}
+
+	return stats, nil
+}