src/client/display.c

Go to the documentation of this file.

 
#include "display.h"
#include "main.h"
#include "../main.h" // Global exit API
 
#include "display.h"
#include "capture.h"
#include "session/keyboard_handler.h"
#include <ascii-chat/ui/keyboard_help.h>
#include <ascii-chat/ui/splash.h>
#include <ascii-chat/platform/abstraction.h>
#include <ascii-chat/platform/keyboard.h>
#include <ascii-chat/options/options.h>
#include <ascii-chat/options/rcu.h> // For RCU-based options access
#include <ascii-chat/video/ascii/ascii.h>
#include <ascii-chat/media/source.h>
#include <ascii-chat/common.h>
 
#include <fcntl.h>
#include <string.h>
#include <stdio.h>
#include <ascii-chat/atomic.h>
#include <ascii-chat/debug/named.h>
#ifndef _WIN32
#include <unistd.h>
#endif
 
/* ============================================================================
 * Session Display Context
 * ============================================================================ */
 
static session_display_ctx_t *g_display_ctx = NULL;
 
static atomic_t g_is_first_frame_of_connection = {0};
 
static bool g_keyboard_enabled = false;
 
static session_capture_ctx_t *g_display_capture_ctx = NULL;
 
/* TTY detection and terminal reset now handled by session display library */
 
/* Frame rendering now handled by session display library */
 
/* ============================================================================
 * Internal Setter Functions (for framework integration)
 * ============================================================================ */
 
void display_set_context(session_display_ctx_t *display_ctx) {
  g_display_ctx = display_ctx;
}
 
/* ============================================================================
 * Public Interface Functions
 * ============================================================================ */
 
int display_init() {
  // Build display configuration from options
  session_display_config_t config = {
      .snapshot_mode = GET_OPTION(snapshot_mode),
      .palette_type = GET_OPTION(palette_type),
      .custom_palette = GET_OPTION(palette_custom_set) ? GET_OPTION(palette_custom) : NULL,
      .color_mode = TERM_COLOR_AUTO // Will be overridden by command-line options
  };
 
  // Create display context using session library
  g_display_ctx = session_display_create(&config);
  if (!g_display_ctx) {
    SET_ERRNO(ERROR_DISPLAY, "Failed to initialize display");
    return -1;
  }
 
  // Check if client mode should load local media (--file or --url)
  const char *media_url = GET_OPTION(media_url);
  const char *media_file = GET_OPTION(media_file);
 
  if ((media_url && strlen(media_url) > 0) || (media_file && strlen(media_file) > 0)) {
    // Client mode with local media file/URL - create capture context for keyboard controls
    session_capture_config_t capture_config = {0};
    capture_config.target_fps = 60;
    capture_config.resize_for_network = false;
    capture_config.should_exit_callback = NULL;
    capture_config.callback_data = NULL;
 
    if (media_url && strlen(media_url) > 0) {
      capture_config.type = MEDIA_SOURCE_FILE;
      capture_config.path = media_url;
    } else {
      if (strcmp(media_file, "-") == 0) {
        capture_config.type = MEDIA_SOURCE_STDIN;
        capture_config.path = NULL;
      } else {
        capture_config.type = MEDIA_SOURCE_FILE;
        capture_config.path = media_file;
      }
    }
 
    capture_config.loop = GET_OPTION(media_loop);
    capture_config.initial_seek_timestamp = GET_OPTION(media_seek_timestamp);
 
    g_display_capture_ctx = session_capture_create(&capture_config);
    if (!g_display_capture_ctx) {
      log_warn("Failed to create capture context for local media - keyboard seek/pause disabled");
      g_display_capture_ctx = NULL;
    }
  }
 
  // Keyboard is pre-initialized from asciichat_shared_init()
  // Only enable keyboard input if stdin is a TTY (avoid interfering with piped/redirected I/O)
  g_keyboard_enabled = terminal_is_stdin_tty();
 
  // Register display atomics with named debug registry
  static bool display_atomics_registered = false;
  if (!display_atomics_registered) {
    NAMED_REGISTER_ATOMIC(&g_is_first_frame_of_connection, "first_frame_of_display_connection", NULL);
    display_atomics_registered = true;
  }
 
  return 0;
}
 
bool display_has_tty() {
  return session_display_has_tty(g_display_ctx);
}
 
void display_full_reset() {
  if (g_display_ctx && !should_exit()) {
    session_display_reset(g_display_ctx);
  }
}
 
void display_reset_for_new_connection() {
  atomic_store_bool(&g_is_first_frame_of_connection, true);
}
 
void display_disable_logging_for_first_frame() {
  // Atomically check if this is the first frame and mark as not-first
  // Avoids TOCTOU race where multiple threads could both think they're first
  bool was_first_frame = atomic_exchange_bool(&g_is_first_frame_of_connection, false);
 
  if (was_first_frame) {
    // Disable terminal logging before clearing display and rendering first frame
    log_set_terminal_output(false);
 
    // Signal the intro splash screen to stop - first frame is ready to render
    splash_intro_done();
  }
}
 
void display_render_frame(const char *frame_data) {
  if (!frame_data) {
    SET_ERRNO(ERROR_INVALID_PARAM, "Attempted to render NULL frame data");
    return;
  }
 
  // Stop splash screen animation on first frame
  // This must be called BEFORE any frame rendering to prevent splash/frame flickering
  display_disable_logging_for_first_frame();
 
  // Render help screen if active (suppresses frame rendering)
  if (keyboard_help_is_active(g_display_ctx)) {
    keyboard_help_render(g_display_ctx);
 
    // Still poll keyboard for interactive controls while help is visible
    if (g_keyboard_enabled) {
      keyboard_key_t key = keyboard_read_nonblocking();
      if (key != KEY_NONE) {
        session_handle_keyboard_input(g_display_capture_ctx, g_display_ctx, key);
      }
    }
    return; // Skip normal frame rendering
  }
 
  // Use session display library for frame rendering
  session_display_render_frame(g_display_ctx, frame_data);
 
  // Poll for keyboard input and handle interactive controls
  // - If client mode has local media (--file/--url), pass capture context for full controls
  //   (seek, pause, play, volume, color mode, flip)
  // - If client mode is network-only, pass NULL (volume, color mode, flip work; seek/pause ignored)
  if (g_keyboard_enabled) {
    keyboard_key_t key = keyboard_read_nonblocking();
    if (key != KEY_NONE) {
      session_handle_keyboard_input(g_display_capture_ctx, g_display_ctx, key);
    }
  }
}
 
void display_cleanup() {
  // Cleanup keyboard input if it was initialized
  if (g_keyboard_enabled) {
    keyboard_destroy();
    g_keyboard_enabled = false;
  }
 
  // Cleanup optional local media capture context if it was created
  if (g_display_capture_ctx) {
    session_capture_destroy(g_display_capture_ctx);
    g_display_capture_ctx = NULL;
  }
 
  // Display context is now managed by session_client_like framework
  // Do not destroy it here - the framework will handle cleanup
  g_display_ctx = NULL;
}