FTG-Modules/url.py at main · GeekTG/FTG-Modules · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
# -*- coding: utf-8 -*-

# Module author: @GovnoCodules, @ftgmodulesbyfl1yd

import logging
import os
import urllib

from requests import post
from telethon import events, functions
from telethon.errors.rpcerrorlist import YouBlockedUserError

from .. import loader, utils

logger = logging.getLogger(__name__)


@loader.tds
class URlMod(loader.Module):
    """URL Module"""

    strings = {
        "name": "URL",
        "some_rong": "<b>You're doing something wrong!\\write</b> <code>.help "
        + "gg.gg</code> <b>for information.</b>",
        "result": "<b>Here you go, help yourself.</b>\n<a href='{}'>{}</a>",
        "default": "How to use Google?",
    }

    async def client_ready(self, client, db):
        self.client = client

    async def ggcmd(self, message):
        """.gg <link or reply_to_link>"""
        m_text = utils.get_args_raw(message)
        if not m_text:
            reply = await message.get_reply_message()
            if not reply:
                await utils.answer(message, self.strings("some_rong", message))
                return
            long_url = reply.raw_text
        else:
            long_url = m_text

        if "http://" not in long_url and "https://" not in long_url:
            long_url = f"http://{long_url}"
        t_check = f"URL: {long_url}\nCheck..."
        await utils.answer(message, t_check)
        check = post(
            "http://gg.gg/check",
            data={
                "custom_path": None,
                "use_norefs": "0",
                "long_url": long_url,
                "app": "site",
                "version": "0.1",
            },
        ).text
        if check != "ok":
            await utils.answer(message, check)
            return
        await utils.answer(message, "Create...")
        short = post(
            "http://gg.gg/create",
            data={
                "custom_path": None,
                "use_norefs": "0",
                "long_url": long_url,
                "app": "site",
                "version": "0.1",
            },
        ).text
        await utils.answer(message, short)

    async def lgtcmd(self, message):
        """Shorten the link using the verylegit.link service"""
        args = utils.get_args_raw(message)
        if not args:
            return await message.edit("No arguments")
        link = os.popen(f"curl verylegit.link/sketchify -d long_url={args}").read()
        await message.edit(f"Ссылка:\n> {link}")

    async def clckcmd(self, message):
        """Shorten the link using the service clck.ru"""
        m_text = utils.get_args_raw(message)
        if not m_text:
            reply = await message.get_reply_message()
            if not reply:
                await utils.answer(message, self.strings("some_rong", message))
                return
            long_url = reply.raw_text
        else:
            long_url = m_text
        await utils.answer(message, "Creating...")
        fetcher = post(f"https://clck.ru/--?url={long_url}").text
        await utils.answer(message, fetcher)

    async def lmgtfycmd(self, message):
        """Use in reply to another message or as .lmgtfy <text>"""
        text = utils.get_args_raw(message)
        if not text:
            if message.is_reply:
                text = (await message.get_reply_message()).message
            else:
                text = self.strings("default", message)
        query_encoded = urllib.parse.quote_plus(text)
        lmgtfy_url = "http://lmgtfy.com/?s=g&iie=1&q={}".format(query_encoded)
        await utils.answer(
            message,
            self.strings("result", message).format(
                utils.escape_html(lmgtfy_url), utils.escape_html(text)
            ),
        )

    async def nullcmd(self, message):
        """Shorten the link using the nullify service"""
        chat = "@nullifybot"
        reply = await message.get_reply_message()
        async with message.client.conversation(chat) as conv:
            if not reply:
                text = utils.get_args_raw(message)
            else:
                text = await message.get_reply_message()
            try:
                response = conv.wait_event(
                    events.NewMessage(incoming=True, from_users=1481485420)
                )
                mm = await message.client.send_message(chat, text)
                response = await response
                await mm.delete()
            except YouBlockedUserError:
                await message.edit("<code>Unblock @nullifybot</code>")
                return
            await message.edit(response.text.replace("🔗 Твоя ссылка: ", ""))
            await message.client(
                functions.messages.DeleteHistoryRequest(
                    peer="nullifybot", max_id=0, just_clear=False, revoke=True
                )
            )