dragon limits

OneShelf.OneDog/OneShelf.OneDog.Processor/Services/PipelineHandlers/AiDialogHandler.cs

@@ -78,6 +78,10 @@ protected override (string? additionalBillingInfo, int? domainId) GetDialogConfi
         return imagesUnavailableUntil;
     }
 
+    protected override async Task<DateTime?> GetChatUnavailableUntil() => null;
+
+    protected override string UnavailableUntilTemplate => throw new InvalidOperationException();
+
     protected override async Task<(string? system, string? version, float? frequencyPenalty, float? presencePenalty, int? imagesVersion)> GetAiParameters()
     {
         var system = _dogContext.Domain.SystemMessage;

OneShelf.OneDragon/OneShelf.OneDragon.Database/DragonDatabase.cs

@@ -22,6 +22,8 @@ public DragonDatabase(DbContextOptions<DragonDatabase> options) : base(options)
 
     public required DbSet<AiParameters> AiParameters { get; set; }
 
+    public required DbSet<Limit> Limits { get; set; }
+
     protected override void OnModelCreating(ModelBuilder modelBuilder)
     {
         base.OnModelCreating(modelBuilder);

OneShelf.OneDragon/OneShelf.OneDragon.Database/Migrations/20240819203114_AddLimits.cs

@@ -0,0 +1,48 @@
+using Microsoft.EntityFrameworkCore.Migrations;
+
+#nullable disable
+
+namespace OneShelf.OneDragon.Database.Migrations
+{
+    /// <inheritdoc />
+    public partial class AddLimits : Migration
+    {
+        /// <inheritdoc />
+        protected override void Up(MigrationBuilder migrationBuilder)
+        {
+            migrationBuilder.AddColumn<bool>(
+                name: "UseLimits",
+                table: "Users",
+                type: "bit",
+                nullable: false,
+                defaultValue: false);
+
+            migrationBuilder.CreateTable(
+                name: "Limits",
+                columns: table => new
+                {
+                    Id = table.Column<int>(type: "int", nullable: false)
+                        .Annotation("SqlServer:Identity", "1, 1"),
+                    Images = table.Column<int>(type: "int", nullable: true),
+                    Texts = table.Column<int>(type: "int", nullable: true),
+                    Window = table.Column<long>(type: "bigint", nullable: false),
+                    IsEnabled = table.Column<bool>(type: "bit", nullable: false)
+                },
+                constraints: table =>
+                {
+                    table.PrimaryKey("PK_Limits", x => x.Id);
+                });
+        }
+
+        /// <inheritdoc />
+        protected override void Down(MigrationBuilder migrationBuilder)
+        {
+            migrationBuilder.DropTable(
+                name: "Limits");
+
+            migrationBuilder.DropColumn(
+                name: "UseLimits",
+                table: "Users");
+        }
+    }
+}

OneShelf.OneDragon/OneShelf.OneDragon.Database/Migrations/DragonDatabaseModelSnapshot.cs

@@ -128,6 +128,31 @@ protected override void BuildModel(ModelBuilder modelBuilder)
                     b.ToTable("Interactions");
                 });
 
+            modelBuilder.Entity("OneShelf.OneDragon.Database.Model.Limit", b =>
+                {
+                    b.Property<int>("Id")
+                        .ValueGeneratedOnAdd()
+                        .HasColumnType("int");
+
+                    SqlServerPropertyBuilderExtensions.UseIdentityColumn(b.Property<int>("Id"));
+
+                    b.Property<int?>("Images")
+                        .HasColumnType("int");
+
+                    b.Property<bool>("IsEnabled")
+                        .HasColumnType("bit");
+
+                    b.Property<int?>("Texts")
+                        .HasColumnType("int");
+
+                    b.Property<long>("Window")
+                        .HasColumnType("bigint");
+
+                    b.HasKey("Id");
+
+                    b.ToTable("Limits");
+                });
+
             modelBuilder.Entity("OneShelf.OneDragon.Database.Model.Update", b =>
                 {
                     b.Property<int>("Id")
@@ -166,6 +191,9 @@ protected override void BuildModel(ModelBuilder modelBuilder)
                         .IsRequired()
                         .HasColumnType("nvarchar(max)");
 
+                    b.Property<bool>("UseLimits")
+                        .HasColumnType("bit");
+
                     b.Property<string>("UserName")
                         .HasColumnType("nvarchar(max)");
 

OneShelf.OneDragon/OneShelf.OneDragon.Database/Model/Limit.cs

@@ -0,0 +1,17 @@
+using System.ComponentModel.DataAnnotations.Schema;
+
+namespace OneShelf.OneDragon.Database.Model;
+
+public class Limit
+{
+    public int Id { get; set; }
+
+    public int? Images { get; set; }
+
+    public int? Texts { get; set; }
+
+    [Column(TypeName = "bigint")]
+    public required TimeSpan Window { get; set; }
+
+    public bool IsEnabled { get; set; }
+}

OneShelf.OneDragon/OneShelf.OneDragon.Database/Model/User.cs

@@ -22,4 +22,6 @@ public class User
     public string? UserName { get; set; }
 
     public string? LanguageCode { get; set; }
+
+    public bool UseLimits { get; set; }
 }