Timeline/Entities/DatabaseContext.cs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

using Microsoft.Data.Sqlite;
using Microsoft.EntityFrameworkCore;
using System;

namespace Timeline.Entities
{
    public class DatabaseContext : DbContext
    {
        public DatabaseContext(DbContextOptions<DatabaseContext> options)
            : base(options)
        {
            if (Database.IsSqlite())
            {
                var connection = (SqliteConnection)Database.GetDbConnection();
                connection.CreateFunction("timeline_create_guid", () => Guid.NewGuid().ToString());
            }
            else
            {
                throw new InvalidOperationException(Resources.Entities.ExceptionOnlySqliteSupported);
            }
        }

        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            modelBuilder.Entity<UserEntity>().Property(e => e.Version).HasDefaultValue(0);
            modelBuilder.Entity<UserEntity>().HasIndex(e => e.Username).IsUnique();
            modelBuilder.Entity<DataEntity>().HasIndex(e => e.Tag).IsUnique();
            modelBuilder.Entity<TimelineEntity>().Property(e => e.UniqueId).HasDefaultValueSql("timeline_create_guid()");
        }

        public DbSet<UserEntity> Users { get; set; } = default!;
        public DbSet<UserAvatarEntity> UserAvatars { get; set; } = default!;
        public DbSet<TimelineEntity> Timelines { get; set; } = default!;
        public DbSet<TimelinePostEntity> TimelinePosts { get; set; } = default!;
        public DbSet<TimelineMemberEntity> TimelineMembers { get; set; } = default!;
        public DbSet<JwtTokenEntity> JwtToken { get; set; } = default!;
        public DbSet<DataEntity> Data { get; set; } = default!;
    }
}