SocialPub/PrivaPub/Api/Mastodon/Mappers/AccountSearch.cs

using PrivaPub.Api.Mastodon.Entities;
using PrivaPub.Federation.Actors;
using PrivaPub.StaticServices;

namespace PrivaPub.Api.Mastodon.Mappers
{
	public class AccountSearch
	{
		readonly MastodonMapper _mapper;
		readonly DbEntities _dbEntities;
		readonly ILocalActorService _localActors;
		readonly IRemoteActorService _remoteActors;

		public AccountSearch(MastodonMapper mapper, DbEntities dbEntities, ILocalActorService localActors, IRemoteActorService remoteActors)
		{
			_mapper = mapper;
			_dbEntities = dbEntities;
			_localActors = localActors;
			_remoteActors = remoteActors;
		}

		public async Task<List<Account>> Find(string q, bool resolve, int limit, CancellationToken token) => await Find(q, resolve, limit, 0, token);

		// gone accounts (remote ones that deleted themselves, personas of banned or deleted roots) are never found
		public async Task<List<Account>> Find(string q, bool resolve, int limit, int offset, CancellationToken token)
		{
			q = q?.Trim().TrimStart('@');
			if (string.IsNullOrEmpty(q))
				return new List<Account>();
			var results = new List<Account>();
			var parts = q.Split('@');
			var localDomain = new Uri(_localActors.BaseAddress).Authority;
			if (parts.Length == 2 && !parts[1].Equals(localDomain, StringComparison.OrdinalIgnoreCase) && !string.IsNullOrEmpty(parts[0]))
			{
				var userName = parts[0];
				var domain = parts[1].ToLowerInvariant();
				var known = await _dbEntities.ForeignAvatars.Match(f => f.UserName == userName && f.Domain == domain && !f.DeletionAt.HasValue).ExecuteFirstAsync(token);
				if (known == default && resolve)
				{
					var actorUri = await _remoteActors.ResolveHandle(q, token);
					known = actorUri == default ? default : await _remoteActors.GetActor(actorUri, refresh: false, token);
				}
				if (known is { DeletionAt: null } && offset == 0)
					results.Add(_mapper.Foreign(known));
				return results;
			}

			var prefix = parts[0].ToLowerInvariant();
			var barred = await Domain.Privacy.Counted.BarredPersonas(token);
			var locals = await _dbEntities.Avatars
				.Match(a => a.UserName.StartsWith(prefix) && !a.DeletionAt.HasValue && a.Settings.IsDiscoverable != false && !barred.Contains(a.ID))
				.Sort(a => a.UserName, MongoDB.Entities.Order.Ascending)
				.Limit(offset + limit).ExecuteAsync(token);
			foreach (var avatar in locals.Skip(offset))
				results.Add(await _mapper.Local(_localActors.FromAvatar(avatar), false, token));
			var skipRemote = Math.Max(0, offset - locals.Count);
			if (results.Count < limit)
				foreach (var foreign in await _dbEntities.ForeignAvatars.Match(f => f.UserName.StartsWith(parts[0]) && f.IsDiscoverable && !f.DeletionAt.HasValue)
					.Sort(f => f.UserName, MongoDB.Entities.Order.Ascending)
					.Skip(skipRemote).Limit(limit - results.Count).ExecuteAsync(token))
					results.Add(_mapper.Foreign(foreign));
			return results;
		}
	}
}