Services/SearchService.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
40
41
42
43
44
45
46

using Microsoft.EntityFrameworkCore;

namespace HyperBooru.Services;

public interface ISearchService {
    public Media[] Search(string query);
}

public class SearchService : ISearchService {
    private IDbContextFactory<HBContext> dbFactory;

    public SearchService(IDbContextFactory<HBContext> dbFactory) =>
        this.dbFactory = dbFactory;

    public Media[] Search(string query) {
        var db = dbFactory.CreateDbContext();

        query = query.ToLower();

        var matchedTag = db.TagDefinitions
            .FirstOrDefault(td => td.Name.ToLower() == query)?.Guid;

        Guid[] guids;

        if(matchedTag is not null) {
            guids = new[] { (Guid) matchedTag };
        } else {
            guids = db.TagDefinitions
                .Where(td => td.Name.ToLower().Contains(query))
                .Select(td => td.Guid)
                .ToArray();
        }

        return db.Media
            .Include(m => m.Tags)
            .ThenInclude(m => m.TagDefinition)
            .AsEnumerable()
            .Where(m => m.Tags.IntersectBy(guids, t => t.TagDefinition.Guid).Any())
            .Concat(db.Media
                .Where(m =>
                    (m.ShortDescription != null && m.ShortDescription.ToLower().Contains(query)) ||
                    (m.LongDescription != null && m.LongDescription.ToLower().Contains(query))))
            .DistinctBy(m => m.Guid)
            .ToArray();
    }
}