src/client/display.c

Go to the documentation of this file.

 
#include "display.h"
#include "main.h"
#include "../main.h" // Global exit API
 
#include <ascii-chat/session/display.h>
#include <ascii-chat/session/capture.h>
#include <ascii-chat/session/keyboard_handler.h>
#include <ascii-chat/ui/help_screen.h>
#include <ascii-chat/ui/splash.h>
#include <ascii-chat/platform/abstraction.h>
#include <ascii-chat/platform/keyboard.h>
#include <ascii-chat/platform/util.h>
#include <ascii-chat/options/options.h>
#include <ascii-chat/options/rcu.h> // For RCU-based options access
#include <ascii-chat/video/ascii.h>
#include <ascii-chat/media/source.h>
#include <ascii-chat/common.h>
 
#include <fcntl.h>
#include <string.h>
#include <stdio.h>
#include <stdatomic.h>
#ifndef _WIN32
#include <unistd.h>
#endif
 
/* ============================================================================
 * Session Display Context
 * ============================================================================ */
 
static session_display_ctx_t *g_display_ctx = NULL;
 
static atomic_bool g_is_first_frame_of_connection = true;
 
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;
    }
  }
 
  // Initialize keyboard input for interactive controls (volume, color mode, flip, seek, pause)
  // Only initialize in TTY mode to avoid interfering with piped/redirected I/O
  if (terminal_is_stdin_tty()) {
    asciichat_error_t kb_result = keyboard_init();
    if (kb_result == ASCIICHAT_OK) {
      g_keyboard_enabled = true;
    } else {
      // Non-fatal: client can work without keyboard support
      log_warn("Failed to initialize keyboard input: %s", asciichat_error_string(kb_result));
      g_keyboard_enabled = false;
    }
  }
 
  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(&g_is_first_frame_of_connection, true);
}
 
void display_disable_logging_for_first_frame() {
  // Disable terminal logging before clearing display and rendering first frame
  if (atomic_load(&g_is_first_frame_of_connection)) {
    log_set_terminal_output(false);
    atomic_store(&g_is_first_frame_of_connection, 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 (session_display_is_help_active(g_display_ctx)) {
    session_display_render_help(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;
}