Onionr/onionr/static-data/default-plugins/flow/main.py

136 lines
5.2 KiB
Python
Raw Normal View History

2018-07-11 07:35:22 +00:00
'''
2019-06-20 07:59:32 +00:00
Onionr - Private P2P Communication
2018-07-11 07:35:22 +00:00
This default plugin handles "flow" messages (global chatroom style communication)
'''
'''
This program 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.
This program 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 this program. If not, see <https://www.gnu.org/licenses/>.
'''
# Imports some useful libraries
2019-04-27 15:43:16 +00:00
import threading, time, locale, sys, os
2018-07-11 07:35:22 +00:00
from onionrblockapi import Block
2019-07-19 19:49:56 +00:00
import logger, config, onionrblocks
2019-07-20 21:21:03 +00:00
from onionrutils import escapeansi, epoch, bytesconverter
2019-04-27 15:43:16 +00:00
locale.setlocale(locale.LC_ALL, '')
from coredb import blockmetadb
2019-07-29 03:00:38 +00:00
from utils import identifyhome, reconstructhash
import deadsimplekv as simplekv
2019-04-27 15:43:16 +00:00
sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)))
import flowapi # import after path insert
flask_blueprint = flowapi.flask_blueprint
2018-07-11 07:35:22 +00:00
plugin_name = 'flow'
2019-04-27 15:43:16 +00:00
PLUGIN_VERSION = '0.0.1'
2018-07-11 07:35:22 +00:00
class OnionrFlow:
def __init__(self):
self.alreadyOutputed = []
self.flowRunning = False
2019-07-29 03:00:38 +00:00
self.channel = ""
2018-07-11 07:35:22 +00:00
return
def start(self):
logger.warn("Please note: everything said here is public, even if a random channel name is used.", terminal=True)
message = ""
self.flowRunning = True
2018-12-09 17:29:39 +00:00
try:
2019-07-29 03:00:38 +00:00
self.channel = logger.readline("Enter a channel name or none for default:").strip()
2018-12-09 17:29:39 +00:00
except (KeyboardInterrupt, EOFError) as e:
self.flowRunning = False
2019-07-29 03:00:38 +00:00
newThread = threading.Thread(target=self.showOutput, daemon=True)
newThread.start()
while self.flowRunning:
2019-07-29 03:00:38 +00:00
if self.channel == "":
self.channel = "global"
try:
message = logger.readline('\nInsert message into flow:').strip().replace('\n', '\\n').replace('\r', '\\r')
except EOFError:
pass
except KeyboardInterrupt:
self.flowRunning = False
2018-12-09 17:29:39 +00:00
else:
if message == "q":
self.flowRunning = False
expireTime = epoch.get_epoch() + 43200
2018-12-09 17:29:39 +00:00
if len(message) > 0:
logger.info('Inserting message as block...', terminal=True)
2019-07-29 03:00:38 +00:00
onionrblocks.insert(message, header='brd', expire=expireTime, meta={'ch': self.channel})
logger.info("Flow is exiting, goodbye", terminal=True)
return
def showOutput(self):
2018-12-09 17:29:39 +00:00
while type(self.channel) is type(None) and self.flowRunning:
time.sleep(1)
try:
while self.flowRunning:
2019-07-29 03:00:38 +00:00
for block in blockmetadb.get_blocks_by_type('brd'):
2019-07-20 21:21:03 +00:00
if block in self.alreadyOutputed:
continue
2019-04-27 15:43:16 +00:00
block = Block(block)
2019-07-20 21:21:03 +00:00
b_hash = bytesconverter.bytes_to_str(block.getHash())
2019-04-27 15:43:16 +00:00
if block.getMetadata('ch') != self.channel:
continue
if not self.flowRunning:
break
logger.info('\n------------------------', prompt = False, terminal=True)
2019-04-27 15:43:16 +00:00
content = block.getContent()
# Escape new lines, remove trailing whitespace, and escape ansi sequences
content = escapeansi.escape_ANSI(content.replace('\n', '\\n').replace('\r', '\\r').strip())
logger.info(block.getDate().strftime("%m/%d %H:%M") + ' - ' + logger.colors.reset + content, prompt = False, terminal=True)
2019-07-20 21:21:03 +00:00
self.alreadyOutputed.append(b_hash)
2019-04-27 15:43:16 +00:00
time.sleep(5)
2018-12-09 17:29:39 +00:00
except KeyboardInterrupt:
self.flowRunning = False
2018-07-11 07:35:22 +00:00
2019-08-05 00:34:57 +00:00
def on_flow_cmd(api, data=None):
OnionrFlow().start()
2018-07-11 07:35:22 +00:00
def on_init(api, data = None):
'''
This event is called after Onionr is initialized, but before the command
inputted is executed. Could be called when daemon is starting or when
just the client is running.
'''
2019-08-06 05:49:31 +00:00
return
2019-07-29 03:00:38 +00:00
def on_processblocks(api, data=None):
metadata = data['block'].bmetadata # Get the block metadata
2019-08-06 05:49:31 +00:00
if data['type'] != 'brd':
2019-08-05 23:09:04 +00:00
return
2019-08-06 05:49:31 +00:00
b_hash = reconstructhash.deconstruct_hash(data['block'].hash) # Get the 0-truncated block hash
board_cache = simplekv.DeadSimpleKV(identifyhome.identify_home() + '/board-index.cache.json') # get the board index cache
2019-07-29 03:00:38 +00:00
# Validate the channel name is sane for caching
try:
ch = metadata['ch']
except KeyError:
ch = 'global'
ch_len = len(ch)
2019-08-05 23:09:04 +00:00
if ch_len == 0:
2019-07-29 03:00:38 +00:00
ch = 'global'
2019-08-05 23:09:04 +00:00
elif ch_len > 12:
2019-07-29 03:00:38 +00:00
return
existing_posts = board_cache.get(ch)
if existing_posts is None:
existing_posts = ''
2019-08-06 05:49:31 +00:00
2019-07-29 03:00:38 +00:00
check_list = existing_posts.split(',')
if len(check_list) > 30:
check_list.pop(0)
existing_posts = ','.join(check_list)
2019-08-06 05:49:31 +00:00
board_cache.put(ch, '%s,%s' % (existing_posts, b_hash))
2019-07-29 03:00:38 +00:00
board_cache.flush()