#pragma once

#if gen_time
#include "gen.hpp"

using namespace gen;

Code gen__array_base()
{
	Code t_allocator_info = def_type( name(AllocatorInfo) );

	Code header = def_struct( name(ArrayHeader), def_struct_body( 3
		, def_variable( t_allocator_info, name(Allocator) )
		, def_variable( t_uw,             name(Capacity) )
		, def_variable( t_uw,             name(Num) )
	));

	Code grow_formula = def_function( name(array_grow_formula), def_param( t_uw, name(value)), t_uw
		, def_execution( code( return 2 * value * 8; ) )
		, def_specifiers( 2, ESpecifier::Static_Member, ESpecifier::Inline )
	);

	return def_global_body( 2, header, grow_formula );
}

Code gen__array( StrC type )
{
	static Code t_allocator_info = def_type( name(AllocatorInfo) );
	static Code v_nullptr        = untyped_str( code(nullptr));

	static Code spec_ct_member     = def_specifiers( 2, ESpecifier::Constexpr, ESpecifier::Static_Member );
	static Code spec_static_inline = def_specifiers( 2, ESpecifier::Static_Member, ESpecifier::Inline );
	static Code spec_static        = def_specifier( ESpecifier::Static_Member );

	static Code using_header    = def_using( name(Header), def_type( name(ArrayHeader) ) );
	static Code ct_grow_formula = def_variable( t_auto, name(grow_formula), untyped_str( code( & array_grow_formula )), spec_ct_member );

	StrC name;
	{
		char const* name_str = str_fmt_buf( "Array_%s\0", type.Ptr );
		s32         name_len = str_len( name_str );

		name = { name_len, name_str };
	};

	Code t_array_type = def_type( name );

	Code t_type     = def_type( type );
	Code t_type_ptr = def_type( type, __, spec_ptr );
	Code t_type_ref = def_type( type, __, spec_ref );

	Code t_alias     = def_type( name(Type) );
	Code t_alias_ptr = def_type( name(Type), __, spec_ptr );
	Code t_alias_ref = def_type( name(Type), __, spec_ref );

	Code t_header     = def_type( name(Header) );
	Code t_header_ptr = def_type( name(Header), __, spec_ptr );
	Code t_header_ref = def_type( name(Header), __, spec_ref );

	Code array = {0};
	{
		Code using_type = def_using( name(Type), t_type );
		Code data       = def_variable( t_alias_ptr, name(Data) );

		Code init = def_function( name(init), def_param( t_allocator_info, name(allocator) ), t_array_type
			, def_execution( code(
				return init_reserve( allocator, grow_formula(0) );
			))
			, spec_static
		);

		Code init_reserve;
		{
			Code params = def_params( 2
				, def_param( t_allocator_info, name(allocator) )
				, def_param( t_sw, name(capacity) )
			);

			Code body = def_execution( code(
				Header* header = rcast( Header*, alloc( allocator, sizeof(Header) + sizeof(Type) ));

				if ( header == nullptr )
					return { nullptr };

				header->Allocator = allocator;
				header->Capacity  = capacity;
				header->Num       = 0;

				return { rcast( Type*, header + 1) };
			));

			init_reserve = def_function( name(init_reserve), params, t_array_type, body, spec_static );
		}

		Code append = def_function( name(append), def_param(t_alias, name(value)), t_bool
			, def_execution( code(
				Header* header = get_header();

				if ( header->Num == header->Capacity )
				{
					if ( ! grow( header->Capacity ))
						return false;

					header = get_header();
				}

				Data[ header->Num ] = value;
				header->Num++;

				return true;
			))
		);

		Code back = def_function( name(back), __, t_alias_ref
			, def_execution( code(
				Header& header = * get_header();
				return Data[ header.Num - 1 ];
			))
		);

		Code clear = def_function( name(clear), __, t_void
			, def_execution( code(
				Header& header = * get_header();
				header.Num = 0;
			))
		);

		Code fill;
		{
			Code params = def_params( 3
				, def_param( t_uw,    name(begin) )
				, def_param( t_uw,    name(end) )
				, def_param( t_alias, name(value) )
			);

			Code body = untyped_str( code(
				Header& header = * get_header();

				if ( begin < 0 || end >= header.Num )
					return false;

				for ( sw idx = begin; idx < end; idx++ )
				{
					Data[ idx ] = value;
				}

				return true;
			));

			fill = def_function( name(fill), params, t_bool, body );
		}

		Code free = def_function( name(free), __, t_void
			, def_execution( code(
				Header* header = get_header();
				gen::free( header->Allocator, header );
			))
		);

		Code get_header = def_function( name(get_header), __, t_header_ptr
			, def_execution( code(
				return rcast( Header*, Data ) - 1;
			))
		);

		Code grow = def_function( name(grow), def_param( t_uw, name(min_capacity)), t_bool
			, def_execution( code(
				Header& header = * get_header();

				uw new_capacity = grow_formula( header.Capacity );

				if ( new_capacity < min_capacity )
					new_capacity = 8;

				return set_capacity( new_capacity );
			))
		);

		Code num = def_function( name(num), __, t_uw
			, def_execution( code(
				return get_header()->Num;
			))
		);

		Code pop = def_function( name(pop), __, t_bool
			, def_execution( code(
				Header& header = * get_header();

				ZPL_ASSERT( header.Num > 0 );
				header.Num--;
			))
		);

		Code remove_at = def_function( name(remove_at), def_param( t_uw, name(idx)), t_void
			, def_execution( code(
				Header* header = get_header();
				ZPL_ASSERT( idx < header->Num );

				mem_move( header + idx, header + idx + 1, sizeof( Type ) * ( header->Num - idx - 1 ) );
				header->Num--;
			))
		);

		Code reserve = def_function( name(reserve), def_param( t_uw, name(new_capacity)), t_bool
			, def_execution( code(
				Header& header = * get_header();

				if ( header.Capacity < new_capacity )
					return set_capacity( new_capacity );

				return true;
			))
		);

		Code resize = def_function( name(resize), def_param( t_uw, name(num)), t_bool
			, def_execution( code(
				Header* header = get_header();

				if ( num > header->Capacity )
				{
					if ( ! grow( header->Capacity ))
						return false;

					header = get_header();
				}

				header->Num = num;
				return true;
			))
		);

		Code set_capacity;
		{
			Code body = def_execution( code(
				Header& header = * get_header();

				if ( new_capacity == header.Capacity )
					return true;

				if ( new_capacity < header.Num )
					header.Num = new_capacity;

				sw      size       = sizeof(Header) + sizeof(Type) * new_capacity;
				Header* new_header = rcast( Header*, alloc( header.Allocator, size ));

				if ( new_header == nullptr )
					return false;

				mem_move( new_header, & header, sizeof( Header ) + sizeof(Type) * header.Num );

				new_header->Capacity = new_capacity;

				gen::free( header.Allocator, & header );

				Data = rcast( Type*, new_header + 1);

				return true;
			));

			set_capacity = def_function( name(set_capacity), def_param( t_uw, name(new_capacity)), t_bool, body );
		}

		Code op_ptr = def_operator_cast( t_type_ptr, def_execution( code(
			return Data;
		)));

		Code body = def_struct_body( 20
			, using_header
			, using_type
			, ct_grow_formula

			, init
			, init_reserve

			, append
			, back
			, clear
			, fill
			, free
			, get_header
			, grow
			, num
			, pop
			, remove_at
			, reserve
			, resize
			, set_capacity

			, op_ptr

			, data
		);
		array = def_struct( name, body );
	}

	return array;
}


