aboutsummaryrefslogtreecommitdiff
path: root/back-end/src/Model/SimpleBookmarkContext.cs
blob: e80153806c57cd9d900636e0bfa1588dcf06200f (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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
// Copyright (C) 2024 Vaughn Nugent
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

using Microsoft.EntityFrameworkCore;

using SimpleBookmark.Model.WebHistory;

using VNLib.Plugins.Extensions.Data;
using VNLib.Plugins.Extensions.Loading.Sql;


namespace SimpleBookmark.Model
{

    internal sealed class SimpleBookmarkContext : DBContextBase, IDbTableDefinition
    {

        public DbSet<BookmarkEntry> Bookmarks { get; set; }

        public DbSet<WebHistoryEntry> WebHistory { get; set; }

        public DbSet<WebExtensionAuth> AuthorizedExtensions { get; set; }

        public SimpleBookmarkContext(DbContextOptions options) : base(options)
        { }

        public SimpleBookmarkContext() : base()
        { }

        public void OnDatabaseCreating(IDbContextBuilder builder, object? userState)
        {
            /*
             * Define the coloumn mappings for the BookmarkEntry table
             */
            builder.DefineTable<BookmarkEntry>(nameof(Bookmarks), table =>
            {
                table.WithColumn(p => p.Id).AllowNull(false);
                table.WithColumn(p => p.Created);
                table.WithColumn(p => p.LastModified);
                table.WithColumn(p => p.UserId).AllowNull(false);
                table.WithColumn(p => p.Name);
                table.WithColumn(p => p.Version);
                table.WithColumn(p => p.Url);
                table.WithColumn(p => p.Description);
                table.WithColumn(p => p.Tags);
            });

            /*
             * Define the coloumn mappings for the WebHistoryEntry table                       
             */
            builder.DefineTable<WebHistoryEntry>(nameof(WebHistory), table =>
            {
                table.WithColumn(p => p.Id).AllowNull(false);
                table.WithColumn(p => p.Created);
                table.WithColumn(p => p.LastModified);
                table.WithColumn(p => p.UserId).AllowNull(false);
                table.WithColumn(p => p.Title);
                table.WithColumn(p => p.Url).AllowNull(false);
            });

            /*
             * Define the coloumn mappings for the authorized extensions table
             */
            builder.DefineTable<WebExtensionAuth>(nameof(AuthorizedExtensions), table =>
            {
                table.WithColumn(p => p.Id).AllowNull(false);
                table.WithColumn(p => p.Created);
                table.WithColumn(p => p.LastModified);
                table.WithColumn(p => p.UserId).AllowNull(false);
                table.WithColumn(p => p.ExtensionId).AllowNull(false);
                table.WithColumn(p => p.PublicKey).AllowNull(false);
                table.WithColumn(p => p.AdditionalData);
            });
        }

    }
}