Implement file manager

README.md

@@ -14,7 +14,9 @@ This program is a PC kill switch Telegram bot. It is a project for New Generatio
 - works on Windows only
 
 ## Usage
-Before starting the bot, copy config-example.py file to config.py and put the password and the token. Administrator privileges are required so that the bot is able to disable input and format partitions. Once the bot is started, send /start to its account on Telegram and enter the password to see what can be done.
+In order to run the bot, you need a Telegram bot token and Google Drive API credentials. A directory on Google Drive where the bot put its files should also be created.
+
+Before starting the bot, copy config-example.py file to config.py and put the Telegram bot token, the bot protection password and the Google Drive folder ID. Put your Google API credentials into credentials.json file. Administrator privileges are required so that the bot is able to disable input and format partitions. Once the bot is started, send /start to its account on Telegram and enter the password to see what can be done.
 
 ## Licensing
 All code in this repository is Unlicensed, see UNLICENSE.

config-example.py

@@ -9,3 +9,6 @@ token = "put your token here"
 
 # Bot password
 master_password = "put your password here"
+
+# Google Drive folder to be used by the program
+root_folder = "put folder ID here"

googledrive.py

@@ -0,0 +1,83 @@
+# Project: Telegram PC kill switch bot
+# Author: Stanislav Mykhailenko
+# License: Unlicense
+
+# This file contains Google Drive interactions
+
+
+from __future__ import print_function
+
+import os
+from mimetypes import MimeTypes
+from google.auth.transport.requests import Request
+from google.oauth2.credentials import Credentials
+from google_auth_oauthlib.flow import InstalledAppFlow
+from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
+from googleapiclient.http import MediaFileUpload
+
+
+creds = None
+
+def authenticate():
+	global creds
+	# The file token.json stores the user's access and refresh tokens, and is
+	# created automatically when the authorization flow completes for the first
+	# time.
+	if os.path.exists(os.path.join(os.path.dirname(__file__), 'token.json')):
+		creds = Credentials.from_authorized_user_file(os.path.join(os.path.dirname(__file__), 'token.json'), ['https://www.googleapis.com/auth/drive'])
+	# If there are no (valid) credentials available, let the user log in.
+	if not creds or not creds.valid:
+		if creds and creds.expired and creds.refresh_token:
+			creds.refresh(Request())
+		else:
+			flow = InstalledAppFlow.from_client_secrets_file(
+				os.path.join(os.path.dirname(__file__), 'credentials.json'), ['https://www.googleapis.com/auth/drive'])
+			creds = flow.run_local_server(port=0)
+		# Save the credentials for the next run
+		with open(os.path.join(os.path.dirname(__file__), 'token.json'), 'w') as token:
+			token.write(creds.to_json())
+
+
+
+def createFolder(name, parent):
+	try:
+		service = build('drive', 'v3', credentials=creds)
+		file_metadata = {
+			'name': [name],
+			'mimeType': 'application/vnd.google-apps.folder',
+			'parents': [parent]
+		}
+
+		file = service.files().create(body=file_metadata, fields='id'
+                                      ).execute()
+		return file.get('id')
+
+	except HttpError as error:
+		print(F'An error occurred: {error}')
+		return None
+
+
+def uploadFile(path, folder):
+	try:
+		service = build('drive', 'v3', credentials=creds)
+
+		name = os.path.basename(os.path.normpath(path))
+
+		file_metadata = {
+			'name': [name],
+			'parents': [folder]
+		}
+
+		mimetype = MimeTypes().guess_type(path)[0]
+
+		media = MediaFileUpload(path,
+				mimetype=mimetype)
+
+		file = service.files().create(body=file_metadata, media_body=media,
+				fields='id').execute()
+		return file.get('id')
+
+	except HttpError as error:
+		print(F'An error occurred: {error}')
+		file = None

main.py

@@ -3,6 +3,8 @@
 # License: Unlicense
 
 import ctypes, sys
+
+from googledrive import authenticate
 from tg import startBot
 
 def main() -> None:
@@ -12,4 +14,5 @@ def main() -> None:
 		ctypes.windll.shell32.ShellExecuteW(None, "runas", sys.executable, " ".join(sys.argv), None, 1)
 
 if __name__ == '__main__':
+	authenticate()
 	main()

payload.py

@@ -4,7 +4,9 @@
 
 # This file contains the payloads
 
-import ctypes, os, pyttsx3
+from googledrive import *
+from config import *
+import ctypes, os, pyttsx3, shutil
 
 def enableInput():
 	ctypes.windll.user32.BlockInput(False)
@@ -23,3 +25,33 @@ def playMessage(message):
 	engine = pyttsx3.init()
 	engine.say(message)
 	engine.runAndWait()
