#version 300 es
precision highp float;

uniform float iTime;
uniform vec2  iResolution;

uniform vec3 baseColor; // value=.75,0,0
uniform vec3 specularColor; // value=1,1,0.9
uniform vec3 ambientColor; // value=0,0,.1
uniform vec3 lightColor; // value=1,1,1
uniform float lightAzimuth; // value=0.69, min=0, max=6.28318530718, step=0.03141592653
uniform float lightZenith; // value=1.10, min=0, max=3.14159265359, step=0.03141592653
uniform float roughness; // value=0.1, min=0, max=1, step=0.01

in vec2 vScreen; // screen coords
out vec4 fragColor;

vec4 phong(vec3 normal, vec3 lightDir, vec3 viewDir, float shininess) {
    vec3 N = normalize(normal);
    vec3 L = normalize(lightDir);
    vec3 V = normalize(viewDir);
    vec3 H = normalize(L + V);
    
    float NdotL = max(dot(N, L), 0.0);
    vec3 diffuse = baseColor * NdotL;
    
    vec3 R = reflect(-L, N);
    float VdotR = max(dot(V, R), 0.0);
    float spec = pow(VdotR, shininess);
    vec3 specular = specularColor * spec;
    
    vec3 color = ambientColor + lightColor * (diffuse + specular);
    return vec4(color, 1.0);
}

vec4 sphereShaders(float radius) {
    // sphere geometry
    float z = sqrt(pow(radius, 2.0) - pow(vScreen.x, 2.0) - pow(vScreen.y, 2.0));
    vec3 fragPos = vec3(vScreen, z);
    vec3 normal = fragPos / radius;
    // prepare other properties
    vec3 cameraPos = vec3(0.0, 0.0, 1.0);
    vec3 viewDir = cameraPos - fragPos;
    vec3 lightDir = vec3(
        sin(lightZenith) * cos(lightAzimuth),
        cos(lightZenith),
        sin(lightZenith) * sin(lightAzimuth)
    );
    // Compute shininess from roughness
    float shininess = max(0.0, (2.0 / (roughness * roughness)) - 2.0);
    return phong(normal, lightDir, viewDir, shininess);
}

vec4 backgroundShaders() {
    return vec4(vec3(0.1), 1.0);
}

void main() {
    float radius = 0.5;
    
    if(length(vScreen) <= radius) {
        fragColor = sphereShaders(radius);
    } else {
        fragColor = backgroundShaders();
    }
}