p2p_document_sync.go

// Copyright 2025 Democratized Data Foundation
//
// Use of this software is governed by the Business Source License
// included in the file licenses/BSL.txt.
//
// As of the Change Date specified in that file, in accordance with
// the Business Source License, use of this software will be governed
// by the Apache License, Version 2.0, included in the file
// licenses/APL.txt.

package cli

import (
	"context"

	"github.com/spf13/cobra"

	"github.com/sourcenetwork/defradb/client/options"
	iIdentity "github.com/sourcenetwork/defradb/internal/identity"
)

func MakeP2PDocumentSyncCommand(ctx context.Context) *cobra.Command {
	var cmd = &cobra.Command{
		Use:   "sync <collection-name> <docID...>",
		Short: "Synchronize specific documents from the network",
		Long: `Synchronize specific documents from the network.

This command allows you to sync documents from a specific collection across the network.
It doesn't automatically subscribe to the collection or the documents.`,
		Args: cobra.MinimumNArgs(2),
		RunE: func(cmd *cobra.Command, args []string) error {
			collectionName := args[0]
			docIDs := args[1:]

			ctx := cmd.Context()
			if timeout, _ := cmd.Flags().GetDuration("timeout"); timeout > 0 {
				var cancel context.CancelFunc
				ctx, cancel = context.WithTimeout(ctx, timeout)
				defer cancel()
			}

			cliClient := mustGetContextCLIClient(cmd)
			opt := options.WithIdentity(options.SyncDocuments(), iIdentity.FromContext(cmd.Context()))
			if blockSyncTimeout, _ := cmd.Flags().GetDuration("block-sync-timeout"); blockSyncTimeout > 0 {
				opt = opt.SetBlockSyncTimeout(blockSyncTimeout)
			}
			return cliClient.SyncDocuments(ctx, collectionName, docIDs, opt)
		},
	}

	EmbedCLIExample(ctx, cmd, "sync single document",
		`defradb client p2p document sync Users bae123`)

	EmbedCLIExample(ctx, cmd, "sync multiple documents",
		`defradb client p2p document sync Users bae123 bae456`)

	cmd.Flags().Duration("timeout", 0, "Timeout for the whole sync operation")
	cmd.Flags().Duration("block-sync-timeout", 0,
		"Per-block fetch timeout for this sync, overriding the node default (e.g. 30s)")
	return cmd
}