#version 330 core

in vec2 texCoord;
in vec3 normals;
in vec3 fragPos;

struct Camera {
	vec3 pos;
};

struct Material {
	float shininess;
	//Do I want a specular color here?
};

struct Light {
	//Directional = 1, Point = 2, Spotlight = 3
	int type;
	vec3 pos;
	vec3 direction;
	vec3 color;

	float falloffConstant;
	float falloffLinear;
	float falloffQuad;

	float cutoff;
};

uniform sampler2D modelTexture;
uniform Camera camera;
uniform Material material;
uniform Light light[100];
uniform int numberOfLights;

out vec4 outColor;

void main() {
	vec4 texColor = texture(modelTexture, texCoord);

	vec3 color = vec3(0.0);

	for (int i = 0; i < numberOfLights; i++) {
		float ambientStrength = 0.2f;
		vec3 ambient = ambientStrength * texColor.rgb;
		if (i == numberOfLights - 1 && numberOfLights == 1) {
			color += ambient;
			break;
		}

		vec3 norm = normalize(normals);
		vec3 lightDir = normalize(light[i].pos - fragPos);
		float diff = max(dot(norm, lightDir), 0.0);
		vec3 diffuse = light[i].color * (diff * texColor.rgb);

		vec3 viewDir = normalize(camera.pos - fragPos);
		vec3 reflectDir = reflect(-lightDir, norm);
		float spec = pow(max(dot(viewDir, reflectDir), 0.0), material.shininess * 32.0f);
		vec3 specular = light[i].color * spec;

		color += (ambient + diffuse + specular);
	}

	outColor = vec4(color, texColor.a);
}