PromptShell/promptshell/format_utils.py at 78856d0cb35f71203548dd34c8e63a7af8ce5c6f · Kirti-Rathi/PromptShell · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import os
import platform

def format_text(fg, bg=None, inverted=False, bold=False):
    """Formats text for terminal output with specified foreground and background colors.

    Args:
        fg (str): The foreground color (e.g., 'red', 'green').
        bg (str, optional): The background color (e.g., 'black', 'white'). Defaults to None.
        inverted (bool, optional): If True, inverts the foreground and background colors. Defaults to False.
        bold (bool, optional): If True, makes the text bold. Defaults to False.

    Returns:
        str: The formatted text string with ANSI escape codes.
    """
    reset = "\033[0m"
    result = reset
    if bold:
        result += "\033[1m"
    if inverted:
        result += "\033[7m"
    fg_codes = {'black': '30', 'red': '31', 'green': '32', 'yellow': '33',
                'blue': '34', 'magenta': '35', 'cyan': '36', 'white': '37'}
    bg_codes = {'black': '40', 'red': '41', 'green': '42', 'yellow': '43',
                'blue': '44', 'magenta': '45', 'cyan': '46', 'white': '47'}
    result += f'\033[{fg_codes.get(fg, "37")}m'
    if bg:
        result += f'\033[{bg_codes.get(bg, "40")}m'
    return result

def reset_format():
    """Resets the text formatting to default.

    Returns:
        str: The ANSI escape code to reset formatting.
    """
    return "\033[0m"

def get_terminal_size():
    """Retrieves the current size of the terminal window.

    Returns:
        tuple: A tuple containing the number of columns and rows in the terminal.
    """
    try:
        columns, rows = os.get_terminal_size(0)
    except OSError:
        columns, rows = os.get_terminal_size(1)
    return columns, rows

def get_current_os():
    """Detects and normalizes the current operating system.

    Returns:
        str: The name of the current operating system ('windows', 'macos', or 'linux').
    """
    system = platform.system().lower()
    return 'windows' if system == 'windows' else 'macos' if system == 'darwin' else 'linux'

def get_os_specific_examples():
    """Returns OS-appropriate command examples based on the current operating system.

    Returns:
        list: A list of command examples specific to the current OS.
    """
    current_os = get_current_os()
    examples = {
        'windows': [
            '"List files" -> dir',
            '"Create directory" -> mkdir projects',
            '"Delete file" -> del example.txt',
            '"Copy file" -> copy source.txt destination.txt',
            '"Search text" -> findstr "pattern" file.txt'
        ],
        'linux': [
            '"List files" -> ls -l',
            '"Create directory" -> mkdir projects',
            '"Delete file" -> rm example.txt',
            '"Copy file" -> cp source.txt destination.txt',
            '"Search text" -> grep "pattern" file.txt'
        ],
        'macos': [
            '"List files" -> ls -lG',
            '"Create directory" -> mkdir projects',
            '"Delete file" -> rm example.txt',
            '"Copy file" -> cp source.txt destination.txt',
            '"Search text" -> grep "pattern" file.txt'
        ]
    }
    return examples[current_os]