summaryrefslogtreecommitdiff
path: root/src/marlin.web/Data/ApplicationDbContext.cs
blob: 2e058356266af2e9840985bcb8d58a07137f4c6d (plain)
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
namespace marlin.web.Data;

using System.Linq.Expressions;
using System.Text.Json;
using core;
using core.models;
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Storage.ValueConversion;
using Models;

// user-based session grant
// agent id/token grant
// agent job claim session grant
// service grant
public interface IActiveGrant
{
    bool CanAccessSource(Guid sourceId);
}

public class ApplicationDbContext : DbContext
{
    public DbSet<VersionControlSource> Sources => this.Set<VersionControlSource>();
    public DbSet<VersionControlledBranch> SourceBranches => this.Set<VersionControlledBranch>();
    public DbSet<VersionControlledCommit> SourceCommits => this.Set<VersionControlledCommit>();
    public DbSet<VersionControlledBlob> SourceBlobs => this.Set<VersionControlledBlob>();
    public DbSet<VersionControlSecret> Secrets => this.Set<VersionControlSecret>();
    public DbSet<Job> Jobs => this.Set<Job>();
    public DbSet<Agent> Agents => this.Set<Agent>();
    
    public ApplicationDbContext(DbContextOptions<ApplicationDbContext> options)
        : base(options) { }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder
            .Entity<Agent>()
            .Property(p => p.Capabilities)
            .HasConversion(
                v => JsonSerializer.Serialize(v, (JsonSerializerOptions?)null),
                v => JsonSerializer.Deserialize<CapabilitySet>(v, (JsonSerializerOptions?)null)!);

        modelBuilder
            .Entity<VersionControlledCommit>()
            .HasAlternateKey(x => new { x.SourceId, x.Hash });

        modelBuilder
            .Entity<VersionControlSecret>()
            .Property(p => p.Content)
            .HasConversion(
                v => JsonSerializer.Serialize(v, (JsonSerializerOptions?)null),
                v => JsonSerializer.Deserialize<Dictionary<string, string>>(v, (JsonSerializerOptions?)null)!);
        
        base.OnModelCreating(modelBuilder);
    }
}