openiddict-core/src/OpenIddict.EntityFrameworkCore/Configurations/OpenIddictTokenConfiguration.cs

/*
 * Licensed under the Apache License, Version 2.0 (http://www.apache.org/licenses/LICENSE-2.0)
 * See https://github.com/openiddict/openiddict-core for more information concerning
 * the license and the contributors participating to this project.
 */

using System;
using System.ComponentModel;
using JetBrains.Annotations;
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Metadata.Builders;
using OpenIddict.EntityFrameworkCore.Models;

namespace OpenIddict.EntityFrameworkCore
{
    /// <summary>
    /// Defines a relational mapping for the Token entity.
    /// </summary>
    /// <typeparam name="TToken">The type of the Token entity.</typeparam>
    /// <typeparam name="TApplication">The type of the Application entity.</typeparam>
    /// <typeparam name="TAuthorization">The type of the Authorization entity.</typeparam>
    /// <typeparam name="TKey">The type of the Key entity.</typeparam>
    [EditorBrowsable(EditorBrowsableState.Never)]
    public class OpenIddictTokenConfiguration<TToken, TApplication, TAuthorization, TKey> : IEntityTypeConfiguration<TToken>
        where TToken : OpenIddictToken<TKey, TApplication, TAuthorization>
        where TApplication : OpenIddictApplication<TKey, TAuthorization, TToken>
        where TAuthorization : OpenIddictAuthorization<TKey, TApplication, TToken>
        where TKey : IEquatable<TKey>
    {
        public void Configure([NotNull] EntityTypeBuilder<TToken> builder)
        {
            if (builder == null)
            {
                throw new ArgumentNullException(nameof(builder));
            }

            // Warning: optional foreign keys MUST NOT be added as CLR properties because
            // Entity Framework would throw an exception due to the TKey generic parameter
            // being non-nullable when using value types like short, int, long or Guid.


            // If primary/foreign keys are strings, limit their length to ensure
            // they can be safely used in indexes, specially when the underlying
            // provider is known to not restrict the default length (e.g MySQL).
            if (typeof(TKey) == typeof(string))
            {
                builder.Property(nameof(OpenIddictToken.Application) + nameof(OpenIddictApplication.Id))
                       .HasMaxLength(50);

                builder.Property(nameof(OpenIddictToken.Authorization) + nameof(OpenIddictApplication.Id))
                       .HasMaxLength(50);

                builder.Property(token => token.Id)
                       .HasMaxLength(50);
            }

            builder.HasKey(token => token.Id);

            builder.HasIndex(token => token.ReferenceId)
                   .IsUnique();

            builder.HasIndex(
                nameof(OpenIddictToken.Application) + nameof(OpenIddictApplication.Id),
                nameof(OpenIddictToken.Status),
                nameof(OpenIddictToken.Subject),
                nameof(OpenIddictToken.Type));

            builder.Property(token => token.ConcurrencyToken)
                   .HasMaxLength(50)
                   .IsConcurrencyToken();

            builder.Property(token => token.ReferenceId)
                   .HasMaxLength(100);

            builder.Property(token => token.Status)
                   .HasMaxLength(25)
                   .IsRequired();

            builder.Property(token => token.Subject)
                   .HasMaxLength(450)
                   .IsRequired();

            builder.Property(token => token.Type)
                   .HasMaxLength(25)
                   .IsRequired();

            builder.ToTable("OpenIddictTokens");
        }
    }
}