nub-lang/compiler/NubType.cs

namespace Compiler;

public abstract class NubType : IEquatable<NubType>
{
    public abstract override string ToString();

    public abstract bool Equals(NubType? other);

    public override bool Equals(object? obj)
    {
        if (obj is NubType otherNubType)
        {
            return Equals(otherNubType);
        }

        return false;
    }

    public abstract override int GetHashCode();

    public static bool operator ==(NubType? left, NubType? right) => Equals(left, right);
    public static bool operator !=(NubType? left, NubType? right) => !Equals(left, right);
}

public sealed class NubTypeVoid : NubType
{
    public override string ToString() => "void";

    public override bool Equals(NubType? other) => other is NubTypeVoid;
    public override int GetHashCode() => HashCode.Combine(typeof(NubTypeVoid));
}

public sealed class NubTypeUInt(int width) : NubType
{
    public readonly int Width = width;

    public override string ToString() => $"u{Width}";

    public override bool Equals(NubType? other) => other is NubTypeUInt otherUInt && Width == otherUInt.Width;
    public override int GetHashCode() => HashCode.Combine(typeof(NubTypeUInt), Width);
}

public sealed class NubTypeSInt(int width) : NubType
{
    public readonly int Width = width;

    public override string ToString() => $"i{Width}";

    public override bool Equals(NubType? other) => other is NubTypeSInt otherUInt && Width == otherUInt.Width;
    public override int GetHashCode() => HashCode.Combine(typeof(NubTypeSInt), Width);
}

public sealed class NubTypeBool : NubType
{
    public override string ToString() => "bool";

    public override bool Equals(NubType? other) => other is NubTypeBool;
    public override int GetHashCode() => HashCode.Combine(typeof(NubTypeBool));
}

public sealed class NubTypeString : NubType
{
    public override string ToString() => "string";

    public override bool Equals(NubType? other) => other is NubTypeString;
    public override int GetHashCode() => HashCode.Combine(typeof(NubTypeString));
}

public sealed class NubTypeStruct : NubType
{
    private List<Field>? _resolvedFields;
    public List<Field> Fields => _resolvedFields ?? throw new InvalidOperationException();

    public void ResolveFields(List<Field> fields)
    {
        if (_resolvedFields != null)
            throw new InvalidOperationException($"{ToString()} already resolved");

        _resolvedFields = fields;
    }

    public override string ToString()
    {
        if (_resolvedFields == null)
            return "struct <unresolved>";

        return $"struct {{ {string.Join(' ', Fields.Select(f => $"{f.Name}: {f.Type}"))} }}";
    }

    public override bool Equals(NubType? other)
    {
        if (other is not NubTypeStruct structType)
            return false;

        if (Fields.Count != structType.Fields.Count)
            return false;

        for (var i = 0; i < Fields.Count; i++)
        {
            if (Fields[i].Name != structType.Fields[i].Name)
                return false;

            if (Fields[i].Type != structType.Fields[i].Type)
                return false;
        }

        return true;
    }

    public override int GetHashCode()
    {
        var hash = new HashCode();
        hash.Add(typeof(NubTypeStruct));
        foreach (var field in Fields)
        {
            hash.Add(field.Name);
            hash.Add(field.Type);
        }

        return hash.ToHashCode();
    }

    public sealed class Field(string name, NubType type)
    {
        public readonly string Name = name;
        public readonly NubType Type = type;
    }
}

public sealed class NubTypePointer(NubType to) : NubType
{
    public readonly NubType To = to;
    public override string ToString() => $"^{To}";

    public override bool Equals(NubType? other) => other is NubTypePointer pointer && To == pointer.To;
    public override int GetHashCode() => HashCode.Combine(typeof(NubTypePointer));
}

public sealed class NubTypeFunc(List<NubType> parameters, NubType returnType) : NubType
{
    public readonly List<NubType> Parameters = parameters;
    public readonly NubType ReturnType = returnType;
    public override string ToString() => $"func({string.Join(' ', Parameters)}): {ReturnType}";

    public override bool Equals(NubType? other) => other is NubTypeFunc func && ReturnType.Equals(func.ReturnType) && Parameters.SequenceEqual(func.Parameters);

    public override int GetHashCode()
    {
        var hash = new HashCode();
        hash.Add(typeof(NubTypeFunc));
        hash.Add(ReturnType);
        foreach (var param in Parameters)
            hash.Add(param);

        return hash.ToHashCode();
    }
}