#include "main.h"
#include "state.h"
#include "text.h"
#include "textures.h"
#include "lua_core.h"

static void checkFontRealloc( int i ) {
	if ( i == state->fontCount ) {
		state->fontCount++;
	}

	if ( state->fontCount == state->fontAlloc ) {
		state->fontAlloc += ALLOC_PAGE_SIZE;
		state->fonts = realloc( state->fonts, state->fontAlloc * sizeof( Font* ) );

		for ( i = state->fontCount; i < state->fontAlloc; i++ ) {
			state->fonts[i] = NULL;
		}
	}
}

bool validFont( size_t id ) {
	if ( id < 0 || state->fontCount < id || state->fonts[ id ] == NULL ) {
		TraceLog( state->logLevelInvalid, "%s %d", "Invalid font", id );
		return false;
	}
	else {
		return true;
	}
}

static int newFont() {
	int i = 0;

	for ( i = 0; i < state->fontCount; i++ ) {
		if ( state->fonts[i] == NULL ) {
			break;
		}
	}
	state->fonts[i] = malloc( sizeof( Font ) );
	checkFontRealloc( i );

	return i;
}

/*
## Text - Loading
*/

/*
> font = RL.LoadFont( string fileName )

Load font from file into GPU memory ( VRAM )

- Failure return -1
- Success return int
*/
int ltextLoadFont( lua_State *L ) {
	if ( !lua_isstring( L, 1 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.LoadFont( string fileName )" );
		lua_pushinteger( L, -1 );
		return 1;
	}
	if ( FileExists( lua_tostring( L, 1 ) ) ) {
		int i = newFont();
		*state->fonts[i] = LoadFont( lua_tostring( L, 1 ) );
		lua_pushinteger( L, i );
		return 1;
	}
	else {
		lua_pushinteger( L, -1 );
		return 1;
	}

	return 1;
}

/*
> font = RL.LoadFontEx( string fileName, int fontSize )

Load font from file with extended parameters. Loading the default character set

- Failure return -1
- Success return int
*/
int ltextLoadFontEx( lua_State *L ) {
	if ( !lua_isstring( L, 1 ) || !lua_isnumber( L, 2 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.LoadFontEx( string fileName, int fontSize )" );
		lua_pushinteger( L, -1 );
		return 1;
	}
	int fontSize = lua_tointeger( L, 2 );

	if ( FileExists( lua_tostring( L, 1 ) ) ) {
		int i = newFont();
		*state->fonts[i] = LoadFontEx( lua_tostring( L, 1 ), fontSize, NULL, 0 );
		lua_pushinteger( L, i );
		return 1;
	}
	else {
		lua_pushinteger( L, -1 );
		return 1;
	}

	return 1;
}

/*
> font = RL.LoadFontFromImage( Image image, Color key, int firstChar )

Load font from Image ( XNA style )

- Failure return -1
- Success return int
*/
int ltextLoadFontFromImage( lua_State *L ) {
	Image *image = luaL_checkudata( L, 1, "Image" );
	Color key = uluaGetColorIndex( L, 2 );
	int firstChar = lua_tointeger( L, 3 );

	int i = newFont();
	*state->fonts[i] = LoadFontFromImage( *image, key, firstChar );
	lua_pushinteger( L, i );

	return 1;
}

/*
> success = RL.UnloadFont( Font font )

Unload Font from GPU memory ( VRAM )

- Failure return false
- Success return true
*/
int ltextUnloadFont( lua_State *L ) {
	if ( !lua_isnumber( L, 1 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.UnloadFont( Font font )" );
		lua_pushboolean( L, false );
		return 1;
	}
	size_t FontId = lua_tointeger( L, 1 );

	if ( !validFont( FontId ) ) {
		lua_pushboolean( L, false );
		return 1;
	}
	UnloadFont( *state->fonts[ FontId ] );
	state->fonts[ FontId ] = NULL;
	lua_pushboolean( L, true );

	return 1;
}

/*
## Text - Draw
*/

/*
> success = RL.DrawFPS( Vector2 pos )

Draw current FPS

- Failure return false
- Success return true
*/
int ltextDrawFPS( lua_State *L ) {
	if ( !lua_istable( L, 1 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.DrawFPS( Vector2 pos )" );
		lua_pushboolean( L, false );
		return 1;
	}
	Vector2 pos = uluaGetVector2Index( L, 1 );

	DrawFPS( pos.x, pos.y );
	lua_pushboolean( L, true );

	return 1;
}

/*
> success = RL.DrawText( Font font, string text, Vector2 position, float fontSize, float spacing, Color tint )

Draw text using font and additional parameters

- Failure return false
- Success return true
*/
int ltextDrawText( lua_State *L ) {
	if ( !lua_isnumber( L, 1 ) || !lua_isstring( L, 2 ) || !lua_istable( L, 3 )
	|| !lua_isnumber( L, 4 ) || !lua_isnumber( L, 5 ) || !lua_istable( L, 6 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.DrawText( Font font, string text, Vector2 position, float fontSize, float spacing, Color tint )" );
		lua_pushboolean( L, false );
		return 1;
	}
	size_t fontId = lua_tointeger( L, 1 );
	Vector2 position = uluaGetVector2Index( L, 3 );
	float fontSize = lua_tonumber( L, 4 );
	float spacing = lua_tonumber( L, 5 );
	Color tint = uluaGetColorIndex( L, 6 );

	if ( !validFont( fontId ) ) {
		lua_pushboolean( L, false );
		return 1;
	}
	DrawTextEx( *state->fonts[ fontId ], lua_tostring( L, 2 ), position, fontSize, spacing, tint );
	lua_pushboolean( L, true );

	return 1;
}

/*
> success = RL.DrawTextPro( Font font, string text, Vector2 position, Vector2 origin, float rotation, float fontSize, float spacing, Color tint )

Draw text using Font and pro parameters ( rotation )

- Failure return false
- Success return true
*/
int ltextDrawTextPro( lua_State *L ) {
	if ( !lua_isnumber( L, 1 ) || !lua_isstring( L, 2 ) || !lua_istable( L, 3 ) || !lua_istable( L, 4 )
	|| !lua_isnumber( L, 5 ) || !lua_isnumber( L, 6 ) || !lua_isnumber( L, 7 ) || !lua_istable( L, 8 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.DrawTextPro( Font font, string text, Vector2 position, Vector2 origin, float rotation, float fontSize, float spacing, Color tint )" );
		lua_pushboolean( L, false );
		return 1;
	}
	size_t fontId = lua_tointeger( L, 1 );
	Vector2 position = uluaGetVector2Index( L, 3 );
	Vector2 origin = uluaGetVector2Index( L, 4 );
	float rotation = lua_tonumber( L, 5 );
	float fontSize = lua_tonumber( L, 6 );
	float spacing = lua_tonumber( L, 7 );
	Color tint = uluaGetColorIndex( L, 8 );

	if ( !validFont( fontId ) ) {
		lua_pushboolean( L, false );
		return 1;
	}
	DrawTextPro( *state->fonts[ fontId ], lua_tostring( L, 2 ), position, origin, rotation, fontSize, spacing, tint );
	lua_pushboolean( L, true );

	return 1;
}

/*
## Text - Misc
*/

/*
> size = RL.MeasureText( Font font, string text, float fontSize, float spacing )

Measure string size for Font

- Failure return false
- Success return Vector2
*/
int ltextMeasureText( lua_State *L ) {
	if ( !lua_isnumber( L, 1 ) || !lua_isstring( L, 2 ) || !lua_isnumber( L, 3 ) || !lua_isnumber( L, 4 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.MeasureText( Font font, string text, float fontSize, float spacing )" );
		lua_pushboolean( L, false );
		return 1;
	}
	size_t fontId = lua_tointeger( L, 1 );
	float fontSize = lua_tonumber( L, 3 );
	float spacing = lua_tonumber( L, 4 );

	if ( !validFont( fontId ) ) {
		lua_pushboolean( L, false );
		return 1;
	}
	uluaPushVector2( L, MeasureTextEx( *state->fonts[ fontId ], lua_tostring( L, 2 ), fontSize, spacing ) );

	return 1;
}

/*
> baseSize = RL.GetFontBaseSize( Font font )

Get font base size ( default chars height )

- Failure return false
- Success return int
*/
int ltextGetFontBaseSize( lua_State *L ) {
	if ( !lua_isnumber( L, 1 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.GetFontBaseSize( Font font )" );
		lua_pushboolean( L, false );
		return 1;
	}
	size_t fontId = lua_tointeger( L, 1 );

	if ( !validFont( fontId ) ) {
		lua_pushboolean( L, false );
		return 1;
	}
	lua_pushinteger( L, state->fonts[ fontId ]->baseSize );

	return 1;
}

/*
> glyphCount = RL.GetFontGlyphCount( Font font )

Get font number of glyph characters

- Failure return false
- Success return int
*/
int ltextGetFontGlyphCount( lua_State *L ) {
	if ( !lua_isnumber( L, 1 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.GetFontGlyphCount( Font font )" );
		lua_pushboolean( L, false );
		return 1;
	}
	size_t fontId = lua_tointeger( L, 1 );

	if ( !validFont( fontId ) ) {
		lua_pushboolean( L, false );
		return 1;
	}
	lua_pushinteger( L, state->fonts[ fontId ]->glyphCount );

	return 1;
}

/*
> glyphPadding = RL.GetFontGlyphPadding( Font font )

Get font padding around the glyph characters

- Failure return false
- Success return int
*/
int ltextGetFontGlyphPadding( lua_State *L ) {
	if ( !lua_isnumber( L, 1 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.GetFontGlyphPadding( Font font )" );
		lua_pushboolean( L, false );
		return 1;
	}
	size_t fontId = lua_tointeger( L, 1 );

	if ( !validFont( fontId ) ) {
		lua_pushboolean( L, false );
		return 1;
	}
	lua_pushinteger( L, state->fonts[ fontId ]->glyphPadding );

	return 1;
}

/*
> textureTable = RL.GetFontTexture( Font font )

Get font texture atlas containing the glyphs. NOTE! Texture in table form.

- Failure return false
- Success return table
*/
int ltextGetFontTexture( lua_State *L ) {
	if ( !lua_isnumber( L, 1 ) ) {
		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.GetFontTexture( Font font )" );
		lua_pushboolean( L, false );
		return 1;
	}
	size_t fontId = lua_tointeger( L, 1 );

	if ( !validFont( fontId ) ) {
		lua_pushboolean( L, false );
		return 1;
	}
	uluaPushTexture( L, state->fonts[ fontId ]->texture );

	return 1;
}