sbox-public/engine/Tools/InteropGen/Arguments/ArgDefinedStruct.cs

using System;

namespace Facepunch.InteropGen;

public class ArgDefinedStruct : Arg
{
	public Struct Type { get; set; }


	public ArgDefinedStruct( Struct c, string name, string[] flags )
	{
		Type = c;
		Name = name;
		Flags = flags;
	}

	public override string ManagedType => Type.ManagedNameWithNamespace;
	public override string NativeType => Type.NativeNameWithNamespace;
	public override string ManagedDelegateType => ManagedType;

	public override string NativeDelegateType => NativeType;

	public override string GetManagedDelegateType( bool incoming )
	{
		return Type.IsPointer
			? $"IntPtr /* PtrHandle:{Type.NativeName}  */"
			: !incoming && Flags == null && !Type.IsEnum && !Type.HasAttribute( "small" )
			? $"{ManagedType}*"
			: base.GetManagedDelegateType( incoming );
	}

	public override string ReturnWrapCall( string functionCall, bool native )
	{
		if ( native )
		{
			if ( Type.CreateUsing != null )
			{
				functionCall = $"{Type.CreateUsing}( {functionCall} )";
			}
		}

		return base.ReturnWrapCall( functionCall, native );
	}

	public override string GetNativeDelegateType( bool incoming )
	{
		if ( incoming && Name != null && Flags == null && !Type.IsPointer && !Type.IsEnum && !Type.HasAttribute( "small" ) )
		{
			return $"{NativeDelegateType}*";
		}

		return base.GetNativeDelegateType( incoming );
	}

	public override string ToInterop( bool native, string code = null )
	{
		return Type.IsPointer
			? code ?? Name
			: !native && code == null && Name != null && Flags == null && !Type.IsEnum && !Type.HasAttribute( "small" )
			? $"&{Name}"
			: base.ToInterop( native, code );
	}

	public override string FromInterop( bool native, string code = null )
	{
		// non-small arg structs are passed as ptr, so read it as ptr
		if ( native && Flags == null && !Type.IsPointer && !Type.IsEnum && !Type.HasAttribute( "small" ) )
		{
			return $"*{code ?? Name}";
		}

		return base.FromInterop( native, code );
	}

	public override string DefaultValue => $"{NativeType}()";
}