bitwarden-server/src/Core/Tokens/Token.cs

using Microsoft.AspNetCore.DataProtection;

namespace Bit.Core.Tokens;

public class Token
{
    private readonly string _token;

    public Token(string token)
    {
        _token = token;
    }

    public Token WithPrefix(string prefix)
    {
        return new Token($"{prefix}{_token}");
    }

    public Token RemovePrefix(string expectedPrefix)
    {
        if (!_token.StartsWith(expectedPrefix))
        {
            throw new BadTokenException($"Expected prefix, {expectedPrefix}, was not present.");
        }

        return new Token(_token[expectedPrefix.Length..]);
    }

    public Token ProtectWith(IDataProtector dataProtector) =>
        new(dataProtector.Protect(ToString()));

    public Token UnprotectWith(IDataProtector dataProtector) =>
        new(dataProtector.Unprotect(ToString()));

    public override string ToString() => _token;
}