agent.py

import webbrowser
import os
import time
import qrcode
import requests
from datetime import datetime
import sys
from art import text2art
from PIL import Image, ImageDraw, ImageFont
import random
import subprocess
from rich.markdown import Markdown
import rich
from rich.console import Console
from rich.theme import Theme
from rich.prompt import Prompt

theme = Theme(
    {"prompt": "bold #FF6E4A", "agent": "italic #00FFFF", "arrow": "blink yellow"}
)
console = Console(theme=theme)


def clear_screen():
    os.system("cls" if os.name == "nt" else "clear")


def get_terminal_width():
    try:
        return os.get_terminal_size().columns
    except OSError:
        return 80


lines = [
    "\033[34m /$$       /$$\033[0m",
    "\033[34m| $$      |__/\033[0m",
    "\033[34m| $$       /$$ /$$$$$$$   /$$$$$$   /$$$$$$   /$$$$$$ \033[0m",
    "\033[34m| $$      | $$| $$__  $$ /$$__  $$ |____  $$ /$$__  $$\033[0m",
    "\033[34m| $$      | $$| $$  | $$| $$_____/ /$$__  $$| $$\033[0m",
    "\033[34m| $$$$$$$$| $$| $$  | $$|  $$$$$$$|  $$$$$$$| $$ \033[0m",
    "\033[34m|________/|__/|__/  |__/ \\_______/ \\_______/|__/ \033[0m",
    "\033[32m            Developed by Linus Shyu \033[0m",
]


def visible_length(s):
    import re

    ansi_escape = re.compile(r"\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])")
    return len(ansi_escape.sub("", s))


max_width = max(visible_length(line) for line in lines)
terminal_width = get_terminal_width()


def display_at_position(padding):
    clear_screen()
    for line in lines:
        print(" " * padding + line)
    sys.stdout.flush()


try:
    for padding in range(0, terminal_width - max_width + 1):
        display_at_position(padding)
        time.sleep(0.05)

    center_position = (terminal_width - max_width) // 2
    for padding in range(terminal_width - max_width, center_position - 1, -1):
        display_at_position(padding)
        time.sleep(0.05)

    display_at_position(center_position)

except KeyboardInterrupt:
    clear_screen()
    for line in lines:
        print(line)


def get_user_rating(username):
    try:
        url = f"https://codeforces.com/api/user.rating?handle={username}"
        response = requests.get(url)
        data = response.json()
        if data["status"] == "OK":
            contests = data["result"]
            return [
                (
                    c["contestId"],
                    c["contestName"],
                    c["ratingUpdateTimeSeconds"],
                    c["newRating"],
                )
                for c in contests
            ]
        else:
            console.print(
                "[bright_red]Error fetching rating data:[/]",
                data.get("comment", "Unknown error"),
            )
            return None
    except Exception as e:
        console.print("[bright_red]Error:[/]", e)
        return None


