using System.Security.Cryptography;
using System.Text;

namespace System.Security.Encryption;

public sealed class SecureString
{
    private static byte[] optionalEntropy;
    private byte[] encryptedValue;

    public string Value {
        get => this.encryptedValue is not null ? Encoding.UTF8.GetString(ProtectedData.Unprotect(this.encryptedValue, optionalEntropy, DataProtectionScope.CurrentUser)) : string.Empty;
        private set => this.encryptedValue = value is not null
            ? ProtectedData.Protect(Encoding.UTF8.GetBytes(value), optionalEntropy, DataProtectionScope.CurrentUser)
            : null;
    }

    public SecureString(string value)
    {
        this.Value = value;
    }

    public override bool Equals(object obj)
    {
        if (obj is string)
        {
            return this == (obj as string);
        }
        else if (obj is SecureString)
        {
            return this == (obj as SecureString);
        }
        else
        {
            return base.Equals(obj);
        }
    }
    public override int GetHashCode()
    {
        return this.Value.GetHashCode();
    }
    public override string ToString()
    {
        return this.Value;
    }

    public static readonly SecureString Empty = new(null);
    public static implicit operator string(SecureString ss) => ss is null ? string.Empty : ss.Value;
    public static implicit operator SecureString(string s) => new(s);
    public static SecureString operator +(SecureString ss1, SecureString ss2)
    {
        if (ss1 is null)
        {
            throw new ArgumentNullException(nameof(ss1));
        }

        if (ss2 is null)
        {
            throw new ArgumentNullException(nameof(ss2));
        }

        return new SecureString(ss1.Value + ss2.Value);
    }
    public static SecureString operator +(SecureString ss1, string s2)
    {
        if (ss1 is null)
        {
            throw new ArgumentNullException(nameof(ss1));
        }

        return new SecureString(ss1.Value + s2);
    }
    public static SecureString operator +(SecureString ss1, char c)
    {
        if (ss1 is null)
        {
            throw new ArgumentNullException(nameof(ss1));
        }

        return new SecureString(ss1.Value + c);
    }
    public static bool operator ==(SecureString ss1, SecureString ss2)
    {
        if (ss1 is null && ss2 is null)
        {
            return true;
        }

        return ss1?.Value == ss2?.Value;
    }
    public static bool operator !=(SecureString ss1, SecureString ss2)
    {
        return !(ss1 == ss2);
    }
    public static bool operator ==(SecureString ss1, string s2)
    {
        return ss1?.Value == s2;
    }
    public static bool operator !=(SecureString ss1, string s2)
    {
        return !(ss1?.Value == s2);
    }

    public static void AddOptionalEntropy(byte[] entropy)
    {
        optionalEntropy = entropy;
    }
}