Newsbot.Collector/Newsbot.Collector.Database/Repositories/SubscriptionsTable.cs

using System.Data;
using Dapper;
using Microsoft.Extensions.Configuration;
using Newsbot.Collector.Domain.Interfaces;
using Newsbot.Collector.Domain.Models;
using Npgsql;

namespace Newsbot.Collector.Database.Repositories;

public class SubscriptionsTable : ISubscriptionRepository
{
    private readonly string _connectionString;

    public SubscriptionsTable(string connectionString)
    {
        _connectionString = connectionString;
    }

    public SubscriptionsTable(IConfiguration configuration)
    {
        var connstr = configuration.GetConnectionString("database");
        if (connstr is null) connstr = "";
        _connectionString = connstr;
    }

    public SubscriptionModel New(SubscriptionModel model)
    {
        model.Id = Guid.NewGuid();
        using var conn = OpenConnection(_connectionString);
        var query =
            "Insert Into subscriptions (ID, DiscordWebHookId, SourceId, codeAllowCommits, codeAllowReleases) Values (@id, @webhookId, @sourceId, @codeAllowCommits, @codeAllowReleases);";
        try
        {
            conn.Execute(query, new
            {
                id = model.Id,
                webhookId = model.DiscordWebHookId,
                sourceId = model.SourceId,
                codeAllowCommits = model.CodeAllowCommits,
                codeAllowReleases = model.CodeAllowReleases
            });
            return model;
        }
        catch
        {
            return new SubscriptionModel();
        }
    }

    public List<SubscriptionModel> List(int page = 0, int count = 25)
    {
        using var conn = OpenConnection(_connectionString);
        var query = @"Select * From subscriptions
        Offset @page Fetch Next @count Rows Only;";
        return conn.Query<SubscriptionModel>(query, new
        {
            page = page * count, count
        }).ToList();
    }

    public List<SubscriptionModel> ListBySourceID(Guid id, int page = 0, int count = 25)
    {
        using var conn = OpenConnection(_connectionString);
        var query = @"Select * From subscriptions 
        Where sourceid = @sourceid
        Offset @page Fetch Next @count Rows Only";
        return conn.Query<SubscriptionModel>(query, new
        {
            page = page * count,
            count,
            sourceid = id
        }).ToList();
    }

    public List<SubscriptionModel> ListByWebhook(Guid id, int page = 0, int count = 25)
    {
        using var conn = OpenConnection(_connectionString);
        var query = @"Select * From subscriptions 
        Where discordwebhookid = @webhookid
        Offset @page Fetch Next @count Rows Only";
        return conn.Query<SubscriptionModel>(query, new
        {
            page = page * count,
            count,
            webhookid = id
        }).ToList();
    }

    public SubscriptionModel GetById(Guid id)
    {
        using var conn = OpenConnection(_connectionString);
        var query = @"Select * From subscriptions Where id = @id;";
        var res = conn.Query<SubscriptionModel>(query, new
        {
            id
        });
        if (res.Count() == 0) return new SubscriptionModel();
        return res.First();
    }

    public SubscriptionModel GetByWebhookAndSource(Guid webhookId, Guid sourceId)
    {
        using var conn = OpenConnection(_connectionString);
        var query = @"Select * From subscriptions 
        Where discordwebhookid = @webhookid 
        and sourceid = @sourceid;";
        var res = conn.Query<SubscriptionModel>(query, new
        {
            webhookid = webhookId,
            sourceid = sourceId
        });
        if (res.Count() == 0) return new SubscriptionModel();
        return res.First();
    }

    public void Delete(Guid id)
    {
        using var conn = OpenConnection(_connectionString);
        var query = "Delete From subscriptions Where id = @id;";
        conn.Execute(query, new
        {
            id
        });
    }

    private IDbConnection OpenConnection(string connectionString)
    {
        var conn = new NpgsqlConnection(_connectionString);
        conn.Open();
        return conn;
    }
}