def draw_ascii_chart(data, width=60, height=20):
    if not data:
        return

    ratings = [x[3] for x in data]
    dates = [datetime.fromtimestamp(x[2]).strftime("%Y-%m") for x in data]

    min_rating = min(ratings)
    max_rating = max(ratings)
    range_rating = max_rating - min_rating
    if range_rating == 0:
        range_rating = 1

    normalized = [int((r - min_rating) / range_rating * (height - 1))
                  for r in ratings]

    chart = []
    for y in range(height - 1, -1, -1):
        line = []
        for i, val in enumerate(normalized):
            if i >= width:
                break
            if val == y:
                line.append("[brigth_red]●[/]")
            elif val > y:
                line.append("[bright_green]│[/]")
            else:
                line.append("[bright_white] [/]")

        current_rating = int(min_rating + (y / (height - 1)) * range_rating)
        line.append(f" {current_rating}")
        chart.append("".join(line))

    x_labels = []
    step = max(1, len(dates) // width)
    for i in range(0, min(len(dates), width), step):
        x_labels.append(dates[i][-2:])

    chart.append(" ".join(x_labels))

    return "\n".join(chart)


def show_rating_history(username):
    rating_data = get_user_rating(username)
    if not rating_data:
        console.print(
            f"[bold red]Could not fetch rating data for {username}[/]")
        return

    console.print(
        f"[bright_green]\nCodeforces Rating History ({username}):[/]")
    console.print("[bright_yellow]=[/]" * 70)

    current_rating = rating_data[-1][3]
    max_rating = max(r[3] for r in rating_data)
    min_rating = min(r[3] for r in rating_data)
    contests_count = len(rating_data)

    console.print(f"[bright_red]Current Rating: {current_rating}[/]")
    console.print(f"[bright_green]Highest Rating: {max_rating}[/]")
    console.print(f"[britht_yellow]Lowest Rating: {min_rating}[/]")
    console.print(f"[bright_blue]Contests Participated: {contests_count}[/]")
    console.print("[bright_magenta]\nRating Chart:[/]")

    chart = draw_ascii_chart(rating_data)
    if chart:
        console.print(chart)

    console.print("[bright_red] \nLast 5 Contests: [/]")
    for contest in rating_data[-5:]:
        date = datetime.fromtimestamp(contest[2]).strftime("%Y-%m-%d")
        console.print(
            f"[bright_green]{date}: {contest[1]} (Rating: {contest[3]})[/]")


def main():
    while True:
        style = "[bright_green]What[/bright_green] [bright_yellow]you[/bright_yellow] [bright_blue]want to[/bright_blue] [agent]do[/agent] [bright_magenta]➜[bright_magenta] "
        command = Prompt.ask(style, console=console)

        if command == "home":
            webbrowser.open("https://atcoder.jp/")

        elif command == "contest":
            webbrowser.open("https://atcoder.jp/contests/")

        elif command == "rank":
            webbrowser.open("https://atcoder.jp/ranking")

        elif command == "userdata":
            username = input("Enter your username: ")
            webbrowser.open(f"https://atcoder.jp/users/{username}")

        elif command == "play":
            contest_name = input("Enter contest name: ")
            webbrowser.open(f"https://atcoder.jp/contests/{contest_name}")

        elif command == "task":
            contest_name = input("Enter contest name: ")
            task_name = input("Enter task name (a-g): ").lower()
            if task_name in "abcdefg":
                webbrowser.open(
                    f"https://atcoder.jp/contests/{contest_name}/tasks/{contest_name}_{task_name}"
                )

        elif command == "submit":
            contest_name = input("Enter contest name: ")
            webbrowser.open(
                f"https://atcoder.jp/contests/{contest_name}/submit")

        elif command == "stand":
            contest_name = input("Enter contest name: ")
            webbrowser.open(
                f"https://atcoder.jp/contests/{contest_name}/standings")

        elif command == "code":
            user = input("Enter your Mac username: ")
            try:
                os.chdir(f"/Users/{user}/Desktop")
                folder_name = input("Enter contest name for folder: ")
                os.mkdir(folder_name)
                os.chdir(f"/Users/{user}/Desktop/{folder_name}")
                os.system("touch A.cpp B.cpp C.cpp D.cpp E.cpp F.cpp G.cpp")

                cpp_code = """#include <bits/stdc++.h>

using namespace std;

typedef unsigned long long ull;
typedef unsigned int uint;
typedef long double lb;
typedef long long ll;
typedef long l;

#define endl '\\n';

void solve()
{

}

int main(void)
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int T;
    cin >> T;
    while(T--)
    {
        solve();
    }
    return 0;
}
"""
                for filename in [
                    "A.cpp",
                    "B.cpp",
                    "C.cpp",
                    "D.cpp",
                    "E.cpp",
                    "F.cpp",
                    "G.cpp",
                ]:
                    with open(filename, "w") as f:
                        f.write(cpp_code)
                os.system(f"nvim /Users/{user}/Desktop/{folder_name}")
            except Exception as e:
                print(
                    f"[bright_red]Error: {e}\nPlease check your username and try again![/]"
                )

        elif command == "clear":
            os.system("clear")

        elif command == "time":
            console.print(
                time.strftime(
                    "[bright_red]%Y-%m-%d %H:%M:%S[/]", time.localtime(
                        time.time())
                )
            )

        elif command == "about":
            print("\033[34m /$$       /$$\033[0m")
            print("\033[34m| $$      |__/\033[0m")
            print(
                "\033[34m| $$       /$$ /$$$$$$$   /$$$$$$   /$$$$$$   /$$$$$$ \033[0m"
            )
            print(
                "\033[34m| $$      | $$| $$__  $$ /$$__  $$ |____  $$ /$$__  $$\033[0m"
            )
            print(
                "\033[34m| $$      | $$| $$  | $$| $$_____/ /$$__  $$| $$\033[0m")
            print(
                "\033[34m| $$$$$$$$| $$| $$  | $$|  $$$$$$$|  $$$$$$$| $$ \033[0m")
            print(
                "\033[34m|________/|__/|__/  |__/ \\_______/ \\_______/|__/ \033[0m")

            print(
                "\033[34mWeChat QR code payment to support this project:\033[0m")
            code = "wxp://f2f08Xmtax1P6TX2gayuRlMjXvgWRIJSXz5TmEnDiWWHgLLc3W7dIqFeUqjb4g8DAPp4"
            qr = qrcode.QRCode(version=1, box_size=1, border=1)
            qr.add_data(code)
            qr.print_ascii()

            print("\033[34mDeveloped by Linus Shyu\033[0m")
            print("\033[34mSupport this project to keep it running!\033[0m")

            print(
                "GitHub Repository: \033[4mhttps://github.com/Linus-Shyu/AT-Tool\033[0m"
            )
            print(
                "Developer Bilibili: \033[4mhttps://space.bilibili.com/411591950\033[0m"
            )
            print(
                "Developer YouTube: \033[4mhttps://www.youtube.com/@LinusShyu\033[0m")
            print(" ")
            print(" ")
            print("----------------------------------------------------")
            print(
                "\033[38;2;255;215;0mBNB Chain support:https://xterminalapp.github.io/BNB/\033[0m"
            )
            print("----------------------------------------------------")
            print(" ")
            print(" ")

            art_text = text2art("> ./hack.sh", font="random")
            img = Image.new("RGB", (300, 300), "black")
            draw = ImageDraw.Draw(img)
            font = ImageFont.load_default()

            draw.text((10, 50), art_text, font=font, fill="cyan")
            img.save("hacker_art.png")
            open("./hacker_art.png")

            url = "https://api.coingecko.com/api/v3/simple/price?ids=binancecoin&vs_currencies=usd"
            response = requests.get(url)
            data = response.json()

            if "binancecoin" in data and "usd" in data["binancecoin"]:
                price = data["binancecoin"]["usd"]
                print(
                    f"\033[32mBNB Price Now: \033[38;2;255;215;0m{price:.4f} USD\033[0m"
                )
            else:
                print("\033[31mFailed to get the price of BNB\033[0m")

        elif command == "exit":
            confirm = console.input(
                "[bright_red]Are you sure to close all web pages? (y/n): [/]"
            )
            if confirm.lower() == "y":
                os.system("pkill Google Chrome")
                break
            if confirm.lower() == "n":
                break

        elif command == "help":
            console.print("[bright_yellow]\nAvailable commands:[/]")
            console.print(
                "[bright_yellow]home       - Open AtCoder home page[/]")
            console.print("[bright_yellow]contest    - Open contests page[/]")
            console.print(
                "[bright_yellow]rank       - Open global rankings[/]")
            console.print("[bright_yellow]userdata   - View user profile[/]")
            console.print(
                "[bright_yellow]play       - Open specific contest[/]")
            console.print(
                "[bright_yellow]task       - Open specific problem (a-g)[/]")
            console.print(
                "[bright_yellow]submit     - Open submission page[/]")
            console.print(
                "[bright_yellow]stand      - Open contest standings[/]")
            console.print(
                "[bright_yellow]code       - Create C++ template files for contest[/]"
            )
            console.print(
                "[bright_yellow]clear      - Clear terminal screen[/]")
            console.print("[bright_yellow]time       - Show current time[/]")
            console.print(
                "[bright_yellow]rating     - Show Codeforces rating graph[/]")
            console.print(
                "[bright_yellow]about      - Show about information[/]")
            console.print("[bright_yellow]exit       - Exit the program[/]")
            console.print(
                "[bright_yellow]help       - Show this help message\n[/]")

        elif command == "rating":
            username = console.input(
                "[bright_white]Enter Codeforces username: [/]")
            show_rating_history(username)

        elif command.strip() == "":
            continue

        elif command == "local":
            while True:
                prompt = "[prompt]Ollama[/prompt] [agent]XTerminal Agent[/agent] [arrow]➜[/arrow] "
                problem = Prompt.ask(prompt, console=console)
                process = subprocess.Popen(
                    ["ollama", "run", "llama3", problem],
                    stdout=subprocess.PIPE,
                    stderr=subprocess.PIPE,
                    text=True,
                    bufsize=1,  # 行缓冲
                    universal_newlines=True,
                )
                for line in process.stdout:
                    rich.print(Markdown(line), end="")
                if problem == "quit":
                    break

        elif command == "remote":
            os.system("./axec")
            time.sleep(1)

        else:
            console.print(
                "[cyan]Command not found. Type 'help' for available commands.[/]"
            )


if __name__ == "__main__":
    main()