Newsbot.Collector/Newsbot.Collector.Api/Controllers/SourcesController.cs

using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Options;
using Newsbot.Collector.Database.Repositories;
using Newsbot.Collector.Domain.Consts;
using Newsbot.Collector.Domain.Dto;
using Newsbot.Collector.Domain.Interfaces;
using Newsbot.Collector.Domain.Models;
using Newsbot.Collector.Services.HtmlParser;

namespace Newsbot.Collector.Api.Controllers;

[ApiController]
[Route("api/sources")]
public class SourcesController : ControllerBase
{
    private readonly IIconsRepository _icons;
    private readonly ILogger<SourcesController> _logger;

    //private readonly ConnectionStrings _settings;
    private readonly ISourcesRepository _sources;

    public SourcesController(ILogger<SourcesController> logger, IOptions<ConnectionStrings> settings)
    {
        _logger = logger;
        //_settings = settings.Value;
        _sources = new SourcesTable(settings.Value.Database);
        _icons = new IconsTable(settings.Value.Database);
    }

    [HttpGet(Name = "GetSources")]
    public IEnumerable<SourceDto> Get(int page)
    {
        var res = new List<SourceDto>();
        var temp = _sources.List(page, 25);
        foreach (var item in temp) res.Add(SourceDto.Convert(item));
        return res;
    }

    [HttpGet("by/type")]
    public IEnumerable<SourceDto> GetByType(string type)
    {
        var res = new List<SourceDto>();
        var temp = _sources.ListByType(type);
        foreach (var item in temp) res.Add(SourceDto.Convert(item));
        return res;
    }

    [HttpPost("new/reddit")]
    public SourceDto NewReddit(string name)
    {
        var res = _sources.GetByNameAndType(name, SourceTypes.Reddit);
        if (res.ID != Guid.Empty) return SourceDto.Convert(res);

        var uri = new Uri($"https://reddit.com/r/{name}");

        var pageReader = new HtmlPageReader(new HtmlPageReaderOptions
        {
            Url = uri.ToString()
        });
        pageReader.Parse();

        var item = _sources.New(new SourceModel
        {
            Site = SourceTypes.Reddit,
            Name = name,
            Type = SourceTypes.Reddit,
            Source = "feed",
            Enabled = true,
            Url = uri.ToString(),
            Tags = $"{SourceTypes.Reddit},{name}"
        });

        // Not all subreddits have an Icon, so we only want to add a record when it has one.
        if (pageReader.Data.Header.Image != "")
            _icons.New(new IconModel
            {
                Id = Guid.NewGuid(),
                FileName = pageReader.Data.Header.Image,
                SourceId = item.ID
            });
        return SourceDto.Convert(item);
    }

    [HttpPost("new/rss")]
    public SourceDto NewRss(string name, string url)
    {
        var res = _sources.GetByNameAndType(name, SourceTypes.Rss);
        if (res.ID != Guid.Empty) return SourceDto.Convert(res);

        var m = new SourceModel
        {
            Site = SourceTypes.Rss,
            Name = name,
            Type = SourceTypes.Rss,
            Source = "feed",
            Enabled = true,
            Url = url,
            Tags = $"{SourceTypes.Rss},{name}"
        };
        var item = _sources.New(m);
        return SourceDto.Convert(item);
    }

    [HttpPost("new/youtube")]
    public SourceDto NewYoutube(string url)
    {
        var res = _sources.GetByUrl(url);
        if (res.ID != Guid.Empty) return SourceDto.Convert(res);

        var htmlClient = new HtmlPageReader(new HtmlPageReaderOptions
        {
            Url = url
        });
        htmlClient.Parse();

        var item = _sources.New(new SourceModel
        {
            Site = SourceTypes.YouTube,
            Type = SourceTypes.YouTube,
            Name = htmlClient.Data.Header.Title,
            Source = "feed",
            Url = "feed",
            Enabled = true,
            Tags = $"{SourceTypes.YouTube},{htmlClient.Data.Header.Title}",
            YoutubeId = htmlClient.Data.Header.YoutubeChannelID ?? ""
        });

        _icons.New(new IconModel
        {
            Id = Guid.NewGuid(),
            FileName = htmlClient.Data.Header.Image,
            SourceId = item.ID
        });

        return SourceDto.Convert(item);
    }

    [HttpPost("new/twitch")]
    public SourceDto NewTwitch(string name)
    {
        var res = _sources.GetByNameAndType(name, SourceTypes.Twitch);
        if (res.ID != Guid.Empty) return SourceDto.Convert(res);

        var item = _sources.New(new SourceModel
        {
            Site = SourceTypes.Twitch,
            Type = SourceTypes.Twitch,
            Name = name,
            Url = $"https://twitch.tv/{name}",
            Source = "api",
            Enabled = true,
            Tags = $"{SourceTypes.Twitch},{name}"
        });
        return SourceDto.Convert(item);
    }

    [HttpPost("new/github")]
    public SourceDto NewGithub(string url)
    {
        if (!url.Contains("github.com")) return new SourceDto();

        var res = _sources.GetByUrl(url);
        if (res.ID != Guid.Empty) return SourceDto.Convert(res);

        var slice = url.Split('/');

        var pageReader = new HtmlPageReader(new HtmlPageReaderOptions
        {
            Url = url
        });
        pageReader.Parse();

        var item = _sources.New(new SourceModel
        {
            Site = SourceTypes.GitHub,
            Type = SourceTypes.GitHub,
            Name = $"{slice[3]}/{slice[4]}",
            Url = url,
            Source = "feed",
            Enabled = true,
            Tags = $"{SourceTypes.GitHub}, {slice[3]}, {slice[4]}"
        });

        _icons.New(new IconModel
        {
            Id = Guid.NewGuid(),
            FileName = pageReader.Data.Header.Image,
            SourceId = item.ID
        });

        return SourceDto.Convert(item);
    }

    [HttpGet("{id}")]
    public SourceDto GetById(Guid id)
    {
        var item = _sources.GetByID(id);
        return SourceDto.Convert(item);
    }

    [HttpPost("{id}/disable")]
    public void Disable(Guid id)
    {
        _sources.Disable(id);
    }

    [HttpPost("{id}/enable")]
    public void Enable(Guid id)
    {
        _sources.Enable(id);
    }
}