+
+def listData(path):
+	try:
+		return os.listdir(path)
+	except NotADirectoryError:
+		return "Is a file."
+	except FileNotFoundError:
+		return "File or directory not found."
+	except:
+		return "An error occurred when trying to access the data."
+
+def uploadFolder(path, parentDirectory):
+	directory = createFolder(os.path.basename(os.path.normpath(path)), parentDirectory)
+	entries = os.listdir(path)
+	for entry in entries:
+		if os.path.isdir(os.path.join(path, entry)):
+			uploadFolder(os.path.join(path, entry), directory)
+		else:
+			uploadFile(os.path.join(path, entry), directory)
+
+def uploadData(path):
+	if not os.path.exists(path):
+		return False
+	if os.path.isdir(path):
+		uploadFolder(path, root_folder)
+	else:
+		uploadFile(path, root_folder)
+
+def deleteData(path):
+	shutil.rmtree(path, ignore_errors=True)

requirements.txt

@@ -1,10 +1,34 @@
 anyio==3.6.2
+cachetools==5.3.0
 certifi==2022.12.7
+charset-normalizer==3.0.1
+comtypes==1.1.14
+google-api-core==2.11.0
+google-api-python-client==2.75.0
+google-auth==2.16.0
+google-auth-httplib2==0.1.0
+google-auth-oauthlib==0.8.0
+google-drive-api==0.0.4
+googleapis-common-protos==1.58.0
 h11==0.14.0
 httpcore==0.16.3
+httplib2==0.21.0
 httpx==0.23.3
 idna==3.4
+oauthlib==3.2.2
+protobuf==4.21.12
+pyasn1==0.4.8
+pyasn1-modules==0.2.8
+pyparsing==3.0.9
+pypiwin32==223
 python-telegram-bot==20.0
 pyttsx3==2.90
+pywin32==305
+requests==2.28.2
+requests-oauthlib==1.3.1
 rfc3986==1.5.0
+rsa==4.9
+six==1.16.0
 sniffio==1.3.0
+uritemplate==4.1.1
+urllib3==1.26.14

tg.py

@@ -8,7 +8,7 @@ from config import *
 from telegram import ReplyKeyboardMarkup, ReplyKeyboardRemove, Update
 from telegram.ext import Application, CommandHandler, ContextTypes, ConversationHandler, filters, MessageHandler
 from payload import *
-import logging
+import logging, googledrive
 from threading import Thread
 
 logging.basicConfig(
@@ -17,15 +17,15 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-PASSWORD, ACTION, FORMAT, MESSAGE = range(4)
+PASSWORD, ACTION, FORMAT, MESSAGE, FILE_MANAGER, LIST, DELETE, UPLOAD = range(8)
 
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
     await update.message.reply_text("Enter password.")
 
     return PASSWORD
 
-async def show_keyboard(update: Update):
+async def action_keyboard(update: Update):
-	reply_keyboard = [["Enable input", "Disable input", "Lock screen", "Format volumes", "Play message"]]
+	reply_keyboard = [["Enable input", "Disable input", "Lock screen", "Format volumes", "Play message", "File manager"]]
 
 	await update.message.reply_text(
 		"Choose your action:",
@@ -34,12 +34,23 @@ async def show_keyboard(update: Update):
 		),
 	)
 
