ui/mdns.c

Go to the documentation of this file.

 
#include <ascii-chat/ui/mdns.h>
#include <ascii-chat/ui/terminal_screen.h>
#include <ascii-chat/ui/frame_buffer.h>
#include <ascii-chat/network/mdns/discovery.h>
#include <ascii-chat/session/session_log_buffer.h>
#include <ascii-chat/log/log.h>
#include <ascii-chat/platform/abstraction.h>
#include <ascii-chat/common.h>
#include <ascii-chat/util/string.h>
 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifndef _WIN32
#include <unistd.h>
#endif
 
/* ============================================================================
 * Log Management
 * ========================================================================== */
 
void ui_mdns_log_init(void) {
  session_log_buffer_t *buf = terminal_screen_log_init();
  if (buf) {
    log_set_session_log_buffer(buf);
  }
}
 
void ui_mdns_log_destroy(void) {
  log_clear_session_log_buffer();
  terminal_screen_log_destroy();
}
 
void ui_mdns_log_clear(void) {
  terminal_screen_log_clear();
}
 
void ui_mdns_log_append(const char *message) {
  if (message) {
    log_info("%s", message);
  }
}
 
ui_mdns_server_t *ui_mdns_query(const ui_mdns_config_t *config, int *out_count) {
  if (!out_count) {
    SET_ERRNO(ERROR_INVALID_PARAM, "out_count pointer is NULL");
    return NULL;
  }
 
  *out_count = 0;
 
  // Apply defaults if needed
  int timeout_ms = (config && config->timeout_ms > 0) ? config->timeout_ms : 2000;
  int max_servers = (config && config->max_servers > 0) ? config->max_servers : 20;
  bool quiet = (config && config->quiet);
 
  // Call the core mDNS discovery function from discovery.c
  return discovery_mdns_query(timeout_ms, max_servers, quiet, out_count);
}
 
void ui_mdns_free_results(ui_mdns_server_t *servers) {
  discovery_mdns_destroy(servers);
}
 
int ui_mdns_prompt_selection(const ui_mdns_server_t *servers, int count) {
  if (!servers || count <= 0) {
    return -1;
  }
 
  // Display available servers
  printf("\nAvailable ascii-chat servers on LAN:\n");
  for (int i = 0; i < count; i++) {
    const ui_mdns_server_t *srv = &servers[i];
    const char *addr = ui_mdns_get_best_address(srv);
    printf("  %d. %s (%s:%u)\n", i + 1, srv->name, addr, srv->port);
  }
 
  // Prompt for selection
  printf("\nSelect server (1-%d) or press Enter to cancel: ", count);
  fflush(stdout);
 
  // Read user input
  char input[32];
  if (fgets(input, sizeof(input), stdin) == NULL) {
    printf("\n");
    return -1; // EOF or error
  }
 
  // Check for empty input (Enter pressed)
  if (input[0] == '\n' || input[0] == '\r' || input[0] == '\0') {
    return -1; // User cancelled
  }
 
  // Parse input as number
  char *endptr;
  long selection = strtol(input, &endptr, 10);
 
  // Validate input
  if (selection < 1 || selection > count) {
    printf("⚠️  Invalid selection. Please enter a number between 1 and %d\n", count);
    return ui_mdns_prompt_selection(servers, count); // Re-prompt
  }
 
  return (int)(selection - 1); // Convert to 0-based index
}
 
int ui_mdns_select(const ui_mdns_server_t *servers, int count) {
  if (!servers || count <= 0) {
    // No servers found - return special code
    // Message will be printed at exit in client main
    return -1;
  }
 
  // Lock terminal to prevent concurrent logging from overwriting TUI
  bool prev_lock_state = log_lock_terminal();
 
  // Clear terminal
  log_plain("\033[2J\033[H");
 
  // Display header
  log_plain("\n");
  log_plain("\033[1m╭─ 🔍 ascii-chat Server Discovery ────────────╮\033[0m\n");
  log_plain("\033[1m│\033[0m\n");
  char found_buf[64];
  snprintf(found_buf, sizeof(found_buf), "Found %d server%s on your local network:", count, count == 1 ? "" : "s");
  log_plain("\033[1m│\033[0m %s\n", colored_string(LOG_COLOR_INFO, found_buf));
  log_plain("\033[1m│\033[0m\n");
 
  // Display server list with formatting
  for (int i = 0; i < count; i++) {
    const ui_mdns_server_t *srv = &servers[i];
    const char *addr = ui_mdns_get_best_address(srv);
 
    char idx_buf[8];
    snprintf(idx_buf, sizeof(idx_buf), "[%d]", i + 1);
    char addr_buf[128];
    snprintf(addr_buf, sizeof(addr_buf), "%s:%u", addr, srv->port);
 
    log_plain("\033[1m│\033[0m  %s %-30s %s\n", colored_string(LOG_COLOR_DEBUG, idx_buf), srv->name,
              colored_string(LOG_COLOR_WARN, addr_buf));
  }
 
  log_plain("\033[1m│\033[0m\n");
  log_plain("\033[1m╰────────────────────────────────────────────╯\033[0m\n");
 
  // Prompt for selection
  log_plain("\n");
  log_plain("Enter server number (1-%d) or press Enter to cancel: ", count);
  fflush(stdout);
 
  // Unlock before waiting for input (user might take time)
  log_unlock_terminal(prev_lock_state);
 
  // Read user input
  char input[32];
  if (fgets(input, sizeof(input), stdin) == NULL) {
    return -1;
  }
 
  // Check for empty input (Enter pressed)
  if (input[0] == '\n' || input[0] == '\r' || input[0] == '\0') {
    return -1;
  }
 
  // Parse input as number
  char *endptr;
  long selection = strtol(input, &endptr, 10);
 
  // Validate input
  if (selection < 1 || selection > count) {
    char err_buf[64];
    snprintf(err_buf, sizeof(err_buf), "Please enter a number between 1 and %d", count);
    printf("%s\n\n", colored_string(LOG_COLOR_WARN, err_buf));
    return ui_mdns_select(servers, count); // Re-prompt
  }
 
  // Lock terminal again for final output
  prev_lock_state = log_lock_terminal();
 
  // Clear screen and show connection status
  log_plain("\033[2J\033[H");
  log_plain("\n");
  char connect_buf[128];
  snprintf(connect_buf, sizeof(connect_buf), "🔗 Connecting to %s...", servers[selection - 1].name);
  log_plain("%s\n", colored_string(LOG_COLOR_INFO, connect_buf));
  log_plain("\n");
  fflush(stdout);
 
  // Brief delay so user can see the selection before logs overwrite it
  platform_sleep_ms(200);
 
  // Unlock terminal now that TUI is complete
  log_unlock_terminal(prev_lock_state);
 
  return (int)(selection - 1); // Convert to 0-based index
}
 
const char *ui_mdns_get_best_address(const ui_mdns_server_t *server) {
  if (!server) {
    return "";
  }
 
  // Prefer IPv4 > name > IPv6
  if (server->ipv4[0] != '\0') {
    return server->ipv4;
  }
  if (server->name[0] != '\0') {
    return server->name;
  }
  if (server->ipv6[0] != '\0') {
    return server->ipv6;
  }
 
  return server->address; // Fallback to address field
}