102 lines
3.4 KiB
Python
Executable File
102 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3.7
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
TFC - Onion-routed, endpoint secure messaging system
|
|
Copyright (C) 2013-2020 Markus Ottela
|
|
|
|
This file is part of TFC.
|
|
|
|
TFC is free software: you can redistribute it and/or modify it under the terms
|
|
of the GNU General Public License as published by the Free Software Foundation,
|
|
either version 3 of the License, or (at your option) any later version.
|
|
|
|
TFC is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
|
|
without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
PURPOSE. See the GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with TFC. If not, see <https://www.gnu.org/licenses/>.
|
|
"""
|
|
|
|
import typing
|
|
|
|
from src.common.output import print_on_previous_line
|
|
from src.common.statics import COMMAND, FILE, MESSAGE, WIN_TYPE_GROUP
|
|
|
|
if typing.TYPE_CHECKING:
|
|
from src.common.db_settings import Settings
|
|
from src.transmitter.windows import TxWindow
|
|
|
|
|
|
def process_aliases(plaintext: str,
|
|
settings: 'Settings',
|
|
window: 'TxWindow'
|
|
) -> str:
|
|
"""Check if plaintext is an alias for another command."""
|
|
aliases = [(' ', '/unread' ),
|
|
(' ', '/exit' if settings.double_space_exits else '/clear'),
|
|
('//', '/cmd' )]
|
|
|
|
for a in aliases:
|
|
if plaintext == a[0]:
|
|
plaintext = a[1]
|
|
|
|
# Replace what the user typed
|
|
print_on_previous_line()
|
|
print(f"Msg to {window.type_print} {window.name}: {plaintext}")
|
|
break
|
|
|
|
return plaintext
|
|
|
|
|
|
def get_input(window: 'TxWindow', settings: 'Settings') -> 'UserInput':
|
|
"""Read and process input from the user and determine its type."""
|
|
while True:
|
|
try:
|
|
plaintext = input(f"Msg to {window.type_print} {window.name}: ")
|
|
if plaintext in ['', '/']:
|
|
raise EOFError
|
|
except (EOFError, KeyboardInterrupt):
|
|
print('')
|
|
print_on_previous_line()
|
|
continue
|
|
|
|
plaintext = process_aliases(plaintext, settings, window)
|
|
|
|
# Determine plaintext type
|
|
pt_type = MESSAGE
|
|
|
|
if plaintext == '/file':
|
|
pt_type = FILE
|
|
|
|
elif plaintext.startswith('/'):
|
|
plaintext = plaintext[len('/'):]
|
|
pt_type = COMMAND
|
|
|
|
# Check if the group was empty
|
|
if pt_type in [MESSAGE, FILE] and window.type == WIN_TYPE_GROUP:
|
|
if window.group is not None and window.group.empty():
|
|
print_on_previous_line()
|
|
print(f"Msg to {window.type_print} {window.name}: Error: The group is empty.")
|
|
print_on_previous_line(delay=0.5)
|
|
continue
|
|
|
|
return UserInput(plaintext, pt_type)
|
|
|
|
|
|
class UserInput(object):
|
|
"""UserInput objects are messages, files or commands.
|
|
|
|
The type of created UserInput object is determined based on input
|
|
by the user. Commands start with a slash, but as files are a special
|
|
case of a command, /file commands are interpreted as the file type.
|
|
The 'type' attribute allows tx_loop to determine what function
|
|
should process the user input.
|
|
"""
|
|
|
|
def __init__(self, plaintext: str, type_: str) -> None:
|
|
"""Create a new UserInput object."""
|
|
self.plaintext = plaintext
|
|
self.type = type_
|