/* SPDX-License-Identifier: MIT */
/* Copyright (c) 2026 Oleksandr Zhylin */

#include "app.h"
#include <locale.h>
#include <ncursesw/ncurses.h>
#include <netinet/in.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <unistd.h>
#include <wchar.h>

void ac_file(WINDOW *win, int is_action_w);

app_t *app_init(app_t *app) {
  app->top_text = NULL;
  /* get values from terminal size */
  int32_t y_max, x_max;
  getmaxyx(stdscr, y_max, x_max);

  /* allocate app struct memory */
  if (app == NULL) {
    app = calloc(1, sizeof(app_t));
  }

  /* associate values for screen size */
  app->coordinates.max_x = app->coordinates.cur_x = x_max;
  app->coordinates.max_y = app->coordinates.cur_y = y_max;

  app->win = newwin(app->coordinates.cur_y, app->coordinates.cur_x, 0, 0);

  /* dialogue */
  app->modal.win = NULL;
  app->modal.is_initiated = 0;

  /* NULL to main_ui */
  MAIN_UI_RESET(app);

  keypad(app->win, TRUE);

  app->callback_after_notification = NULL;

  app->modal.is_initiated = false;  
  app->modal.needs_destroy = false;  

  /* refresh the windows */
  wnoutrefresh(app->win);
  doupdate();

  return app;
}

void app_init_nc(void) {
  initscr();
  cbreak();
  keypad(stdscr, TRUE);
  noecho();
  curs_set(false);
  set_escdelay(50);
  setlocale(LC_ALL, "");

  if (!has_colors()) {
    printf("Your terminal does not support color\n");
  } else {
    start_color();
    init_pair(1, COLOR_RED, COLOR_BLACK);
    init_pair(2, COLOR_GREEN, COLOR_BLACK);
    init_pair(modal_color_pair, COLOR_BLUE, COLOR_WHITE);
    init_pair(4, COLOR_CYAN, COLOR_WHITE);
  }
}

void app_draw_borders(app_t *app) {
  if (app->modal.win != NULL && !app->modal.needs_update) {
    return;
  }
  clear();
  box(app->win, 0, 0);
}

int32_t print_bottom_menu_option(WINDOW *win, char *key, char *title, int32_t y,
                                 int32_t x, int32_t size) {
  int32_t key_len = 0;
  int32_t title_len = 0;

  wattrset(win, A_BOLD);
  mvwprintw(win, y, x, " %s%n", key, &key_len);
  size -= key_len + strlen(title) + 1;
  wattrset(win, COLOR_PAIR(4) | A_REVERSE);
  mvwprintw(win, y, x + key_len, "%s %*s%n", title, size, "", &title_len);
  wattrset(win, COLOR_PAIR(0) | A_REVERSE);
  return key_len + title_len;
}

void app_draw_bbar(app_t *app);

void app_draw_bars(app_t *app) {
  char top_text[64] = "Hello!";

  wattrset(app->win, A_REVERSE);

  /* print top and bottom bars */
  for (int i = 1; i < app->coordinates.cur_x - 1; i++) {
    mvwprintw(app->win, 1, i, " ");
    mvwprintw(app->win, app->coordinates.cur_y - 2, i, " ");
  }

  if (app->top_text != NULL) {
    mvwaddnwstr(app->win, 1, 2, app->top_text, wcslen(app->top_text));
  } else if (app->params && app->params->is_connected) {
    int32_t ip_address = app->params->addr;
    u_char ip_addr[4];
    mempcpy(ip_addr, &ip_address, 4);
    sprintf(top_text, "Connected to %u.%u.%u.%u, user: %s", ip_addr[0],
            ip_addr[1], ip_addr[2], ip_addr[3], app->params->uname);
    mvwprintw(app->win, 1, 2, "%s", top_text);
  }

  wattroff(app->win, A_REVERSE);

  /* add content to the bottom bar */
  if (app->main_ui.b_keys_len > 0) {
    app_draw_bbar(app);
  }
}

void app_draw_bbar(app_t *app) {
  wattrset(app->win, A_REVERSE);
  int t_margin = 0; /* tab margin */
  int c_size =
      app->coordinates.max_x / app->main_ui.b_keys_len; /* column size */
  int incr_last = app->coordinates.max_x %
                  app->main_ui.b_keys_len; /* increment size fo the last element
                                              (fill empty space) */

  struct action_key *k = NULL;
  if (app->main_ui.b_keys_len > 1) {
    for (int i = 0; i < app->main_ui.b_keys_len - 1; i++) {
      k = &(app->main_ui.b_keys)[i];
      t_margin += print_bottom_menu_option(app->win, k->key, k->title,
                                           app->coordinates.cur_y - 2, t_margin,
                                           c_size);
    }
  }

  k = &(app->main_ui.b_keys)[app->main_ui.b_keys_len - 1];

  t_margin += print_bottom_menu_option(app->win, k->key, k->title,
                                       app->coordinates.cur_y - 2, t_margin,
                                       c_size + incr_last);
  wattroff(app->win, A_REVERSE);
}

void app_refresh(app_t *app) {
  app_draw_borders(app);
  wnoutrefresh(app->win);
  if (app->modal.win != NULL) {
    if (app->modal.is_initiated != 0) {
      wnoutrefresh(app->modal.win);
    }
  } else if (app->main_ui.cb_ui_refresh != NULL) {
    app->main_ui.cb_ui_refresh(app);
  }
  doupdate();
}

void app_destroy(app_t *app, int32_t exit_code) {
  delwin(app->win);
  endwin();
  app->win = NULL;
  exit(exit_code);
}