RedisManager/Commands/BitCommands.cs

using CommandLine;
using StackExchange.Redis;
using System;
using System.Collections.Generic;
using System.Linq;
using RedisManager.Utils;

using RedisManager.Commands;

namespace RedisManager.Commands
{
    [Verb("bitcount", HelpText = "Count set bits in a string.")]
    public class BitCountOptions : InstanceOptions
    {
        [Value(0, MetaName = "key", Required = true, HelpText = "Key name.")]
        public string Key { get; set; }
        [Option("start", Default = 0L, HelpText = "Start byte.")]
        public long Start { get; set; }
        [Option("end", Default = -1L, HelpText = "End byte.")]
        public long End { get; set; }
    }

    [Verb("bitfield", HelpText = "Perform arbitrary bitfield integer operations on strings.")]
    public class BitFieldOptions : InstanceOptions
    {
        [Value(0, MetaName = "key", Required = true, HelpText = "Key name.")]
        public string Key { get; set; }
        [Value(1, MetaName = "operations", Min = 1, Required = true, HelpText = "Bitfield operations.")]
        public IEnumerable<string> Operations { get; set; }
    }

    [Verb("bitop", HelpText = "Perform bitwise operations between strings.")]
    public class BitOpOptions : InstanceOptions
    {
        [Value(0, MetaName = "operation", Required = true, HelpText = "Operation (AND, OR, XOR, NOT).")]
        public string Operation { get; set; }
        [Value(1, MetaName = "dest-key", Required = true, HelpText = "Destination key.")]
        public string DestKey { get; set; }
        [Value(2, MetaName = "source-keys", Min = 1, Required = true, HelpText = "Source keys.")]
        public IEnumerable<string> SourceKeys { get; set; }
    }

    [Verb("bitpos", HelpText = "Find first bit set or clear in a string.")]
    public class BitPosOptions : InstanceOptions
    {
        [Value(0, MetaName = "key", Required = true, HelpText = "Key name.")]
        public string Key { get; set; }
        [Value(1, MetaName = "bit", Required = true, HelpText = "Bit value (0 or 1).")]
        public int Bit { get; set; }
        [Option("start", Default = 0L, HelpText = "Start byte.")]
        public long Start { get; set; }
        [Option("end", Default = -1L, HelpText = "End byte.")]
        public long End { get; set; }
    }

    [Verb("getbit", HelpText = "Returns the bit value at offset in the string value stored at key.")]
    public class GetBitOptions : InstanceOptions
    {
        [Value(0, MetaName = "key", Required = true, HelpText = "Key name.")]
        public string Key { get; set; }
        [Value(1, MetaName = "offset", Required = true, HelpText = "Bit offset.")]
        public long Offset { get; set; }
    }

    [Verb("setbit", HelpText = "Sets or clears the bit at offset in the string value stored at key.")]
    public class SetBitOptions : InstanceOptions
    {
        [Value(0, MetaName = "key", Required = true, HelpText = "Key name.")]
        public string Key { get; set; }
        [Value(1, MetaName = "offset", Required = true, HelpText = "Bit offset.")]
        public long Offset { get; set; }
        [Value(2, MetaName = "bit", Required = true, HelpText = "Bit value (0 or 1).")]
        public int Bit { get; set; }
    }

    public static class BitCommands
    {
        /// <summary>
        /// Executes the BITCOUNT command to count set bits in a string.
        /// </summary>
        /// <param name="opts">The BitCountOptions containing instance, key, start, and end.</param>
        /// <param name="config">The RedisManager configuration.</param>
        /// <returns>Exit code (0 for success, non-zero for failure).</returns>
        public static int RunBitCount(BitCountOptions opts, Config config)
        {
            var instance = RedisUtils.GetInstance(config, opts.Instance);
            using var redis = RedisUtils.ConnectRedis(instance);
            var db = redis.GetDatabase();

            var args = new List<object> { opts.Key };
            if (opts.Start != 0 || opts.End != -1) { args.Add(opts.Start); args.Add(opts.End); }
            var result = db.Execute("BITCOUNT", args.ToArray());
            Console.WriteLine(Output.Green($"Bit count: {result}"));
            return 0;
        }

