# Proprietary License Agreement # Copyright (c) 2026-2029 Archquise # Permission is hereby granted to any person obtaining a copy of this software and associated documentation files (the "Software"), to use the Software for personal and non-commercial purposes, subject to the following conditions: # 1. The Software may not be modified, altered, or otherwise changed in any way without the explicit written permission of the author. # 2. Redistribution of the Software, in original or modified form, is strictly prohibited without the explicit written permission of the author. # 3. The Software is provided "as is", without warranty of any kind, express or implied, including but not limited to the warranties of merchantability, fitness for a particular purpose, and non-infringement. In no event shall the author or copyright holder be liable for any claim, damages, or other liability, whether in an action of contract, tort, or otherwise, arising from, out of, or in connection with the Software or the use or other dealings in the Software. # 4. Any use of the Software must include the above copyright notice and this permission notice in all copies or substantial portions of the Software. # 5. By using the Software, you agree to be bound by the terms and conditions of this license. # For any inquiries or requests for permissions, please contact archquise@gmail.com # --------------------------------------------------------------------------------- # Name: Shortener # Description: Module for using bit.ly API # Author: @hikka_mods # --------------------------------------------------------------------------------- # meta developer: @hikka_mods # scope: Shortener # scope: Shortener 0.0.1 # --------------------------------------------------------------------------------- import logging import re import aiohttp from .. import loader, utils logger = logging.getLogger(__name__) @loader.tds class Shortener(loader.Module): """Module for using bit.ly API""" strings = { "name": "Shortener", "no_api": "❌ You have not specified an API token from the site bit.ly", "statclcmd": "📊 Statistics on clicks for this link: {c}", "shortencmd": "✅ Your shortened link is ready: {c}", "no_args": "❌ Please provide a URL to shorten.", "invalid_url": "❌ Invalid URL format.", "api_error": "❌ API error: {error}", "_cls_doc": "Module for using bit.ly API", } strings_ru = { "no_api": "❌ Вы не указали api токен с сайта bit.ly", "statclcmd": "📊 Статистика о переходе по этой ссылке: {c}", "shortencmd": "✅ Ваша сокращённая ссылка готова: {c}", "no_args": "❌ Пожалуйста, укажите URL для сокращения.", "invalid_url": "❌ Неверный формат URL.", "api_error": "❌ Ошибка API: {error}", "_cls_doc": "Модуль для использования API bit.ly", } def __init__(self): self.config = loader.ModuleConfig( loader.ConfigValue( "token", None, lambda: "Need a token with https://app.bitly.com/settings/api/", validator=loader.validators.Hidden(), ) ) def _validate_url(self, url: str) -> bool: """Validate URL format""" if not url: return False url_pattern = re.compile( r"^https?://" r"(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+[A-Z]{2,6}\.?|" r"localhost|" r"\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})" r"(?::\d+)?" r"(?:/?|[/?]\S+)$", re.IGNORECASE, ) return url_pattern.match(url) is not None async def shorten_url(self, url: str, token: str) -> str: async with aiohttp.ClientSession() as session: async with session.post("https://api-ssl.bitly.com/v4/shorten", json={'long_url': url}, headers={"Authorization": f"Bearer {token}"}) as resp: if resp.status == 201: json_response = await resp.json() return json_response['link'] else: logger.error(f"Error occurred! Status code: {resp.status}") return async def get_bitlink_stats(self, bitlink: str, token: str) -> str: async with aiohttp.ClientSession() as session: async with session.get(f"https://api-ssl.bitly.com/v4/bitlinks/{bitlink}/clicks/summary", headers={"Authorization": f"Bearer {token}"}) as resp: if resp.status == 200: json_response = await resp.json() return json_response['total_clicks'] else: logger.error(f"Error occurred! Status code: {resp.status}") return @loader.command( ru_doc="Сократить ссылку через bit.ly (ссылка с https://)", en_doc="Shorten the link via bit.ly (url with https://)", ) async def shortencmd(self, message): """Shorten URL using bit.ly API""" if self.config["token"] is None: await utils.answer(message, self.strings("no_api")) return args = utils.get_args_raw(message) if not args: await utils.answer(message, self.strings("no_args")) return if not self._validate_url(args): await utils.answer(message, self.strings("invalid_url")) return try: short_url = await self.shorten_url(url=args, token=self.config['token']) await utils.answer(message, self.strings("shortencmd").format(c=short_url)) except Exception as e: logger.error(f"Error shortening URL: {e}") await utils.answer(message, self.strings("api_error").format(error=str(e))) @loader.command( ru_doc="Посмотреть статистику ссылки через bit.ly (ссылка без https:// | Доступно только на платных аккаунтах)", en_doc="View link statistics via bit.ly (link without https:// | Works only on paid accounts)", ) async def statclcmd(self, message): """Get click statistics for shortened URL""" if self.config["token"] is None: await utils.answer(message, self.strings("no_api")) return args = utils.get_args_raw(message) if not args: await utils.answer(message, self.strings("no_args")) return try: if not args.startswith("bit.ly/"): await utils.answer(message, self.strings("invalid_url")) return else: clicks = await self.get_bitlink_stats(bitlink=args, token=self.config['token']) await utils.answer(message, self.strings("statclcmd").format(c=clicks)) except Exception as e: logger.error(f"Error getting statistics: {e}") await utils.answer(message, self.strings("api_error").format(error=str(e)))