Added tasks integration

Reviewed-on: #1

.gitignore

@@ -1,2 +1,3 @@
 config.ini
+venv/
 __pycache__

README.md

@@ -7,3 +7,6 @@ sudo apt install python3-slixmpp
 
 ## Install dependencies with pip
 pip install -r requirements.txt
+
+## Setup
+Create `config.ini` based on the `config.ini.example`, with your credentials

functions.py

@@ -1,5 +1,29 @@
+import ollama
+import scraper_functions as sf
+
 def processmsg(msg, rcpt):
-	if msg.startswith("!"):
-		return ""
-	elif "youtube.com/watch" in msg:
+    if "youtube.com/watch" in msg:
         return msg.replace("youtube.com", "iv.datura.network")
+    elif msg.startswith("!"):
+        return command(msg, "")
+    elif "good bot" in msg:
+        return "^_^"
+
+def command(msg, rcpt):
+    if msg.startswith("!help"):
+        response = "chatbot commands:"  + "\n"
+        response += "!help            Show this help page" + "\n"
+        response += "!ai [message]    Ask llama2" + "\n"
+        response += "!wiki [message]    Ask wiki"
+        return response
+    elif msg.startswith("!ai"):
+        client = ollama.Client(host='https://ollama.krov.dmz.rs')
+        response = client.chat(model='llama2-uncensored:latest', messages=[{'role':'user','content':f'{msg[4:]}'}])
+        return(response['message']['content'])
+    elif msg.startswith("!wiki"):
+        cmd, query = msg.split(" ", 1)
+        return sf.query_external_website("https://en.wikipedia.org", "/wiki/" + query)
+    elif msg.startswith("!tasks"):
+        content = sf.getDmzTasks()
+        return content
+

requirements.txt

@@ -1 +1,4 @@
 slixmpp
+ollama
+requests
+beautifulsoup4

scraper_functions.py

@@ -0,0 +1,33 @@
+import requests
+from bs4 import BeautifulSoup
+from urllib.parse import quote
+
+def query_external_website(base_url, query):
+    try:
+        page = requests.get(base_url + quote(query))
+        soup = BeautifulSoup(page.content, "html.parser")
+        title = soup.find(id="firstHeading").text
+        mainContentElement = soup.find(id="mw-content-text")
+        if "This page is a redirect" in mainContentElement.text:
+            redirectLink = mainContentElement.find(class_="redirectMsg").find_all("a")[0]["href"]
+            return query_external_website(base_url, redirectLink)
+        content = next((paragraph for paragraph in mainContentElement.select("p") if not paragraph.has_attr("class")), None)
+        if content == None:
+            raise Exception("Can't parse search result :(")
+        return "\nTITLE:\n" + title + "\n\nCONTENT:\n" + content.text + "\n\nFULL LINK:\n" + base_url + quote(query)
+    except Exception as e:
+        return e
+
+def getDmzTasks():
+    try:
+        page = requests.get("https://todo.dmz.rs/")
+        soup = BeautifulSoup(page.content, "html.parser")
+        tasks = soup.find_all(class_="task")
+        result = "\nActive tasks:\n"
+        for task in tasks:
+            taskIndex = task.select("div")[0].text
+            taskTitle = task.select("div")[1].text
+            result += taskIndex + " " + taskTitle + "\n"
+        return result
+    except Exception as e:
+        return e