        /// <summary>
        /// Executes the BITFIELD command to perform arbitrary bitfield integer operations on strings.
        /// </summary>
        /// <param name="opts">The BitFieldOptions containing instance, key, and operations.</param>
        /// <param name="config">The RedisManager configuration.</param>
        /// <returns>Exit code (0 for success, non-zero for failure).</returns>
        public static int RunBitField(BitFieldOptions opts, Config config)
        {
            var instance = RedisUtils.GetInstance(config, opts.Instance);
            using var redis = RedisUtils.ConnectRedis(instance);
            var db = redis.GetDatabase();

            var args = new List<object> { opts.Key };
            args.AddRange(opts.Operations);
            var result = db.Execute("BITFIELD", args.ToArray());
            RedisResult[] results;
            if (result is Array)
                results = (RedisResult[])result;
            else
                results = new RedisResult[] { result };

            Console.WriteLine(Output.Green("Bitfield results:"));
            var operationsList = opts.Operations.ToList();
            for (int i = 0; results != null && i < results.Length; i++)
            {
                var res = results[i];
                if (res != null)
                {
                    Console.WriteLine(Output.Cyan($"{operationsList[i]}: {res}"));
                }
                else
                {
                    Console.WriteLine(Output.Red($"{operationsList[i]}: null"));
                }
            }
            return 0;
        }

        /// <summary>
        /// Executes the BITOP command to perform bitwise operations between strings.
        /// </summary>
        /// <param name="opts">The BitOpOptions containing instance, operation, destination key, and source keys.</param>
        /// <param name="config">The RedisManager configuration.</param>
        /// <returns>Exit code (0 for success, non-zero for failure).</returns>
        public static int RunBitOp(BitOpOptions opts, Config config)
        {
            var instance = RedisUtils.GetInstance(config, opts.Instance);
            using var redis = RedisUtils.ConnectRedis(instance);
            var db = redis.GetDatabase();

            var args = new List<object> { opts.Operation, opts.DestKey };
            args.AddRange(opts.SourceKeys);
            var result = db.Execute("BITOP", args.ToArray());
            Console.WriteLine(Output.Green($"BITOP result: {result}"));
            return 0;
        }

        /// <summary>
        /// Executes the BITPOS command to find the first bit set or clear in a string.
        /// </summary>
        /// <param name="opts">The BitPosOptions containing instance, key, bit, start, and end.</param>
        /// <param name="config">The RedisManager configuration.</param>
        /// <returns>Exit code (0 for success, non-zero for failure).</returns>
        public static int RunBitPos(BitPosOptions opts, Config config)
        {
            var instance = RedisUtils.GetInstance(config, opts.Instance);
            using var redis = RedisUtils.ConnectRedis(instance);
            var db = redis.GetDatabase();

            var args = new List<object> { opts.Key, opts.Bit };
            if (opts.Start != 0 || opts.End != -1) { args.Add(opts.Start); args.Add(opts.End); }
            var result = db.Execute("BITPOS", args.ToArray());
            Console.WriteLine(Output.Green($"First {opts.Bit} bit at position: {result}"));
            return 0;
        }

        /// <summary>
        /// Executes the GETBIT command to return the bit value at offset in the string value stored at key.
        /// </summary>
        /// <param name="opts">The GetBitOptions containing instance, key, and offset.</param>
        /// <param name="config">The RedisManager configuration.</param>
        /// <returns>Exit code (0 for success, non-zero for failure).</returns>
        public static int RunGetBit(GetBitOptions opts, Config config)
        {
            var instance = RedisUtils.GetInstance(config, opts.Instance);
            using var redis = RedisUtils.ConnectRedis(instance);
            var db = redis.GetDatabase();

            var result = db.Execute("GETBIT", new object[] { opts.Key, opts.Offset });
            Console.WriteLine(Output.Green($"Bit at offset {opts.Offset}: {result}"));
            return 0;
        }

        /// <summary>
        /// Executes the SETBIT command to set or clear the bit at offset in the string value stored at key.
        /// </summary>
        /// <param name="opts">The SetBitOptions containing instance, key, offset, and bit value.</param>
        /// <param name="config">The RedisManager configuration.</param>
        /// <returns>Exit code (0 for success, non-zero for failure).</returns>
        public static int RunSetBit(SetBitOptions opts, Config config)
        {
            var instance = RedisUtils.GetInstance(config, opts.Instance);
            using var redis = RedisUtils.ConnectRedis(instance);
            var db = redis.GetDatabase();

            var result = db.Execute("SETBIT", new object[] { opts.Key, opts.Offset, opts.Bit });
            Console.WriteLine(Output.Green($"Set bit at offset {opts.Offset} to {opts.Bit}: {result}"));
            return 0;
        }
    }
}