#pragma once

#if gen_time
#include "gen.hpp"

using namespace gen;

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

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

	return def_global_body( 1, header );
}

Code gen__buffer( StrC type, sw type_size )
{
	static Code t_allocator_info = def_type( name(AllocatorInfo));

	static Code using_header = def_using( name(Header), def_type( name(BufferHeader) ) );

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

		name = { name_len, name_str };
	};

	Code t_buffer_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_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 buffer = {0};
	{
		Code using_type = def_using( name(Type), t_type );
		Code data       = def_variable( t_type_ptr, name(Data) );

		Code init;
		{
			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) + capacity * sizeof(Type) ) );

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

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

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

			init = def_function( name(init), params, t_buffer_type, body, spec_static_member );
		}

		Code init_copy;
		{
			Code params = def_params( 2
				, def_param( t_allocator_info, name(allocator))
				, def_param( t_buffer_type,    name(other))
			);

			init_copy = def_function( name(init), params, t_buffer_type
				, def_execution( code(
					Header& other_header = other.get_header();
					Header* header = rcast( Header*, alloc( allocator, sizeof(Header) + other_header.Capacity * sizeof(Type) ) );

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

					header->Backing  = allocator;
					header->Capacity = other_header.Capacity;
					header->Num      = other_header.Num;

					mem_copy( header + 1, other.Data, other_header.Num * sizeof(Type) );
					return { rcast( Type*, header + 1) };
				))
			);
		}

		Code append = def_function( name(append), def_param( t_type, name(value)), t_void
			, def_execution( code(
				Header& header = get_header();
				Data[ header.Num ] = value;
				header.Num++;
			))
			, spec_inline
		);

		Code appendv;
		{
			Code params = def_params( 2
				, def_param( t_type_ptr, name( values))
				, def_param( t_sw, 	 name( num))
			);

			appendv = def_function( name(append), params, t_void
				, def_execution( code(
					Header& header = get_header();

					ZPL_ASSERT( header.Num + num <= header.Capacity);

					mem_copy( Data + header.Num, values, num * sizeof( Type ) );

					header.Num += num;
				))
				, spec_inline
			);
		}

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

		Code end = def_function( name(end), __, t_type_ref
			, def_execution( code(
				Header& header = get_header();
				return Data[ header.Num - 1 ];
			))
			, spec_inline
		);

		Code free = def_function( name(free), __, t_void
			, def_execution( code(
				Header& header = get_header();
				zpl::free( header.Backing, & header );
			))
			, spec_inline
		);

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

		Code num = def_function( name(num), __, t_sw
			, def_execution( code(
				return get_header().Num;
			))
			, spec_inline
		);

		Code pop = def_function( name(pop), __, t_type
			, def_execution( code(
				Header& header = get_header();
				header.Num--;
				return Data[ header.Num ];
			))
			, spec_inline
		);

		Code wipe = def_function( name(wipe), __, t_void
			, def_execution( code(
				Header& header = get_header();
				header.Num = 0;
				mem_set( Data, 0, header.Capacity * sizeof( Type ) );
			))
			, spec_inline
		);

		Code op_type_ptr = untyped_str( code(
			operator Type*()
			{
				return Data;
			}
		));

		buffer = def_struct( name, def_struct_body( 14
			, using_header
			, using_type

			, init
			, init_copy
			, append
			, appendv
			, clear
			, end
			, free
			, get_header
			, num
			, wipe

			, op_type_ptr

			, data
		));
	}

	return buffer;
}

struct GenBufferRequest
{
	StrC Dependency;
	StrC Type;
	sw   TypeSize;
};
Array(GenBufferRequest) GenBufferRequests;

void gen__buffer_request( StrC type, sw size, StrC dep = {} )
{
	do_once_start
		array_init( GenBufferRequests, g_allocator );
	do_once_end

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

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

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

	GenBufferRequest request = { dep, type, size};
	array_append( GenBufferRequests, request );
}
#define gen_buffer( type ) gen__buffer_request( { txt_n_len(type) }, sizeof( type ))

u32 gen_buffer_file()
{
	Builder
	gen_buffer_file;
	gen_buffer_file.open( "buffer.gen.hpp" );

	gen_buffer_file.print( def_include( StrC::from("Bloat.hpp")) );
	gen_buffer_file.print( gen__buffer_base() );

	GenBufferRequest* current = GenBufferRequests;
	s32 left = array_count( GenBufferRequests );
	while (left--)
	{
		GenBufferRequest const& request = * current;

		Code generated_buffer = gen__buffer( current->Type, current->TypeSize );

		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_buffer_file.print( cmt );
			gen_buffer_file.print( include );
		}

		gen_buffer_file.print( generated_buffer );
		current++;
	}

	gen_buffer_file.write();
	return 0;
}

#endif // gen_time