+async def file_keyboard(update: Update):
+	reply_keyboard = [["List files", "Upload files", "Delete files", "Go back"]]
+
+	await update.message.reply_text(
+		"Choose your action:",
+		reply_markup=ReplyKeyboardMarkup(
+			reply_keyboard, one_time_keyboard=True, input_field_placeholder="Choose your action"
+		),
+	)
+
+
 async def password(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
 	user = update.message.from_user
 
 	if update.message.text == master_password:
 		logger.info("User %s logged in.", user.first_name)
-		await show_keyboard(update)
+		await action_keyboard(update)
 
 		return ACTION
 
@@ -56,28 +67,51 @@ async def action(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
 			logger.info("Got a request from %s to enable input.", user.first_name)
 			enableInput()
 			await update.message.reply_text("Input enabled.", reply_markup=ReplyKeyboardRemove())
-			await show_keyboard(update)
+			await action_keyboard(update)
 			return ACTION
 		case "Disable input":
 			logger.info("Got a request from %s to disable input.", user.first_name)
 			disableInput()
 			await update.message.reply_text("Input disabled.", reply_markup=ReplyKeyboardRemove())
-			await show_keyboard(update)
+			await action_keyboard(update)
 			return ACTION
 		case "Lock screen":
 			logger.info("Got a request from %s to lock screen.", user.first_name)
 			lockScreen()
 			await update.message.reply_text("Screen locked.", reply_markup=ReplyKeyboardRemove())
-			await show_keyboard(update)
+			await action_keyboard(update)
 			return ACTION
 		case "Format volumes":
 			logger.info("Got a request from %s to format volumes.", user.first_name)
-			await update.message.reply_text("Please type a space-separated list of the volumes you want to format.", reply_markup=ReplyKeyboardRemove())
+			await update.message.reply_text("Please enter a space-separated list of the volumes you want to format.", reply_markup=ReplyKeyboardRemove())
 			return FORMAT
 		case "Play message":
 			logger.info("Got a request from %s to play a message.", user.first_name)
-			await update.message.reply_text("Please type the message you want to play.", reply_markup=ReplyKeyboardRemove())
+			await update.message.reply_text("Please enter the message you want to play.", reply_markup=ReplyKeyboardRemove())
 			return MESSAGE
+		case "File manager":
+			await file_keyboard(update)
+			return FILE_MANAGER
+
+async def file_manager(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+	user = update.message.from_user
+
+	match update.message.text:
+		case "List files":
+			logger.info("Got a request from %s to list files.", user.first_name)
+			await update.message.reply_text("Please enter the path.", reply_markup=ReplyKeyboardRemove())
+			return LIST
+		case "Upload files":
+			logger.info("Got a request from %s to upload files.", user.first_name)
+			await update.message.reply_text("Please enter the path.", reply_markup=ReplyKeyboardRemove())
+			return UPLOAD
+		case "Delete files":
+			logger.info("Got a request from %s to delete files.", user.first_name)
+			await update.message.reply_text("Please enter the path.", reply_markup=ReplyKeyboardRemove())
+			return DELETE
+		case "Go back":
+			await action_keyboard(update)
+			return ACTION
 
 async def format(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
 	user = update.message.from_user
@@ -86,7 +120,7 @@ async def format(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
 	Thread(target=formatVolumes,args=(update.message.text.split(),)).start()
 	await update.message.reply_text("Command to format the volumes sent.")
 
-	await show_keyboard(update)
+	await action_keyboard(update)
 	return ACTION
 
 async def message(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
@@ -95,9 +129,38 @@ async def message(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
 	Thread(target=playMessage,args=(update.message.text,)).start()
 	await update.message.reply_text("Command to play the message sent.")
 
-	await show_keyboard(update)
+	await action_keyboard(update)
 	return ACTION
 
+async def list(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+	user = update.message.from_user
+
+	logger.info("Got a request from %s to list files at %s.", user.first_name, update.message.text)                
+	await update.message.reply_text(listData(os.path.join(update.message.text)))
+
+	await file_keyboard(update)
+	return FILE_MANAGER
+
+async def upload(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+	user = update.message.from_user
+
+	logger.info("Got a request from %s to upload %s.", user.first_name, update.message.text)                
+	Thread(target=uploadData,args=(os.path.join(update.message.text),)).start()
+	await update.message.reply_text("Command to upload the data sent.")
+
+	await file_keyboard(update)
+	return FILE_MANAGER
+
+async def delete(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+	user = update.message.from_user
+
+	logger.info("Got a request from %s to delete %s.", user.first_name, update.message.text)                
+	Thread(target=deleteData,args=(os.path.join(update.message.text),)).start()
+	await update.message.reply_text("Command to delete the data sent.")
+
+	await file_keyboard(update)
+	return FILE_MANAGER
+
 async def cancel(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
 	user = update.message.from_user
 	logger.info("User %s cancelled the conversation.", user.first_name)
@@ -112,9 +175,13 @@ def startBot():
 		entry_points=[CommandHandler("start", start)],
 		states={
 			PASSWORD: [MessageHandler(filters.TEXT & ~filters.COMMAND, password)],
-			ACTION: [MessageHandler(filters.Regex("^(Enable input|Disable input|Lock screen|Format volumes|Play message)$"), action)],
+			ACTION: [MessageHandler(filters.Regex("^(Enable input|Disable input|Lock screen|Format volumes|Play message|File manager)$"), action)],
 			FORMAT: [MessageHandler(filters.TEXT & ~filters.COMMAND, format)],
 			MESSAGE: [MessageHandler(filters.TEXT & ~filters.COMMAND, message)],
+			FILE_MANAGER: [MessageHandler(filters.Regex("^(List files|Upload files|Delete files|Go back)$"), file_manager)],
+			LIST: [MessageHandler(filters.TEXT & ~filters.COMMAND, list)],
+			DELETE: [MessageHandler(filters.TEXT & ~filters.COMMAND, delete)],
+			UPLOAD: [MessageHandler(filters.TEXT & ~filters.COMMAND, upload)],
 		},
 		fallbacks=[CommandHandler("cancel", cancel)],
 	)