Cleaning root.

.gitignore

@@ -4,3 +4,6 @@ assets/cert*
 main.db
 main.db-journal
 settings.ini
+code.png
+*.swp
+*.swo

README.md

@@ -3,8 +3,27 @@
 # Create your own spy bot in Telegram ¯\_(ツ)_/¯
 
 This bot is able to get every message in your group, log it and send simple stat about top 10 used words. 
-Also bot can summon every active member of group with @here command.
 
+
+/code - highlight any code snippet following command. There is lexer guesser but you are able to specify language by adding last line comment like this:
+```
+/code
+class Cat {
+  constructor(name) {
+    this.name = name;
+  }
+}
+function get_name(any_object) {
+  console.log(any_object.name);
+}
+var my_cat = new Cat('Cassandra');
+get_name(my_cat)
+#js
+```
+/sql - perform an SQL request to internal bot SQLite database. Any WRITE requests are prohibited.
+/stat - your top 10 words in conf.
+
+Also bot can summon every active member of group with @here command or just with mentioning bot username like @test_huy_bot .
 **Copyright** 
 
 *As probabilistic morphological analyzer (PMA) this bot uses mystem app, which developed by yandex and provided as binary package.*

database.py

@@ -1,5 +1,6 @@
 import datetime as dt
 
+
 class DataBase:
     def __init__(self, basefile, scheme):
         import sqlite3

webhook.py

@@ -10,6 +10,7 @@ import settings
 # fuckin dirty hack. idk the best way to inherit return func into
 # RequestHandler class
 
