Iceshrimp.NET/Iceshrimp.Frontend/Core/Services/ApiClient.cs

using System.Net.Http.Headers;
using System.Net.Http.Json;
using System.Net.Mime;
using System.Text;
using System.Text.Json;
using Iceshrimp.Frontend.Core.Miscellaneous;
using Iceshrimp.Shared.Configuration;
using Iceshrimp.Shared.Schemas.Web;
using Microsoft.AspNetCore.Components.Forms;

namespace Iceshrimp.Frontend.Core.Services;

internal class ApiClient(HttpClient client)
{
	private string? _token;

	public void SetToken(string token) => _token = token;

	public async Task CallAsync(HttpMethod method, string path, QueryString? query = null, object? data = null)
	{
		var res = await MakeRequestAsync(method, path, query, data);
		if (res.IsSuccessStatusCode)
			return;

		var error = await res.Content.ReadFromJsonAsync<ErrorResponse>(JsonSerialization.Options);
		if (error == null)
			throw new Exception("Deserialized API error was null");
		throw new ApiException(error);
	}

	public async Task<bool> CallNullableAsync(
		HttpMethod method, string path, QueryString? query = null, object? data = null
	)
	{
		var res = await MakeRequestAsync(method, path, query, data);
		if (res.IsSuccessStatusCode)
			return true;

		var error = await res.Content.ReadFromJsonAsync<ErrorResponse>(JsonSerialization.Options);
		if (error == null)
			throw new Exception("Deserialized API error was null");
		throw new ApiException(error);
	}

	public async Task<T> CallAsync<T>(
		HttpMethod method, string path, QueryString? query = null, object? data = null
	) where T : class
	{
		var res = await CallInternalAsync<T>(method, path, query, data);
		if (res.result != null)
			return res.result;
		throw new ApiException(res.error ?? throw new Exception("Deserialized API error was null"));
	}

	public async Task<T?> CallNullableAsync<T>(
		HttpMethod method, string path, QueryString? query = null, object? data = null
	) where T : class
	{
		var res = await CallInternalAsync<T>(method, path, query, data);
		if (res.result != null)
			return res.result;

		var err = res.error ?? throw new Exception("Deserialized API error was null");
		if (err.StatusCode == 404)
			return null;

		throw new ApiException(err);
	}

	private async Task<(T? result, ErrorResponse? error)> CallInternalAsync<T>(
		HttpMethod method, string path, QueryString? query, object? data
	) where T : class
	{
		var res = await MakeRequestAsync(method, path, query, data);

		if (res.IsSuccessStatusCode)
		{
			var deserialized = await res.Content.ReadFromJsonAsync<T>(JsonSerialization.Options);
			if (deserialized == null)
				throw new Exception("Deserialized API response was null");
			return (deserialized, null);
		}

		var error = await res.Content.ReadFromJsonAsync<ErrorResponse>(JsonSerialization.Options);
		if (error == null)
			throw new Exception("Deserialized API error was null");
		return (null, error);
	}

	private async Task<HttpResponseMessage> MakeRequestAsync(
		HttpMethod method, string path, QueryString? query, object? data
	)
	{
		var request = new HttpRequestMessage(method, "/api/iceshrimp/" + path.TrimStart('/') + query);
		request.Headers.Accept.ParseAdd(MediaTypeNames.Application.Json);
		if (_token != null) request.Headers.Authorization = new AuthenticationHeaderValue("Bearer", _token);

		if (data is IBrowserFile file)
		{
			request.Content = new MultipartFormDataContent
			{
				new StreamContent(file.OpenReadStream(1024 * 1024 * 50))
				{
					Headers =
					{
						ContentType = new MediaTypeHeaderValue(file.ContentType),
						ContentDisposition = new ContentDispositionHeaderValue("form-data")
						{
							Name     = "file",
							FileName = file.Name,
							Size     = file.Size
						}
					}
				}
			};
		}
		else if (data is not null)
		{
			request.Content = new StringContent(JsonSerializer.Serialize(data, JsonSerialization.Options),
			                                    Encoding.UTF8, MediaTypeNames.Application.Json);
		}

		return await client.SendAsync(request);
	}
}