display.c

Go to the documentation of this file.

 
#include "display.h"
#include "main.h"
 
#include "platform/abstraction.h"
#include "platform/util.h"
#include "options/options.h"
#include "options/rcu.h" // For RCU-based options access
#include "video/ascii.h"
 
#include <fcntl.h>
#include <string.h>
#include <stdio.h>
#include <stdatomic.h>
 
/* ============================================================================
 * TTY Management
 * ============================================================================ */
 
tty_info_t g_tty_info = {-1, NULL, false};
 
static bool g_has_tty = false;
 
static atomic_bool g_is_first_frame_of_connection = true;
 
/* ============================================================================
 * TTY Detection and Initialization
 * ============================================================================ */
 
static tty_info_t display_get_current_tty(void) {
  // Use the platform abstraction layer function
  return get_current_tty();
}
 
static void full_terminal_reset(int fd) {
  // Skip terminal control sequences in snapshot mode - just print raw ASCII
  if (!GET_OPTION(snapshot_mode)) {
    terminal_reset(fd);             // Reset using the proper TTY fd
    console_clear(fd);              // This calls terminal_clear_screen() + terminal_cursor_home(fd)
    terminal_clear_scrollback(fd);  // Clear scrollback using the proper TTY fd
    terminal_hide_cursor(fd, true); // Hide cursor on the specific TTY
    terminal_flush(fd);             // Flush the specific TTY fd
  }
}
 
/* ============================================================================
 * Frame Rendering Functions
 * ============================================================================ */
 
static void write_frame_to_output(const char *frame_data, bool use_direct_tty) {
  // Validate parameters
  if (!frame_data) {
    SET_ERRNO(ERROR_INVALID_PARAM, "write_frame_to_output: NULL frame_data");
    return;
  }
 
  // Calculate length safely to avoid potential segfault in strlen
  size_t frame_len = strnlen(frame_data, 1024 * 1024); // Max 1MB frame
  if (frame_len == 0) {
    SET_ERRNO(ERROR_INVALID_PARAM, "write_frame_to_output: Empty frame data");
    return;
  }
 
  if (use_direct_tty) {
    // Direct TTY for interactive use
    if (g_tty_info.fd >= 0) {
      // Always position cursor for TTY output (even in snapshot mode)
      cursor_reset(g_tty_info.fd);
      platform_write(g_tty_info.fd, frame_data, frame_len);
      // Flush terminal buffer to ensure immediate visibility
      terminal_flush(g_tty_info.fd);
    } else {
      log_error("Failed to open TTY: %s", g_tty_info.path ? g_tty_info.path : "unknown");
    }
  } else {
    // stdout for pipes/redirection/testing
    // Skip cursor reset in snapshot mode - just print raw ASCII
    if (!GET_OPTION(snapshot_mode)) {
      cursor_reset(STDOUT_FILENO);
    }
    platform_write(STDOUT_FILENO, frame_data, frame_len);
    // Flush stdout to ensure immediate visibility
    (void)fflush(stdout);
    // Only fsync if we have a valid file descriptor and not on Windows console
    if (!platform_isatty(STDOUT_FILENO)) {
      platform_fsync(STDOUT_FILENO);
    }
  }
}
 
/* ============================================================================
 * Public Interface Functions
 * ============================================================================ */
 
int display_init() {
  // Get TTY info for direct terminal access (if needed for interactive mode)
  g_tty_info = display_get_current_tty();
 
  // Determine if we should use interactive TTY output
  // For output routing: only consider stdout status (not controlling terminal)
  // Additional TTY validation: if stdout is redirected but we have a controlling TTY,
  // we can still show interactive output on the terminal
  if (g_tty_info.fd >= 0) {
    g_has_tty = platform_isatty(g_tty_info.fd) != 0; // We have a valid controlling terminal
    // Initialize ASCII output for this connection (only if we have a valid TTY)
    ascii_write_init(g_tty_info.fd, false);
  }
 
  return 0;
}
 
bool display_has_tty() {
  return g_has_tty;
}
 
void display_full_reset() {
  if (g_tty_info.fd >= 0 && !should_exit()) {
    full_terminal_reset(g_tty_info.fd);
  }
}
 
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);
  }
}
 
void display_render_frame(const char *frame_data, bool is_snapshot_frame) {
  if (!frame_data) {
    SET_ERRNO(ERROR_INVALID_PARAM, "Attempted to render NULL frame data");
    return;
  }
 
  // For terminal: print every frame until final snapshot
  // For non-terminal: only print the final snapshot frame
  if (g_has_tty || (!g_has_tty && GET_OPTION(snapshot_mode) && is_snapshot_frame)) {
    if (is_snapshot_frame) {
      // Write the final frame to the terminal as well, not just to stdout
      write_frame_to_output(frame_data, true);
    }
 
    // The real ASCII data frame write call
    write_frame_to_output(frame_data, g_has_tty && !is_snapshot_frame);
 
    if (GET_OPTION(snapshot_mode) && is_snapshot_frame) {
      // A newline at the end of the snapshot of ASCII art to end the file
      printf("\n");
    }
  }
}
 
void display_cleanup() {
  // Cleanup ASCII rendering
  ascii_write_destroy(g_tty_info.fd, true);
 
  // Close the controlling terminal if we opened it
  if (g_tty_info.owns_fd && g_tty_info.fd >= 0) {
    platform_close(g_tty_info.fd);
    g_tty_info.fd = -1;
    g_tty_info.owns_fd = false;
  }
 
  g_has_tty = false;
}