tools

var Flags = &flags{}

var GetFirehoseClientCmd = func(zlog *zap.Logger, tracer logging.Tracer, transformsSetter TransformsSetter) *cobra.Command {
	out := &cobra.Command{
		Use:   "firehose-client",
		Short: "Connects to a Firehose endpoint over gRPC and print block stream as JSON to terminal",
		Args:  cobra.ExactArgs(3),
		RunE:  getFirehoseClientE(zlog, tracer, transformsSetter),
	}
	out.Flags().StringP("api-token-env-var", "a", "FIREHOSE_API_TOKEN", "Look for a JWT in this environment variable to authenticate against endpoint")
	out.Flags().String("compression", "none", "The HTTP compression: use either 'none', 'gzip' or 'zstd'")
	out.Flags().String("cursor", "", "Use this cursor with the request to resume your stream at the following block pointed by the cursor")
	out.Flags().BoolP("plaintext", "p", false, "Use plaintext connection to Firehose")
	out.Flags().BoolP("insecure", "k", false, "Use SSL connection to Firehose but skip SSL certificate validation")
	out.Flags().Bool("print-cursor-only", false, "Skip block decoding, only print the step cursor (useful for performance testing)")
	out.Flags().Bool("final-blocks-only", false, "Only ask for final blocks")

	return out
}

var GetFirehosePrometheusExporterCmd = func(zlog *zap.Logger, tracer logging.Tracer, transformsSetter TransformsSetter) *cobra.Command {
	out := &cobra.Command{
		Use:   "firehose-prometheus-exporter <endpoint:port>",
		Short: "stream blocks near the chain HEAD and report to prometheus",
		Args:  cobra.ExactArgs(1),
		RunE:  runPrometheusExporterE(zlog, tracer, transformsSetter),
	}
	out.Flags().StringP("api-token-env-var", "a", "FIREHOSE_API_TOKEN", "Look for a JWT in this environment variable to authenticate against endpoint")

	out.Flags().BoolP("plaintext", "p", false, "Use plaintext connection to firehose")
	out.Flags().BoolP("insecure", "k", false, "Skip SSL certificate validation when connecting to firehose")

	return out
}

var GetFirehoseSingleBlockClientCmd = func(zlog *zap.Logger, tracer logging.Tracer) *cobra.Command {
	out := &cobra.Command{
		Use:     "firehose-single-block-client {endpoint} {block_num|block_num:block_id|cursor}",
		Short:   "fetch a single block from firehose and print as JSON",
		Args:    cobra.ExactArgs(2),
		RunE:    getFirehoseSingleBlockClientE(zlog, tracer),
		Example: "firehose-single-block-client --compression=gzip my.firehose.endpoint:443 2344:0x32d8e8d98a798da98d6as9d69899as86s9898d8ss8d87",
	}
	out.Flags().StringP("api-token-env-var", "a", "FIREHOSE_API_TOKEN", "Look for a JWT in this environment variable to authenticate against endpoint")
	out.Flags().String("compression", "none", "http compression: use either 'none', 'gzip' or 'zstd'")
	out.Flags().BoolP("plaintext", "p", false, "Use plaintext connection to firehose")
	out.Flags().BoolP("insecure", "k", false, "Skip SSL certificate validation when connecting to firehose")
	return out
}

var GetMergedBlocksUpgrader = func(zlog *zap.Logger, tracer logging.Tracer, tweakFunc func(cmd *cobra.Command, block *bstream.Block) (*bstream.Block, error)) *cobra.Command {
	out := &cobra.Command{
		Use:   "upgrade-merged-blocks <source> <destination> <start> <stop>",
		Short: "from a merged-blocks source, rewrite blocks to a new merged-blocks destination, while applying all possible upgrades",
		Args:  cobra.ExactArgs(4),
		RunE:  getMergedBlockUpgrader(zlog, tracer, tweakFunc),
	}

	return out
}