+
 class RequestHandler(SimpleHTTPRequestHandler):
     def __init__(self,
                  request,
@@ -36,7 +37,6 @@ class RequestHandler(SimpleHTTPRequestHandler):
         pass
 
 
-
 class WebHook:
     def __init__(self,
                  certfile,

worker.py

@@ -7,17 +7,63 @@ from database import DataBase
 import os
 import urllib.request
 from urllib.parse import urlencode
+import requests
+import json
 import settings
 import re
 
+from pygments import highlight
+from pygments.lexers import PythonLexer
+from pygments.lexers import guess_lexer, get_lexer_by_name
+from pygments.styles import get_style_by_name
+from pygments.formatters import ImageFormatter
+
+
 class MessageWorker:
-    def __init__(self, db, stop_words = 'assets/stop-word.ru'):
+    def __init__(self, db, stop_words='assets/stop-word.ru', settings=settings):
         self.stop_words = stop_words
         self.db = db
+        self.telegram_key = settings.parser.get('bot', 'telegram_key')
+        self.telegram_api = settings.parser.get('bot', 'telegram_api')
+        self.me = self.getMe()
+        print("My name is %s" % self.me['result']['username'])
+
+    def getMe(self):
+        url = self.telegram_api + 'bot' + self.telegram_key + '/getMe'
+        print(url)
+        urllib.request.Request(url)
+        request = urllib.request.Request(url)
+        raw = urllib.request.urlopen(request).read().decode()
+        return json.loads(raw)
 
     def handleUpdate(self, msg):
         try:
-            if msg['message']['text'] == '/scheme':
+            try:
+                input_message = msg['message']['text']
+                if ('@here' in input_message) or (' @'+self.me['result']['username'] in input_message):
+                    conf_id = msg['message']['chat']['id']
+                    user_id = msg['message']['from']['id']
+                    chat_title = msg['message']['chat']['title']
+                    self.db.add_conf(conf_id, chat_title)
+                    if msg['message']['text'] != '@here':
+                        message = msg['message']['text'].replace('@here', '\n').replace(' @'+self.me['result']['username'], '\n')
+                    else:
+                        message = """I summon you!\n"""
+
+                    users = self.db.here(
+                        user_id=user_id,
+                        conf_id=conf_id
+                    )
+                    for user in users:
+                        message += ' @%s ' % (user[0])
+                    self.send(id=conf_id, msg=message)
+                    return True
+            
+                input_message = msg['message']['text'].replace(
+                    '@' + self.me['result']['username'], '')
+            except:
+                input_message = msg['message']['text']
+            if input_message == '/scheme':
                 conf_id = msg['message']['chat']['id']
                 user_id = msg['message']['from']['id']
                 chat_title = msg['message']['chat']['title']
@@ -25,7 +71,7 @@ class MessageWorker:
                 self.send(id=conf_id, msg='```\n' + self.db.scheme + '\n```')
                 return True
 
-            if msg['message']['text'] == '/stat':
+            if input_message == '/stat':
                 conf_id = msg['message']['chat']['id']
                 user_id = msg['message']['from']['id']
                 chat_title = msg['message']['chat']['title']
@@ -41,7 +87,7 @@ class MessageWorker:
                 self.send(id=conf_id, msg=message)
                 return True
 
-            if msg['message']['text'] == '/reset':
+            if input_message == '/reset':
                 conf_id = msg['message']['chat']['id']
                 user_id = msg['message']['from']['id']
                 chat_title = msg['message']['chat']['title']
@@ -53,7 +99,7 @@ class MessageWorker:
                     user_id=user_id)
                 return True
 
-            if msg['message']['text'][:4] == '/sql':
+            if input_message[:4] == '/sql':
                 conf_id = msg['message']['chat']['id']
                 user_id = msg['message']['from']['id']
                 chat_title = msg['message']['chat']['title']
@@ -76,21 +122,49 @@ class MessageWorker:
                 self.send(id=conf_id, msg=msg + ' ```')
                 return True
 
-            if msg['message']['text'][:5] == '@here' or \
+#           if '@here' in input_message:
-                msg['message']['text'][5:] == '@here':
+#               conf_id = msg['message']['chat']['id']
+#               user_id = msg['message']['from']['id']
+#               chat_title = msg['message']['chat']['title']
+#               self.db.add_conf(conf_id, chat_title)
+#               if msg['message']['text'] != '@here':
+#                   message = msg['message']['text'].replace('@here', '\n')
+#               else:
+#                   message = """I summon you!\n"""
+
+#               users = self.db.here(
+#                   user_id=user_id,
+#                   conf_id=conf_id
+#               )
+#               for user in users:
+#                   message += ' @%s ' % (user[0])
+#               self.send(id=conf_id, msg=message)
+#               return True
+            if input_message[:5] == '/code':
                 conf_id = msg['message']['chat']['id']
                 user_id = msg['message']['from']['id']
                 chat_title = msg['message']['chat']['title']
                 self.db.add_conf(conf_id, chat_title)
-
+                if len(msg['message']['text'][6:]) < 10000:
-                message = """I summon you!\n"""
+                    code = msg['message']['text'][6:]
-                users = self.db.here(
+                    code_tag = code[code.rfind('#') + 1:]
-                    user_id=user_id,
+                    try:
-                    conf_id=conf_id
+                        lexer = get_lexer_by_name(code_tag)
-                )
+                        code = code[:code.rfind('#')]
-                for user in users:
+                        print("Lexer is defined as %s" % lexer)
-                    message += '@%s ' % (user[0])
+                    except:
-                self.send(id=conf_id, msg=message)
+                        lexer = guess_lexer(code)
+                    print("lexer is %s" % lexer)
+                    if lexer.name == 'Text only':
+                        lexer = get_lexer_by_name('python')
+                    highlight(code, lexer, ImageFormatter(
+                        font_size=16,
+                        line_number_bg="#242e0c",
+                        line_number_fg="#faddf2",
+                        line_number_bold=True,
+                        font_name='DejaVuSansMono',
+                        style=get_style_by_name('monokai')), outfile="code.png")
+                self.send_img(conf_id)
                 return True
         except:
             return False
@@ -108,7 +182,6 @@ class MessageWorker:
             user_id = msg['message']['from']['id']
             chat_id = msg['message']['chat']['id']
             chat_title = msg['message']['chat']['title']
-            #print(self.clean_text(text))
         except:
             return False
 
@@ -124,12 +197,12 @@ class MessageWorker:
         for word in collection:
             self.db.add_relation(word=word, user_id=user_id, conf_id=chat_id)
 
-        
     def clean_text(self, s):
         file = open(self.stop_words, 'rt')
         sw = file.read().split('\n')
         file.close()
-        s = re.sub(r'(https?:\/\/)?([\da-z\.-]+)\.([\/\w\.-]*)*\/?\S','',s,flags=re.MULTILINE)
+        s = re.sub(
+            r'(https?:\/\/)?([\da-z\.-]+)\.([\/\w\.-]*)*\/?\S', '', s, flags=re.MULTILINE)
         print(s)
         # dirty hack with dat fucking file
         fh = open("tmp.txt", "w")
@@ -155,10 +228,8 @@ class MessageWorker:
         return collection
 
     def send(self, id, msg):
-        print(msg)
+        # print(msg)
-        url =  settings.parser.get('bot', 'telegram_api') + \
+        url = self.telegram_api + 'bot' + self.telegram_key + '/sendMessage'
-            'bot'+ settings.parser.get('bot', 'telegram_key') \
-            + '/sendMessage'
         post_fields = {
             'text': msg,
             'chat_id': id,
@@ -169,3 +240,9 @@ class MessageWorker:
         request = urllib.request.Request(url, urlencode(post_fields).encode())
         json = urllib.request.urlopen(request).read().decode()
         return json
+
+    def send_img(self, id):
+        url = self.telegram_api + 'bot' + self.telegram_key + '/sendPhoto'
+        data = {'chat_id': id}
+        files = {'photo': open('code.png', 'rb')}
+        r = requests.post(url, files=files, data=data)