Spotify.NET/SpotifyAPI.Web.Auth/EmbedIOAuthServer.cs

using System.Reflection;
using System.Threading;
using System.Web;
using System.Globalization;
using System.Text;
using System;
using System.Threading.Tasks;
using EmbedIO;
using EmbedIO.Actions;

namespace SpotifyAPI.Web.Auth
{
  public class EmbedIOAuthServer : IAuthServer
  {
    public event Func<object, AuthorizationCodeResponse, Task> AuthorizationCodeReceived;
    public event Func<object, ImplictGrantResponse, Task> ImplictGrantReceived;

    private const string CallbackPath = "/";
    private const string AssetsResourcePath = "SpotifyAPI.Web.Auth.Resources.auth_assets";
    private const string DefaultResourcePath = "SpotifyAPI.Web.Auth.Resources.default_site";

    private CancellationTokenSource _cancelTokenSource;
    private readonly WebServer _webServer;

    public EmbedIOAuthServer(Uri baseUri, int port)
      : this(baseUri, port, Assembly.GetExecutingAssembly(), DefaultResourcePath) { }

    public EmbedIOAuthServer(Uri baseUri, int port, Assembly resourceAssembly, string resourcePath)
    {
      Ensure.ArgumentNotNull(baseUri, nameof(baseUri));

      BaseUri = baseUri;
      Port = port;

      _webServer = new WebServer(port)
        .WithModule(new ActionModule("/", HttpVerbs.Post, (ctx) =>
        {
          var query = ctx.Request.QueryString;
          if (query["error"] != null)
          {
            throw new AuthException(query["error"], query["state"]);
          }

          var requestType = query.Get("request_type");
          if (requestType == "token")
          {
            ImplictGrantReceived?.Invoke(this, new ImplictGrantResponse(
              query["access_token"], query["token_type"], int.Parse(query["expires_in"])
            )
            {
              State = query["state"]
            });
          }
          if (requestType == "code")
          {
            AuthorizationCodeReceived?.Invoke(this, new AuthorizationCodeResponse(query["code"])
            {
              State = query["state"]
            });
          }

          return ctx.SendStringAsync("OK", "text/plain", Encoding.UTF8);
        }))
        .WithEmbeddedResources("/auth_assets", Assembly.GetExecutingAssembly(), AssetsResourcePath)
        .WithEmbeddedResources("/", resourceAssembly, resourcePath);
    }

    public Uri BaseUri { get; }
    public Uri RedirectUri { get => new Uri(BaseUri, CallbackPath); }
    public int Port { get; }

    public Task Start()
    {
      _cancelTokenSource = new CancellationTokenSource();
      _webServer.Start(_cancelTokenSource.Token);
      return Task.CompletedTask;
    }

    public Task Stop()
    {
      _cancelTokenSource.Cancel();
      return Task.CompletedTask;
    }

    public Uri BuildLoginUri(LoginRequest request)
    {
      Ensure.ArgumentNotNull(request, nameof(request));

      var callbackUri = new Uri(BaseUri, CallbackPath);

      StringBuilder builder = new StringBuilder(SpotifyUrls.Authorize.ToString());
      builder.Append($"?client_id={request.ClientId}");
      builder.Append($"&response_type={request.ResponseTypeParam.ToString().ToLower()}");
      builder.Append($"&redirect_uri={HttpUtility.UrlEncode(callbackUri.ToString())}");
      if (!string.IsNullOrEmpty(request.State))
      {
        builder.Append($"&state={HttpUtility.UrlEncode(request.State)}");
      }
      if (request.Scope != null)
      {
        builder.Append($"&scope={HttpUtility.UrlEncode(string.Join(" ", request.Scope))}");
      }
      if (request.ShowDialog != null)
      {
        builder.Append($"&show_dialog={request.ShowDialog.Value}");
      }

      return new Uri(builder.ToString());
    }

    public void Dispose()
    {
      Dispose(true);
      GC.SuppressFinalize(this);
    }

    protected virtual void Dispose(bool disposing)
    {
      if (disposing)
      {
        _webServer?.Dispose();
      }
    }
  }
}