Created check for Matrix mode

Reviewed-on: #1

.drone.yml

@@ -4,11 +4,6 @@ type: docker
 name: publish-bot
 
 steps:
-- name: build-dotnet
-  image: mcr.microsoft.com/dotnet/sdk:6.0
-  commands:
-  - dotnet publish
-
 - name: build-image
   image: plugins/docker
   settings:
@@ -17,10 +12,8 @@ steps:
       from_secret: DOCKER_USER
     password:
       from_secret: DOCKER_PASS
-    repo: git.kizaing.ca/kizaing/TeleTok
+    repo: git.kizaing.ca/kizaing/teletok
     tags: latest
-    platform: linux/amd64,linux/arm64
-
 
 # Commented out until stuff actually works
 trigger:

.gitignore

@@ -1,2 +1,4 @@
 bin
 obj
+.vscode
+config.Dev.json

Dockerfile

@@ -1,7 +1,14 @@
-FROM mcr.microsoft.com/dotnet/runtime:6.0-alpine3
+#Builds the bot from source
+FROM mcr.microsoft.com/dotnet/sdk:6.0 AS build-task
+
+COPY . /build
+RUN cd /build && dotnet publish
+
+# Actually runs the bot
+FROM mcr.microsoft.com/dotnet/runtime:6.0
 
 WORKDIR /app/teletok
 
