#!/usr/bin/env python3

import os
import sys
import subprocess
from datetime import datetime

# Import helper functions
sys.path.append(os.path.join(os.path.expanduser("~/.dotfiles"), "bin"))
from helpers.functions import printfe, logo, _rainbow_color, COLORS

def get_last_ssh_login():
    """Get information about the last SSH login"""
    try:
        # Using lastlog2 to get the last login information
        result = subprocess.run(['lastlog2', '-u', os.environ.get("USER", "")], 
                               capture_output=True, text=True)
        if result.returncode == 0:
            lines = result.stdout.strip().split('\n')
            if len(lines) >= 2:  # Header line + data line
                # Parse the last login line - example format:
                # menno            ssh      100.99.23.98                               Mon Mar 10 19:09:43 +0100 2025
                parts = lines[1].split()
                if len(parts) >= 7 and 'ssh' in parts[1]:  # Check if it's an SSH login
                    # Extract IP address from the third column
                    ip = parts[2]
                    # Time is the rest of the line starting from position 3
                    time_str = ' '.join(parts[3:])
                    return f"{COLORS['cyan']}Last SSH login{COLORS['reset']}{COLORS['yellow']} {time_str}{COLORS['cyan']} from{COLORS['yellow']} {ip}"
        return None
    except Exception as e:
        # For debugging, you might want to print the exception
        # print(f"Error getting SSH login: {str(e)}")
        return None

def check_dotfiles_status():
    """Check if the dotfiles repository is dirty"""
    dotfiles_path = os.environ.get("DOTFILES_PATH", os.path.expanduser("~/.dotfiles"))
    try:
        if not os.path.isdir(os.path.join(dotfiles_path, ".git")):
            return None
            
        # Check for git status details
        status = {
            'is_dirty': False,
            'untracked': 0,
            'modified': 0,
            'staged': 0,
            'commit_hash': '',
            'unpushed': 0
        }
        
        # Get status of files
        result = subprocess.run(['git', 'status', '--porcelain'], 
                              cwd=dotfiles_path,
                              capture_output=True, text=True)
        
        if result.stdout.strip():
            status['is_dirty'] = True
            for line in result.stdout.splitlines():
                if line.startswith('??'):
                    status['untracked'] += 1
                if line.startswith(' M') or line.startswith('MM'):
                    status['modified'] += 1
                if line.startswith('M ') or line.startswith('A '):
                    status['staged'] += 1
        
        # Get current commit hash
        result = subprocess.run(['git', 'rev-parse', '--short', 'HEAD'],
                              cwd=dotfiles_path,
                              capture_output=True, text=True)
        if result.returncode == 0:
            status['commit_hash'] = result.stdout.strip()
        
        # Count unpushed commits
        # Fix: Remove capture_output and set stdout explicitly
        result = subprocess.run(['git', 'log', '--oneline', '@{u}..'],
                              cwd=dotfiles_path,
                              stdout=subprocess.PIPE,
                              stderr=subprocess.DEVNULL,
                              text=True)
        if result.returncode == 0:
            status['unpushed'] = len(result.stdout.splitlines())
        
        return status
    except Exception as e:
        print(f"Error checking dotfiles status: {str(e)}")
        return None

def get_condensed_status():
    """Generate a condensed status line for trash and git status"""
    status_parts = []
    
    # Check trash status
    trash_path = os.path.expanduser("~/.local/share/Trash/files")
    try:
        if os.path.exists(trash_path):
            items = os.listdir(trash_path)
            count = len(items)
            if count > 0:
                status_parts.append(f"[!] {count} file(s) in trash")
    except Exception:
        pass
    
    # Check dotfiles status
    dotfiles_status = check_dotfiles_status()
    if dotfiles_status is not None:
        if dotfiles_status['is_dirty']:
            status_parts.append(f"{COLORS['yellow']}dotfiles is dirty{COLORS['reset']}")
            status_parts.append(f"{COLORS['red']}[{dotfiles_status['untracked']}] untracked{COLORS['reset']}")
            status_parts.append(f"{COLORS['yellow']}[{dotfiles_status['modified']}] modified{COLORS['reset']}")
            status_parts.append(f"{COLORS['green']}[{dotfiles_status['staged']}] staged{COLORS['reset']}")
        
        if dotfiles_status['commit_hash']:
            status_parts.append(f"[{COLORS['blue']}{dotfiles_status['commit_hash']}{COLORS['reset']}]")
        
        if dotfiles_status['unpushed'] > 0:
            status_parts.append(f"[!] You have {dotfiles_status['unpushed']} commit(s) to push")
    else:
        status_parts.append("Unable to check dotfiles status")
    
    if status_parts:
        return " - ".join(status_parts)
    return None

def welcome():
    """Display welcome message with hostname and username"""
    print()
    
    # Get hostname and username
    hostname = os.uname().nodename
    username = os.environ.get("USER", os.environ.get("USERNAME", "user"))
    
    # Get SSH login info first
    ssh_login = get_last_ssh_login()
    
    print(f"{COLORS['cyan']}You're logged in on [", end="")
    print(_rainbow_color(hostname), end="")
    print(f"{COLORS['cyan']}] as [", end="")
    print(_rainbow_color(username), end="")
    print(f"{COLORS['cyan']}]{COLORS['reset']}")
    
    # Display last SSH login info if available
    if ssh_login:
        print(f"{ssh_login}{COLORS['reset']}")
    
    # Display condensed status line
    condensed_status = get_condensed_status()
    if condensed_status:
        print(f"{COLORS['yellow']}{condensed_status}{COLORS['reset']}")

def main():
    logo(continue_after=True)
    welcome()
    return 0

if __name__ == "__main__":
    sys.exit(main())