#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# Fenrir TTY screen reader
# By Chrys, Storm Dragon, and contributors.

import re

from fenrirscreenreader.core import debug


class TableManager:
    def __init__(self):
        self.headLine = ""
        self.defaultSeparators = ["+", ";", "|", " "]
        self.noOfHeadLineColumns = 0
        self.headColumnSep = ""
        self.rowColumnSep = ""
        self.headerRow = []
        self.tableMode = False
        self.currentColumn = 0
        self.currentRow = 0

    def initialize(self, environment):
        self.env = environment

    def shutdown(self):
        pass
    
    def is_table_mode(self):
        return self.tableMode
    
    def set_table_mode(self, active):
        self.tableMode = active
        if not active:
            self.clear_header_row()
        else:
            # Initialize current position when entering table mode
            self.sync_table_position_from_cursor()
    
    def set_header_row_from_cursor(self):
        """Set header row from current cursor position"""
        if not self.env["runtime"]["CursorManager"].is_review_mode():
            return False
        
        cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
        if not cursor_pos:
            return False
        
        # Get the line text at cursor position
        line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
        if not line_text:
            return False
        
        # Parse the line into columns
        self.headerRow = self.parse_line_into_columns(line_text)
        return len(self.headerRow) > 0
    
    def clear_header_row(self):
        """Clear the stored header row"""
        self.headerRow = []
    
    def has_header_row(self):
        """Check if header row is set"""
        return len(self.headerRow) > 0
    
    def get_column_header(self, column_index):
        """Get header for a specific column (0-based)"""
        if column_index < 0:
            return f"Column {column_index + 1}"
        
        if len(self.headerRow) == 0:
            return f"Column {column_index + 1}"
        
        if column_index >= len(self.headerRow):
            return f"Column {column_index + 1}"
        
        header = self.headerRow[column_index].strip()
        if not header:  # If header is empty
            return f"Column {column_index + 1}"
        
        return header
    
    def get_column_count(self):
        """Get number of columns based on header row"""
        return len(self.headerRow)
    
    def parse_line_into_columns(self, line_text):
        """Parse a line into columns using various separators"""
        if not line_text:
            return []
        
        # Try different separators in order of preference
        separators = [',', '|', ';', '\t']
        
        for sep in separators:
            if sep in line_text:
                columns = line_text.split(sep)
                if len(columns) > 1:
                    return columns
        
        # If no clear separator, try to detect aligned columns
        return self.detect_aligned_columns(line_text)
    
    def detect_aligned_columns(self, line_text):
        """Detect columns in space-aligned text"""
        if not line_text:
            return []
        
        # Split on multiple spaces (2 or more)
        parts = re.split(r'  +', line_text)
        if len(parts) > 1:
            return parts
        
        # Fallback: treat as single column
        return [line_text]
    
    def get_cell_content(self, line_text, column_index):
        """Get content of a specific cell in a line"""
        columns = self.parse_line_into_columns(line_text)
        if column_index < 0 or column_index >= len(columns):
            return ""
        return columns[column_index].strip()
    
    def get_table_cell_info(self, cursor_pos):
        """Get table cell information for speech output"""
        line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
        if not line_text:
            return None
        
        columns = self.parse_line_into_columns(line_text)
        if not columns:
            return None
        
        # Find which column we're in based on cursor x position
        column_index = self.get_column_index_from_x_position(line_text, cursor_pos["x"])
        
        if column_index < 0 or column_index >= len(columns):
            return None
        
        # Use header if available, otherwise generic column name
        if self.has_header_row():
            column_header = self.get_column_header(column_index)
        else:
            column_header = f"Column {column_index + 1}"
        
        return {
            'column_index': column_index,
            'column_header': column_header,
            'cell_content': columns[column_index].strip(),
            'total_columns': len(columns),
            'row_number': cursor_pos["y"] + 1
        }
    
    def get_column_index_from_x_position(self, line_text, x_pos):
        """Determine which column an x position falls into"""
        columns = self.parse_line_into_columns(line_text)
        if not columns:
            return 0
        
        # Handle CSV/delimited text
        if ',' in line_text or '|' in line_text or ';' in line_text or '\t' in line_text:
            current_pos = 0
            for i, column in enumerate(columns):
                column_end = current_pos + len(column)
                if current_pos <= x_pos <= column_end:
                    return i
                current_pos = column_end + 1  # +1 for separator
        else:
            # Handle space-aligned text - find the column by position ranges
            current_pos = 0
            for i, column in enumerate(columns):
                # Find where this column starts in the original text
                if i == 0:
                    column_start = 0
                else:
                    # Find the column by searching from current position
                    column_start = line_text.find(column.strip(), current_pos)
                    if column_start == -1:
                        column_start = current_pos
                
                column_end = column_start + len(column.strip())
                if column_start <= x_pos <= column_end:
                    return i
                current_pos = column_end
        
        # If past the end, return last column
        return len(columns) - 1
    
    def sync_table_position_from_cursor(self):
        """Sync internal table position from current cursor position"""
        if not self.env["runtime"]["CursorManager"].is_review_mode():
            return
        
        cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
        if cursor_pos:
            old_column = self.currentColumn
            self.currentRow = cursor_pos["y"]
            line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
            if line_text:
                self.currentColumn = self.get_column_index_from_x_position(line_text, cursor_pos["x"])
            else:
                self.currentColumn = 0
            
            self.env["runtime"]["DebugManager"].write_debug_out(
                f"TableManager sync: old_column={old_column}, new_column={self.currentColumn}, cursor_x={cursor_pos['x']}, cursor_y={cursor_pos['y']}",
                debug.DebugLevel.INFO
            )
    
    def move_to_next_column(self):
        """Move to next column in table"""
        self.env["runtime"]["DebugManager"].write_debug_out(
            f"TableManager move_to_next_column: currentColumn={self.currentColumn}",
            debug.DebugLevel.INFO
        )
        
        if not self.env["runtime"]["CursorManager"].is_review_mode():
            return None
        
        cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
        if not cursor_pos:
            return None
        
        line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
        if not line_text:
            return None
        
        columns = self.parse_line_into_columns(line_text)
        if not columns:
            return None
        
        self.env["runtime"]["DebugManager"].write_debug_out(
            f"TableManager: line_text='{line_text}', columns={columns}, currentColumn={self.currentColumn}",
            debug.DebugLevel.INFO
        )
        
        # Don't sync from cursor position - maintain our own tracking
        # Check if we're already at the last column
        if self.currentColumn >= len(columns) - 1:
            # At end of line - return special indicator but keep position
            return {"at_end": True, "current_info": self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)}
        
        # Move to next column
        self.currentColumn += 1
        
        self.env["runtime"]["DebugManager"].write_debug_out(
            f"TableManager: moved to column {self.currentColumn}",
            debug.DebugLevel.INFO
        )
        
        # Return info for the new column without moving cursor position
        return self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)
    
    def move_to_prev_column(self):
        """Move to previous column in table"""
        if not self.env["runtime"]["CursorManager"].is_review_mode():
            return None
        
        cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
        if not cursor_pos:
            return None
        
        line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
        if not line_text:
            return None
        
        columns = self.parse_line_into_columns(line_text)
        if not columns:
            return None
        
        # Don't sync from cursor position - maintain our own tracking
        # Check if we're already at the first column
        if self.currentColumn <= 0:
            # At beginning of line - return special indicator but keep position
            return {"at_start": True, "current_info": self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)}
        
        # Move to previous column
        self.currentColumn -= 1
        
        # Return info for the new column without moving cursor position
        return self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)
    
    def get_current_table_cell_info(self):
        """Get current table cell info using internal position tracking"""
        if not self.env["runtime"]["CursorManager"].is_review_mode():
            return None
        
        cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
        if not cursor_pos:
            return None
        
        return self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)
    
    def get_table_cell_info_by_indices(self, row, column_index):
        """Get table cell info for specific row and column indices"""
        line_text = self.env["runtime"]["ScreenManager"].get_line_text(row)
        if not line_text:
            return None
        
        columns = self.parse_line_into_columns(line_text)
        if not columns or column_index < 0 or column_index >= len(columns):
            return None
        
        # Always get column header (fallback to generic if needed)
        column_header = self.get_column_header(column_index)
        
        cell_content = columns[column_index].strip()
        if not cell_content:  # Handle empty cells
            cell_content = "blank"
        
        return {
            'column_index': column_index,
            'column_header': column_header,
            'cell_content': cell_content,
            'total_columns': len(columns),
            'row_number': row + 1
        }
    
    def get_column_start_position(self, line_text, column_index):
        """Get the starting x position of a specific column"""
        columns = self.parse_line_into_columns(line_text)
        if not columns or column_index < 0 or column_index >= len(columns):
            return 0
        
        # For CSV/delimited text - find the actual position in the original line
        if ',' in line_text or '|' in line_text or ';' in line_text or '\t' in line_text:
            # Determine the separator being used
            separator = ','
            if '|' in line_text and line_text.count('|') > line_text.count(','):
                separator = '|'
            elif ';' in line_text and line_text.count(';') > line_text.count(','):
                separator = ';'
            elif '\t' in line_text:
                separator = '\t'
            
            # Find the position by splitting and calculating
            if column_index == 0:
                return 0
            
            # Count characters up to the target column
            parts = line_text.split(separator)
            position = 0
            for i in range(column_index):
                if i < len(parts):
                    position += len(parts[i]) + 1  # +1 for separator
            return position
        else:
            # For space-aligned text, find the actual position of the column
            if column_index == 0:
                return 0
            
            # Find the column text in the line
            target_text = columns[column_index].strip()
            search_start = 0
            for i in range(column_index):
                prev_text = columns[i].strip()
                found_pos = line_text.find(prev_text, search_start)
                if found_pos != -1:
                    search_start = found_pos + len(prev_text)
            
            column_pos = line_text.find(target_text, search_start)
            return column_pos if column_pos != -1 else search_start

    def is_cursor_within_current_cell(self, cursor_x, cursor_y):
        """Check if the given cursor position is within the current table cell"""
        if not self.is_table_mode():
            return False
        
        line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_y)
        if not line_text:
            return False
        
        columns = self.parse_line_into_columns(line_text)
        if not columns or self.currentColumn < 0 or self.currentColumn >= len(columns):
            return False
        
        # Get the bounds of the current column
        column_start = self.get_column_start_position(line_text, self.currentColumn)
        column_text = columns[self.currentColumn]
        column_end = column_start + len(column_text)
        
        # Check if cursor is within the column bounds
        return column_start <= cursor_x < column_end

    def move_to_first_cell(self):
        """Move to first cell in current row"""
        if not self.env["runtime"]["CursorManager"].is_review_mode():
            return None
        
        cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
        if not cursor_pos:
            return None
        
        line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
        if not line_text:
            return None
        
        columns = self.parse_line_into_columns(line_text)
        if not columns:
            return None
        
        # Set current column to first column
        self.currentColumn = 0
        
        # Return info for the first column
        return self.get_table_cell_info_by_indices(cursor_pos["y"], 0)
    
    def move_to_last_cell(self):
        """Move to last cell in current row"""
        if not self.env["runtime"]["CursorManager"].is_review_mode():
            return None
        
        cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
        if not cursor_pos:
            return None
        
        line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
        if not line_text:
            return None
        
        columns = self.parse_line_into_columns(line_text)
        if not columns:
            return None
        
        # Set current column to last column
        self.currentColumn = len(columns) - 1
        
        # Return info for the last column
        return self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)
    
    def move_to_first_char_in_cell(self):
        """Move to first character in current cell"""
        if not self.env["runtime"]["CursorManager"].is_review_mode():
            return None
        
        cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
        if not cursor_pos:
            return None
        
        line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
        if not line_text:
            return None
        
        columns = self.parse_line_into_columns(line_text)
        if not columns or self.currentColumn < 0 or self.currentColumn >= len(columns):
            return None
        
        # Get column start position
        column_start = self.get_column_start_position(line_text, self.currentColumn)
        
        # Find first non-space character in the column
        column_text = columns[self.currentColumn]
        first_char_offset = len(column_text) - len(column_text.lstrip())
        
        # Set cursor position to first character in cell
        new_x = column_start + first_char_offset
        self.env["runtime"]["CursorManager"].set_review_cursor_position(new_x, cursor_pos["y"])
        
        # Get the character at the new position
        from fenrirscreenreader.utils import char_utils
        (
            self.env["screen"]["newCursorReview"]["x"],
            self.env["screen"]["newCursorReview"]["y"],
            curr_char,
        ) = char_utils.get_current_char(
            new_x,
            cursor_pos["y"],
            self.env["screen"]["new_content_text"],
        )
        
        return {
            'cell_content': column_text.strip(),
            'column_header': self.get_column_header(self.currentColumn),
            'character': curr_char,
            'position': 'first'
        }
    
    def move_to_last_char_in_cell(self):
        """Move to last character in current cell"""
        if not self.env["runtime"]["CursorManager"].is_review_mode():
            return None
        
        cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
        if not cursor_pos:
            return None
        
        line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
        if not line_text:
            return None
        
        columns = self.parse_line_into_columns(line_text)
        if not columns or self.currentColumn < 0 or self.currentColumn >= len(columns):
            return None
        
        # Get column start position
        column_start = self.get_column_start_position(line_text, self.currentColumn)
        column_text = columns[self.currentColumn]
        
        # Find last non-space character in the column
        trimmed_text = column_text.rstrip()
        if not trimmed_text:
            # If empty cell, go to column start
            new_x = column_start
        else:
            # Find the position of the last character
            new_x = column_start + len(trimmed_text) - 1
        
        # Set cursor position to last character in cell
        self.env["runtime"]["CursorManager"].set_review_cursor_position(new_x, cursor_pos["y"])
        
        # Get the character at the new position
        from fenrirscreenreader.utils import char_utils
        (
            self.env["screen"]["newCursorReview"]["x"],
            self.env["screen"]["newCursorReview"]["y"],
            curr_char,
        ) = char_utils.get_current_char(
            new_x,
            cursor_pos["y"],
            self.env["screen"]["new_content_text"],
        )
        
        return {
            'cell_content': column_text.strip(),
            'column_header': self.get_column_header(self.currentColumn),
            'character': curr_char,
            'position': 'last'
        }

    def reset_table_mode(self):
        self.set_head_line()

    def set_head_column_sep(self, columnSep=""):
        self.headColumnSep = columnSep
        if columnSep == "":
            self.noOfHeadLineColumns = 0
        else:
            self.coun_no_of_head_columns()

    def coun_no_of_head_columns(self):
        if self.headLine and self.headColumnSep:
            self.noOfHeadLineColumns = len(self.headLine.split(self.headColumnSep))

    def search_for_head_column_sep(self, headLine):
        """Find the most likely column separator in a header line"""
        separators = [',', '|', ';', '\t']
        
        for sep in separators:
            if sep in headLine:
                return sep
        
        # Check for multiple spaces (aligned columns)
        if '  ' in headLine:
            return '  '
        
        return ""

    def set_row_column_sep(self, columnSep=""):
        self.rowColumnSep = columnSep

    def set_head_line(self, headLine=""):
        self.set_head_column_sep()
        self.set_row_column_sep()
        if headLine != "":
            sep = self.search_for_head_column_sep(headLine)
            if sep != "":
                self.headLine = headLine
                self.set_head_column_sep(sep)