-COPY bin/Debug/net6.0/publish/* /app/teletok/
+COPY --from=build-task /build/bin/Debug/net6.0/publish/* /app/teletok/
 
-CMD [ "TeleTok" ]
+CMD [ "./TeleTok" ]

MatrixListener.cs

@@ -0,0 +1,14 @@
+using System;
+using System.Configuration;
+using Matrix.Sdk;
+
+namespace TeleTok
+{
+    public class MatrixListener
+    {
+        public void RunListener()
+        {
+            
+        }
+    }
+}

Program.cs

@@ -10,17 +10,92 @@ namespace TeleTok
             .AddJsonFile("config.json", true)
             .Build();
 
+        // Value to see what bot mode to run in
+        public static string botMode = config.GetSection("TeleTokConf:botMode").Value;
+
+        // Telegram bot config
         public static string token = config.GetSection("TeleTokConf:token").Value;
-        public static string ptInstance = config.GetSection("TeleTokConf:proxitokInstance").Value;
+        public static string ptInstance = config.GetSection("TeleTokConf:ptInstance").Value;
+
+        // Matrix bot config
+        public static string matrixAddress = config.GetSection("TeleTokConf:matrixAddress").Value;
+        public static string mBotUser = config.GetSection("TeleTokConf:mBotUser").Value;
+        public static string mBotPass = config.GetSection("TeleTokConf:mBotPass").Value;
 
         static async Task Main(string[] args)
         {
+            //Checks to see what mode to run the bot in
 
-            TelegramListener listener = new TelegramListener();
+            if(botMode == "telegram")
-            Console.WriteLine("Now listening...");
+            {
+                //Checks if the config json data is valid
+                if(ConfigCheck(token) == false)
+                {
+                    LogMessage("Telegram bot token is invalid! Exiting...");
+                }
+                else if(ConfigCheck(ptInstance) == false)
+                {
+                    LogMessage("Proxitok instance is invalid! Exiting...");
+                }
+                else
+                {
+                    TelegramListener tListener = new TelegramListener();
+                    LogMessage("Now listening...");
 
-            listener.RunListener();
+                    tListener.RunListener();
+                }
+            }
+            else if(botMode == "matrix")
+            {
+                //Checks if the config json data is valid
+                if(ConfigCheck(matrixAddress) == false)
+                {
+                    LogMessage("Synapse address is invalid! Exiting...");
+                }
+                else if(ConfigCheck(mBotUser) == false)
+                {
+                    LogMessage("Matrix bot username is invalid! Exiting...");
+                }
+                else if(ConfigCheck(mBotPass) == false)
+                {
+                    LogMessage("Matrix bot password is invalid! Exiting...");
+                }
+                else
+                {
+                    MatrixListener mListener = new MatrixListener();
+                    LogMessage("Now listening...");
 
+                    mListener.RunListener();
+                }              
+            }
+            else
+            {
+                LogMessage("Bot mode is not configured! Enter either \'telegram\' or \'matrix\'");
+            }
+        }
+
+        public static bool ConfigCheck(string confItem) 
+        {
+            if(confItem == "" || confItem == null
+                || confItem == "INSERT TOKEN HERE"
+                || confItem == "PROXITOK INSTANCE URL"
+                || confItem == "SYNAPSE SERVER URL"
+                || confItem == "MATRIX BOT USERNAMEL"
+                || confItem == "MATRIX BOT PASSWORD")
+            {
+                return false;
+            }
+            else
+            {
+                return true;
+            }
+        }
+
+        public static void LogMessage(string text)
+        {   
+            DateTime now =DateTime.Now;
+
+            Console.WriteLine("[" + now.ToString() + "] " + text);
         }
     }
 }

README.md

@@ -1,5 +1,33 @@
 [![Build Status](https://ci.kizaing.ca/api/badges/kizaing/TeleTok/status.svg)](https://ci.kizaing.ca/kizaing/TeleTok)
 
 # TeleTok Telegram Bot
+This bot will monitor any chats for TikTok links that are posted, and then will run the link through a [ProxiTok](https://github.com/pablouser1/ProxiTok) instance to generate a download link. The resulting video file will then get directly uploaded directly to your chat.
 
-This bot is in early development, but the end goal is to have this bot automatically convert all tiktok links in a chat into a videofile that can be re-uploaded to the telegram chat
+## Building
+Requirements: Dotnet SDK 6.0
+
+Run `dotnet build` for a Debug instance
+Run `dotnet publish` for a full release
+
+## Installation
+### Requirements
+ * A telegram bot token
+ * A working [ProxiTok](https://github.com/pablouser1/ProxiTok) instance, either public or hosted yourself. (Please be kind and take bandwidth into account when using a public instance)
+
+### Binary
+ 1. Either build the application or download a release zip for your OS/architecture
+ 2. Edit the config.json and put your telegram bot token in the "token" field, and your chosen ProxiTok instance in the "ptInstance" field
+ 3. Run the TeleTok.exe/TeleTok executable
+
+### Docker
+ 1. You can build an image for yourself with `docker build teletok .` or you can pull a prebuilt image with `docker pull kizaing/teletok:latest`
+ 2. Copy the placeholder config.json from the repo to a location of your choosing
+ 3. Edit the config.json and put your telegram bot token in the "token" field, and your chosen ProxiTok instance in the "ptInstance" field
+ 4. Run the container with `docker run -v ${PWD}/config.json:/app/teletok/config.json -d kizaing/teletok:latest`
+
+
+## To Do
+ - [x] Add better error and link handling
+ - [x] Process videos into telegram
+ - [ ] Add [Matrix](https://matrix.org) support
+ - [ ] Automate binary release publishing

TeleTok.csproj

@@ -8,6 +8,7 @@
   </PropertyGroup>
 
   <ItemGroup>
+    <PackageReference Include="Matrix.Sdk" Version="1.0.7" />
     <PackageReference Include="Microsoft.Extensions.Configuration.Binder" Version="7.0.2" />
     <PackageReference Include="Microsoft.Extensions.Configuration.Json" Version="7.0.0" />
     <PackageReference Include="Telegram.Bot" Version="17.0.0" />

TelegramListener.cs

@@ -54,8 +54,6 @@ namespace TeleTok
 
             // Passes the url along to the video downloader if it is valid AND a tiktok link
             if (isUri)
-            {
-                try
             {
                 if(messageText.Contains("tiktok.com"))
                 {
@@ -68,11 +66,6 @@ namespace TeleTok
                     );
                 }
             }
-                catch 
-                {
-                    Console.WriteLine("Valid TikTok URI was sent, but was not a video!");
-                }
-            }
         }
 
         Task HandlePollingErrorAsync(ITelegramBotClient botClient, Exception exception, CancellationToken cancellationToken)
@@ -84,7 +77,7 @@ namespace TeleTok
                 _ => exception.ToString()
             };
 
-            Console.WriteLine(ErrorMessage);
+            TeleTok.LogMessage(ErrorMessage);
             return Task.CompletedTask;
         }
     }

VidDownload.cs

@@ -12,13 +12,16 @@ namespace TeleTok
             string url = videourl;
             string proxyUrl;
 
+            TeleTok.LogMessage("Video for " + videourl + " processing..");
+
             if(url.Contains("vm.tiktok.com"))
             {
                 url = UnshortenUrl(url);
             }
 
-            proxyUrl = TeleTok.ptInstance + "/download?url=" + url;
+            proxyUrl = CreateDownloadLink(url);
-            Console.WriteLine("Video for " + url + " has been sent..");
+
+            TeleTok.LogMessage("Sending video link for " + proxyUrl);
 
             return proxyUrl;
         }
@@ -33,5 +36,25 @@ namespace TeleTok
 
             return realUrl;
         }
+
+        //Breaks apart the URL and extracts the User and Video ID to be processed into a working download link
+        static string CreateDownloadLink(string videourl)
+        {
+            Uri segmentedUri = new Uri(videourl);
+            segmentedUri = new Uri(segmentedUri.AbsoluteUri.Replace(segmentedUri.Query, string.Empty));
+
+            string videoUser = segmentedUri.Segments[1];
+            videoUser = videoUser.Replace(@"/", "");
+            string videoID = segmentedUri.Segments[3];
+
+            string fixedUrl = "https://www.tiktok.com/" + videoUser + "/video/" + videoID + @"&id=" + videoID + @"&user=" + videoUser.Remove(0, 1);
+
+            string proxyLink = TeleTok.ptInstance + "/download?url=" + fixedUrl;
+
+            TeleTok.LogMessage("Input User ID is: " + videoUser);
+            TeleTok.LogMessage("Input video ID is: " + videoID);
+
+            return proxyLink;
+        }
     }
 }

config.json

@@ -1,6 +1,10 @@
 {
     "TeleTokConf": {
+        "botMode": "telegram",
         "token": "INSERT TOKEN HERE",
-        "proxitokInstance": "PROXITOK INSTANCE URL"
+        "ptInstance": "PROXITOK INSTANCE URL",
+        "matrixAddress": "SYNAPSE SERVER URL",
+        "mBotUser": "MATRIX BOT USERNAME",
+        "mBotPass": "MATRIX BOT PASSWORD"
     }
 }