struct GenArrayRequest
{
	StrC Dependency;
	StrC Type;
};
Array<GenArrayRequest> GenArrayRequests;

void gen__array_request( StrC type, StrC dep = {} )
{
	do_once_start
		GenArrayRequests = Array<GenArrayRequest>::init( Memory::GlobalAllocator );
	do_once_end

	// Make sure we don't already have a request for the type.
	for ( sw idx = 0; idx < GenArrayRequests.num(); ++idx )
	{
		StrC const reqest_type = GenArrayRequests[ idx ].Type;

		if ( reqest_type.Len != type.Len )
			continue;

		if ( str_compare( reqest_type.Ptr, type.Ptr, reqest_type.Len ) == 0 )
			return;
	}

	GenArrayRequest request = { dep, type };
	GenArrayRequests.append( request );
}
#define gen_array( type ) gen__array_request( code(type) )

u32 gen_array_file()
{
	Builder
	gen_array_file;
	gen_array_file.open( "array.Upfront.gen.hpp" );

	Code include_zpl = def_include( txt_StrC("Bloat.hpp") );
	gen_array_file.print( include_zpl );

	gen_array_file.print( def_using_namespace( name(gen)));

	Code array_base = gen__array_base();
	gen_array_file.print( array_base );

	GenArrayRequest* current = GenArrayRequests;
	s32 left = GenArrayRequests.num();
	while (left--)
	{
		GenArrayRequest const& request = * current;

		Code generated_array = gen__array( request.Type );

		if ( request.Dependency )
		{
			char const* cmt_str = str_fmt_buf( "// Dependency for %s type", request.Type );
			s32         cmt_len = str_len( cmt_str );

			Code cmt     = def_comment( { cmt_len, cmt_str } );
			Code include = def_include( request.Dependency );

			gen_array_file.print( cmt );
			gen_array_file.print( include );
		}

		gen_array_file.print( generated_array );
		current++;
	}

	gen_array_file.write();
	return 0;
}

#endif