#include "funkey_prod_screens.h"
#include <signal.h> 

#define SHELL_CMD_PID_RECORD                "pid record"
#define SHELL_CMD_POWERDOWN_HANDLE          "powerdown handle"

/* Static variable */
static int stop_menu_loop = 0;
static int res = EXIT_FAILURE;

/* Handler for SIGUSR1, caused by closing the console */
void handle_sigusr1(int sig) 
{ 

    /* Send command to cancel any previously scheduled powerdown */
    if (popen(SHELL_CMD_POWERDOWN_HANDLE, "r") == NULL)
    {
        /* Countdown is still ticking, so better do nothing
           than start writing and get interrupted!
        */
        printf("Failed to cancel scheduled shutdown\n");
        //exit(0);
    }

    //printf("Caught signal USR1 %d\n", sig); 
    stop_menu_loop = 1;
    res = 0;
}

/// -------------- FUNCTIONS IMPLEMENTATION --------------
static int wait_event_loop(){

    SDL_Event event;
 
    /// -------- Main loop ---------
    while (!stop_menu_loop)
    {
        /// -------- Handle Keyboard Events ---------
        while (SDL_PollEvent(&event))
        switch(event.type)
        {
            case SDL_QUIT:
                stop_menu_loop = 1;
        break;
        case SDL_KEYDOWN:
            switch (event.key.keysym.sym)
            {

                case SDLK_m:
                    stop_menu_loop = 1;
                    res = ERROR_MANUAL_FAIL;
                    break;

                //case SDLK_n:
                case SDLK_ESCAPE:
                    stop_menu_loop = 1;
                    res = 0;
                    break;

                default:
                    break;
            }
        }

        /* To inverstigate but with Buildroot, we need this: */
        SDL_Flip(hw_surface);

        /* Sleep for some time */
        SDL_Delay(SLEEP_PERIOD_MS);
    }

    return res;
}

int launch_prod_screen_magnet(int argc, char *argv[]){
    SDL_Surface *text_surface = NULL;
    SDL_Rect text_pos;

    /* Record current PID */
    char shell_cmd[100];
    sprintf(shell_cmd, "%s %d", SHELL_CMD_PID_RECORD, getpid());
    system(shell_cmd);

    /* Init Signals */
    signal(SIGUSR1, handle_sigusr1); 

    /* Fill screen white */
    SDL_FillRect(hw_surface, NULL, SDL_MapRGBA(hw_surface->format, bg_color.r, bg_color.g, bg_color.b, 0) );

    /* Write Title */
    text_surface = TTF_RenderText_Shaded(font_title, prog_title, text_color, bg_color);
    text_pos.x = display_width/2 - text_surface->w/2;
    text_pos.y = Y_PADDING;
    SDL_BlitSurface(text_surface, NULL, hw_surface, &text_pos);
    SDL_FreeSurface(text_surface);

    /* Write 
    "   MAGNET TEST 
    (close the console)*/
    int y_pad_tmp = 11;
    text_surface = TTF_RenderText_Shaded(font_title, "MAGNET TEST", text_color, bg_color);
    text_pos.x = display_width/2 - text_surface->w/2;
    text_pos.y = display_height/2 - text_surface->h/2 - y_pad_tmp;
    SDL_BlitSurface(text_surface, NULL, hw_surface, &text_pos);
    SDL_FreeSurface(text_surface);
    
    text_surface = TTF_RenderText_Shaded(font_title, "(close the console)", text_color, bg_color);
    text_pos.x = display_width/2 - text_surface->w/2;
    text_pos.y = display_height/2 - text_surface->h/2 + y_pad_tmp;
    SDL_BlitSurface(text_surface, NULL, hw_surface, &text_pos);
    SDL_FreeSurface(text_surface);

    /* Write:
        "Press 
        L=FAIL
    */
    SDL_Color red_color={220,20,20};
    text_surface = TTF_RenderText_Shaded(font_info, "Press", red_color, bg_color);
    text_pos.x = X_PADDING;
    text_pos.y = display_height - Y_PADDING - 2*text_surface->h;
    SDL_BlitSurface(text_surface, NULL, hw_surface, &text_pos);
    SDL_FreeSurface(text_surface);
    text_surface = TTF_RenderText_Shaded(font_info, "L=FAIL", red_color, bg_color);
    text_pos.x = X_PADDING;
    text_pos.y = display_height - Y_PADDING - text_surface->h;
    SDL_BlitSurface(text_surface, NULL, hw_surface, &text_pos);
    SDL_FreeSurface(text_surface);

    /* Write:
        Press 
        R=OK"
    */
    /*SDL_Color green_color={20,220,20};
    text_surface = TTF_RenderText_Shaded(font_info, "Press", green_color, bg_color);
    text_pos.x = display_width - text_surface->w - X_PADDING;
    text_pos.y = display_height - Y_PADDING - 2*text_surface->h;
    SDL_BlitSurface(text_surface, NULL, hw_surface, &text_pos);
    SDL_FreeSurface(text_surface);
    text_surface = TTF_RenderText_Shaded(font_info, "R=OK", green_color, bg_color);
    text_pos.x = display_width - text_surface->w - X_PADDING;
    text_pos.y = display_height - Y_PADDING - text_surface->h;
    SDL_BlitSurface(text_surface, NULL, hw_surface, &text_pos);
    SDL_FreeSurface(text_surface);*/

    /// Render screen
    //SDL_Flip(hw_surface);

    /// 
    int res = wait_event_loop();
    return res;
}