ai-ebash 0.1.2__py3-none-any.whl

ai_ebash-0.1.2.dist-info/METADATA

@@ -0,0 +1,55 @@
+Metadata-Version: 2.4
+Name: ai-ebash
+Version: 0.1.2
+Summary: Console utility for integrating artificial intelligence into a Linux terminal.
+Author: Andrey Bochkarev
+Author-email: andrey.bch.1976@gmail.com
+License: MIT
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: certifi==2025.8.3
+Requires-Dist: charset-normalizer==3.4.3
+Requires-Dist: idna==3.10
+Requires-Dist: markdown-it-py==3.0.0
+Requires-Dist: mdurl==0.1.2
+Requires-Dist: platformdirs==4.4.0
+Requires-Dist: Pygments==2.19.2
+Requires-Dist: requests==2.32.5
+Requires-Dist: rich==14.1.0
+Requires-Dist: urllib3==2.5.0
+
+# Ai-bash!
+Console utility for integrating artificial intelligence into a Linux terminal. Allows you to ask an AI question and execute the scripts and commands suggested by the AI in the terminal. It will be useful for novice Linux administrators. 
+  
+The project is in the pre-alpha stage. In case of problems with the installation or operation of the Ai-bash utility, please contact me.
+
+## Setup
+
+### Ubuntu/Debian
+Download `.deb` из [Releases](https://github.com/yourname/ai-bash/releases) and install:
+
+```bash
+sudo dpkg -i ai-bash_0.1.0-1_all.deb
+sudo apt -f install  # it will tighten up the missing dependencies
+```
+
+
+
+### Run
+```bash
+ai [-run] Your request to the AI
+```
+
+### Example
+```bash
+ai Write a script in bash that outputs a list of files in the current directory.
+```
+or
+```bash
+ai -run Write a script in bash that outputs a list of files in the current directory.
+```
+
+## Remove
+```bash
+sudo apt remove ai-bash
+```

ai_ebash-0.1.2.dist-info/RECORD

@@ -0,0 +1,11 @@
+aiebash/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aiebash/__main__.py,sha256=DHAyTjr2VmSnXP9VGLzUOVVJGTgErBUtDjWkizC2rWU,3720
+aiebash/api_client.py,sha256=-hcgY3CP0lsV9Jct_-HQ6WpDosDChwalJjy4aFQ4CIM,2004
+aiebash/block_runner.py,sha256=JNcdBklJmrlO7ZX7xXiO0fUGqrcNJiRqdwM-yr3OP3k,1599
+aiebash/formatter_text.py,sha256=NRz_pCpakZ9k8477_4Gx1scDXrEbbOaOi-U7WRp6ZWo,847
+aiebash/settings.py,sha256=Ml9_ixXl-DVpFYfCygMAhGSH-1luTxN22YYz1mHlsfw,2671
+ai_ebash-0.1.2.dist-info/METADATA,sha256=ZpxRWic7mwoJDXe_2eelWAnENkSFDdBKKD5eHaG5iAw,1595
+ai_ebash-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ai_ebash-0.1.2.dist-info/entry_points.txt,sha256=bstvFHHX76KHqYpjM8FPoRVvNbQprST74CI30Mr5TIM,45
+ai_ebash-0.1.2.dist-info/top_level.txt,sha256=JJ-bWdhWlFCfCQGLhmqzN4_bNO0pfVPRsdbx8o9A9AA,8
+ai_ebash-0.1.2.dist-info/RECORD,,

ai_ebash-0.1.2.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

ai_ebash-0.1.2.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ai = aiebash.__main__:main

ai_ebash-0.1.2.dist-info/top_level.txt

@@ -0,0 +1 @@
+aiebash

aiebash/__main__.py

@@ -0,0 +1,100 @@
+#!/usr/bin/env python3
+import sys
+from pathlib import Path
+
+# Убедимся, что parent (src) в sys.path, чтобы можно было import aibash.*
+# Это нужно, если вы запускаете файл напрямую: python src/aibash/__main__.py
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+
+from rich.markdown import Markdown
+from rich.console import Console
+from rich.rule import Rule
+import threading
+import time
+
+
+from aiebash.api_client import send_prompt
+from aiebash.formatter_text import annotate_bash_blocks
+from aiebash.block_runner import run_code_selection  # добавлен импорт
+
+from aiebash.settings import settings as user_settings
+
+# нормализуем DEBUG (возможные варианты: True/False, "true"/"False", "1"/"0")
+_raw_debug = user_settings.get("DEBUG")
+if isinstance(_raw_debug, bool):
+    DEBUG = _raw_debug
+else:
+    DEBUG = str(_raw_debug).strip().lower() in ("1", "true", "yes", "on")
+CONTEXT = user_settings.get("CONTEXT")
+MODEL = user_settings.get("MODEL")
+API_URL = user_settings.get("API_URL")
+API_KEY = user_settings.get("API_KEY")
+
+# Флаг для остановки потока (заменён на Event)
+import threading
+stop_event = threading.Event()
+
+def run_progress():
+    console = Console()
+    with console.status("[bold green]Ai печатает...[/bold green]", spinner="dots") as status:
+         while not stop_event.is_set():
+            time.sleep(0.1)
+
+def main():
+    # Если нет аргументов, выводим подсказку по использованию
+    if len(sys.argv) < 2:
+        print("Использование: ai [-run] ваш запрос к ИИ без кавычек")
+        sys.exit(0)
+
+    console = Console()
+
+    # Проверяем ключ -run
+    run_mode = False
+    args = sys.argv[1:]
+    if "-run" in args:
+        run_mode = True
+        args.remove("-run")
+
+    # Собираем текст запроса из оставшихся аргументов
+    prompt = " ".join(args)
+
+    try:
+        # Запуск прогресс-бара в отдельном потоке
+        progress_thread = threading.Thread(target=run_progress)
+        progress_thread.start()
+
+        # Получаем ответ от API через новый интерфейс
+        answer = send_prompt(prompt, MODEL, API_URL, API_KEY, CONTEXT)
+
+        # Сигнализируем потоку прогресса остановиться
+        stop_event.set()
+        progress_thread.join()  # Ждём завершения потока
+
+        # В режиме DEBUG выводим исходную (неформатированную) версию ответа
+        if DEBUG:
+            print("=== RAW RESPONSE (from send_prompt) ===")
+            print(answer)
+            print("=== /RAW RESPONSE ===\n")
+
+        # Размечаем bash-блоки и получаем список кодов
+        annotated_answer, code_blocks = annotate_bash_blocks(answer)
+        
+
+        # Если включён режим выполнения и есть блоки кода — предлагаем выбрать
+        if run_mode and code_blocks:
+            console.print(Markdown(annotated_answer))
+            run_code_selection(console, code_blocks)
+        else:
+            console.print(Markdown(answer))
+        
+        console.print(Rule("", style="green"))
+
+    except Exception as e:
+        # Прочие ошибки (сеть, JSON, и т.д.)
+        print("Ошибка:", e)
+
+
+
+if __name__ == "__main__":
+    main()

aiebash/api_client.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python3
+"""
+api_client — небольшой интерфейс для отправки chat-запросов к API.
+Экспортирует send_chat_request (работает с messages) и удобный send_prompt.
+"""
+import json
+import requests
+import typing as t
+
+
+def send_chat_request(messages: t.List[dict], model , api_url, api_key = None, timeout: int = 60) -> dict:
+    """
+    Отправляет chat-style запрос (messages) к API и возвращает распарсенный JSON-ответ.
+    messages — список словарей вида {"role": "user|system|assistant", "content": "..."}.
+    """
+    if not api_url:
+        raise RuntimeError("API_URL не задан в config.ini")
+
+    payload = {
+        "model": model,
+        "messages": messages,
+    }
+
+    headers = {"Content-Type": "application/json"}
+    if api_key:
+        headers["Authorization"] = f"Bearer {api_key}"
+
+    resp = requests.post(api_url, headers=headers, data=json.dumps(payload), timeout=timeout)
+    resp.raise_for_status()
+    data = resp.json()
+
+    return data
+
+
+def send_prompt(prompt: str, model, api_url,api_key = None, system_context: str = "", timeout: int = 30) -> str:
+    """
+    Удобная обёртка: формирует messages из system_context + user prompt,
+    отправляет запрос и возвращает текст ответа ассистента.
+    """
+    messages = []
+    if system_context:
+        messages.append({"role": "system", "content": system_context})
+    messages.append({"role": "user", "content": prompt})
+
+    data = send_chat_request(messages, model, api_url=api_url, api_key=api_key, timeout=timeout)
+    # ожидаемый формат: choices -> [ { message: { content: "..." } } ]
+    if "choices" in data and data["choices"]:
+        return data["choices"][0]["message"]["content"]
+    raise RuntimeError("Unexpected API response format")

aiebash/block_runner.py

@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+import subprocess
+from rich.console import Console
+from rich.markdown import Markdown
+
+
+def run_code_selection(console: Console, code_blocks: list):
+    """
+    Интерактивный цикл: спрашивает номер блока, выводит его содержимое и при выборе выполняет его.
+    Завершается при вводе 0, q, exit или при Ctrl+C / EOF.
+    """
+    try:
+        while True:
+            choice = console.input("[blue]\nВведите номер блока кода для запуска (0 — выход): [/blue]").strip()
+            if choice.lower() in ("0", "q", "exit"):
+                console.print("Выход.")
+                break
+            if not choice.isdigit():
+                console.print("[red]Введите число или 0 для выхода.[/red]")
+                continue
+            idx = int(choice)
+            if idx < 1 or idx > len(code_blocks):
+                console.print(f"[red]Неверный номер: у вас {len(code_blocks)} блоков. Попробуйте снова.[/red]")
+                continue
+            console.print(f"\n>>> Выполняем блок #{idx}:\n", style="blue")
+            console.print(code_blocks[idx - 1])
+            # Выполнение — риск: выполняется произвольный код из ответа ИИ
+            subprocess.run(code_blocks[idx - 1], shell=True)
+    except (EOFError, KeyboardInterrupt):
+        console.print("\nВыход.")

aiebash/formatter_text.py

@@ -0,0 +1,23 @@
+import re
+
+def annotate_bash_blocks(md_text):
+    """
+    Находит fenced code blocks с языком bash, собирает их содержимое в список
+    и добавляет над каждым блоком строку-метку вида "**Блок кода bash [#{counter}]**".
+    Возвращает (annotated_md, list_of_code_strings).
+    """
+    code_blocks = []
+    counter = 0
+
+    pattern = re.compile(r"```bash[^\n]*\n(.*?)```", re.DOTALL | re.IGNORECASE)
+
+    def repl(m):
+        nonlocal counter
+        counter += 1
+        code = m.group(1).rstrip("\n")
+        code_blocks.append(code)
+        label = f"\n**Блок кода bash [#{counter}]**\n"
+        return label + "```bash\n" + code + "\n```"
+
+    annotated = pattern.sub(repl, md_text)
+    return annotated, code_blocks

aiebash/settings.py

@@ -0,0 +1,65 @@
+
+from pathlib import Path
+import configparser
+from platformdirs import user_config_dir
+
+APP_NAME = "ai-bash"
+
+# Дефолтные значения хранятся в словаре
+DEFAULTS = {
+    "API_URL": "https://openai-proxy.andrey-bch-1976.workers.dev/v1/chat/completions", # Прокси для OpenAI API
+    "API_KEY": "",  # Ключ API (если нужен) 
+    "MODEL": "gpt-4o-mini", # Модель по умолчанию
+    "CONTEXT": ( # Контекст по умолчанию
+        "Ты профессиональный системный администратор, помощник по Linux. "
+        "Пользователь - новичок в Linux, помоги ему. "
+        "Система Ubuntu, оболочка $SHELL. "
+        "При ответе учитывай, что пользователь работает в терминале Bash."
+        "Отвечай всегда на русском языке. "
+        "Ты разговариваешь с пользователем в терминале. "
+    ), 
+    "DEBUG": False  # Включить отладочную информацию
+}
+
+# Где хранится INI-файл
+# Ubuntu → ~/.config/ai-bash/config.ini
+# Windows → C:\Users\<user>\AppData\Local\ai-bash\ai-bash\config.ini
+
+class Settings:
+    def __init__(self):
+        # Кроссплатформенная директория конфигов
+        self.config_dir = Path(user_config_dir(APP_NAME))
+        self.config_dir.mkdir(parents=True, exist_ok=True)
+
+        self.config_file = self.config_dir / "config.ini"
+        self.config = configparser.ConfigParser()
+
+        # Если файл есть — читаем, иначе создаём с дефолтами
+        if self.config_file.exists():
+            self.config.read(self.config_file)
+            if "settings" not in self.config:
+                self.config["settings"] = DEFAULTS
+                self._save()
+        else:
+            self.config["settings"] = DEFAULTS
+            self._save()
+            print(f"[INFO] Создан новый конфиг: {self.config_file}")
+
+    def _save(self):
+        with open(self.config_file, "w") as f:
+            self.config.write(f)
+
+    # Получение значений
+    def get(self, key):
+        return self.config["settings"].get(key, DEFAULTS.get(key))
+
+    # Изменение значений
+    def set(self, key, value):
+        self.config["settings"][key] = str(value)
+        self._save()
+
+
+# Глобальный объект настроек
+settings = Settings()
+
+