| |
| |
| |
| |
| |
| |
|
|
| from . import get_help |
|
|
| __doc__ = get_help("help_beautify") |
|
|
|
|
| import os |
| import random |
|
|
| from telethon.utils import get_display_name |
| from urllib.parse import urlencode |
| from . import Carbon, ultroid_cmd, get_string, inline_mention |
| from secrets import token_hex |
|
|
| _colorspath = "resources/colorlist.txt" |
|
|
| if os.path.exists(_colorspath): |
| with open(_colorspath, "r") as f: |
| all_col = f.read().split() |
| else: |
| all_col = [] |
|
|
|
|
| @ultroid_cmd( |
| pattern="(rc|c)arbon", |
| ) |
| async def cr_bn(event): |
| xxxx = await event.eor(get_string("com_1")) |
| te = event.pattern_match.group(1) |
| col = random.choice(all_col) if te[0] == "r" else "White" |
| if event.reply_to_msg_id: |
| temp = await event.get_reply_message() |
| if temp.media: |
| b = await event.client.download_media(temp) |
| with open(b) as a: |
| code = a.read() |
| os.remove(b) |
| else: |
| code = temp.message |
| else: |
| try: |
| code = event.text.split(" ", maxsplit=1)[1] |
| except IndexError: |
| return await xxxx.eor(get_string("carbon_2")) |
| xx = await Carbon(code=code, file_name="ultroid_carbon", backgroundColor=col) |
| if isinstance(xx, dict): |
| await xxxx.edit(f"`{xx}`") |
| return |
| await xxxx.delete() |
| await event.reply( |
| f"Carbonised by {inline_mention(event.sender)}", |
| file=xx, |
| ) |
|
|
|
|
| @ultroid_cmd( |
| pattern="ccarbon( (.*)|$)", |
| ) |
| async def crbn(event): |
| match = event.pattern_match.group(1).strip() |
| if not match: |
| return await event.eor(get_string("carbon_3")) |
| msg = await event.eor(get_string("com_1")) |
| if event.reply_to_msg_id: |
| temp = await event.get_reply_message() |
| if temp.media: |
| b = await event.client.download_media(temp) |
| with open(b) as a: |
| code = a.read() |
| os.remove(b) |
| else: |
| code = temp.message |
| else: |
| try: |
| match = match.split(" ", maxsplit=1) |
| code = match[1] |
| match = match[0] |
| except IndexError: |
| return await msg.eor(get_string("carbon_2")) |
| xx = await Carbon(code=code, backgroundColor=match) |
| await msg.delete() |
| await event.reply( |
| f"Carbonised by {inline_mention(event.sender)}", |
| file=xx, |
| ) |
|
|
|
|
| RaySoTheme = [ |
| "meadow", |
| "breeze", |
| "raindrop", |
| "candy", |
| "crimson", |
| "falcon", |
| "sunset", |
| "noir", |
| "midnight", |
| "bitmap", |
| "ice", |
| "sand", |
| "forest", |
| "mono", |
| ] |
|
|
|
|
| @ultroid_cmd(pattern="rayso") |
| async def pass_on(ult): |
| try: |
| from playwright.async_api import async_playwright |
| except ImportError: |
| await ult.eor( |
| "`playwright` is not installed!\nPlease install it to use this command.." |
| ) |
| return |
|
|
| proc = await ult.eor(get_string("com_1")) |
| spli = ult.text.split() |
| theme, dark, title, text = None, True, get_display_name(ult.chat), None |
| if len(spli) > 1: |
| if spli[1] in RaySoTheme: |
| theme = spli[1] |
| if len(spli) > 2: |
| text = " ".join(spli[2:]) |
| else: |
| text = " ".join(spli[1:]) |
| if ult.is_reply: |
| try: |
| msg = await ult.get_reply_message() |
| text = msg.message if not text else text |
| title = get_display_name(msg.sender) |
| if not theme and spli[1] in RaySoTheme: |
| theme = spli[1] |
| except Exception as sam: |
| LOGS.exception(sam) |
| if not text: |
| await proc.eor("No text to beautify!") |
| return |
| if not theme: |
| theme = random.choice(RaySoTheme) |
| cleaned_text = "\n".join([line.strip() for line in text.splitlines()]) |
| name = token_hex(8) + ".png" |
| data = {"darkMode": dark, "theme": theme, "title": title} |
| url = f"https://ray.so/#{urlencode(data)}" |
| async with async_playwright() as play: |
| try: |
| browser = await play.chromium.launch() |
| page = await browser.new_page() |
| await page.goto(url) |
| await page.wait_for_load_state("networkidle") |
| try: |
| await page.wait_for_selector( |
| "div[class*='Editor_editor__']", timeout=60000 |
| ) |
| editor = await page.query_selector("div[class*='Editor_editor__']") |
| await editor.focus() |
| await editor.click() |
|
|
| for line in cleaned_text.split("\n"): |
| await page.keyboard.type(line) |
| await page.keyboard.press("Enter") |
|
|
| await page.evaluate( |
| """() => { |
| const button = document.querySelector('button[aria-label="Export as PNG"]'); |
| button.click(); |
| }""" |
| ) |
|
|
| async with page.expect_download() as download_info: |
| download = await download_info.value |
| await download.save_as(name) |
| except playwright._impl._errors.TimeoutError: |
| LOGS.error("Timeout error: Selector not found within 60 seconds.") |
| await proc.eor("Failed to find the editor within 60 seconds.") |
| return |
| except Exception as e: |
| LOGS.error(f"Error occurred during playwright operation: {e}") |
| await proc.eor("An error occurred during the operation.") |
| return |
| finally: |
| if os.path.exists(name): |
| try: |
| await ult.reply(file=name) |
| await proc.try_delete() |
| os.remove(name) |
| except Exception as e: |
| LOGS.error(f"Error occurred while replying with the file: {e}") |
| await proc.eor("Failed to send the file.") |
| else: |
| LOGS.error(f"Error: File {name} not found or inaccessible.") |
| await proc.eor("Failed to save the file.") |
|
|
