# 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 from typing import Optional 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(), ) ) self._session: Optional[aiohttp.ClientSession] = None async def _get_session(self) -> aiohttp.ClientSession: if self._session is None or self._session.closed: self._session = aiohttp.ClientSession( timeout=aiohttp.ClientTimeout(total=15) ) return self._session async def on_unload(self): if self._session and not self._session.closed: await self._session.close() 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) -> Optional[str]: session = await self._get_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 None async def get_bitlink_stats(self, bitlink: str, token: str) -> Optional[int]: session = await self._get_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 None @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"]) if short_url is None: await utils.answer( message, self.strings("api_error").format(error="Failed to shorten URL"), ) return 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"] ) if clicks is None: await utils.answer( message, self.strings("api_error").format( error="Failed to get statistics" ), ) return 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)))