2024-02-28 23:54:48 +01:00
|
|
|
#!/usr/bin/env python3
|
2024-04-05 17:00:19 +02:00
|
|
|
# -*- coding: utf-8 -*-
|
2024-02-28 23:54:48 +01:00
|
|
|
|
|
|
|
|
import slixmpp
|
2024-05-31 10:42:37 +02:00
|
|
|
from bukubot.xmpp.chat import Chat
|
|
|
|
|
from bukubot.config import Configuration
|
|
|
|
|
from bukubot.about import Documentation
|
2024-04-05 17:00:19 +02:00
|
|
|
|
2024-05-31 10:42:37 +02:00
|
|
|
from bukubot.log import Logger
|
|
|
|
|
from bukubot.version import __version__
|
2024-04-05 17:00:19 +02:00
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
import tomllib
|
|
|
|
|
except:
|
|
|
|
|
import tomli as tomllib
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# time_now = datetime.now()
|
|
|
|
|
# time_now = time_now.strftime("%H:%M:%S")
|
|
|
|
|
|
|
|
|
|
# def print_time():
|
|
|
|
|
# # return datetime.now().strftime("%H:%M:%S")
|
|
|
|
|
# now = datetime.now()
|
|
|
|
|
# current_time = now.strftime("%H:%M:%S")
|
|
|
|
|
# return current_time
|
|
|
|
|
|
|
|
|
|
logger = Logger(__name__)
|
|
|
|
|
|
|
|
|
|
class Client(slixmpp.ClientXMPP):
|
2024-02-28 23:54:48 +01:00
|
|
|
|
|
|
|
|
"""
|
2024-05-31 10:42:37 +02:00
|
|
|
bukubot - Bookmark manager chat bot for Jabber/XMPP.
|
|
|
|
|
bukubot is a bookmark manager bot based on buku and slixmpp.
|
2024-02-28 23:54:48 +01:00
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
def __init__(self, jid, password):
|
|
|
|
|
slixmpp.ClientXMPP.__init__(self, jid, password)
|
|
|
|
|
|
2024-04-05 17:00:19 +02:00
|
|
|
print('client')
|
|
|
|
|
|
|
|
|
|
self.register_plugin('xep_0030') # Service Discovery
|
|
|
|
|
self.register_plugin('xep_0004') # Data Forms
|
|
|
|
|
self.register_plugin('xep_0060') # Publish-Subscribe
|
|
|
|
|
self.register_plugin('xep_0050') # Ad-Hoc Commands
|
|
|
|
|
self.register_plugin('xep_0115') # Entity Capabilities
|
|
|
|
|
self.register_plugin('xep_0122') # Data Forms Validation
|
|
|
|
|
self.register_plugin('xep_0199') # XMPP Ping
|
|
|
|
|
|
|
|
|
|
# Connect to the XMPP server and start processing XMPP stanzas.
|
|
|
|
|
# self.connect()
|
|
|
|
|
# self.process()
|
|
|
|
|
|
2024-02-28 23:54:48 +01:00
|
|
|
# The session_start event will be triggered when
|
|
|
|
|
# the bot establishes its connection with the server
|
|
|
|
|
# and the XML streams are ready for use. We want to
|
|
|
|
|
# listen for this event so that we we can initialize
|
|
|
|
|
# our roster.
|
2024-04-05 17:00:19 +02:00
|
|
|
self.add_event_handler("session_start", self.process_session_start)
|
2024-02-28 23:54:48 +01:00
|
|
|
|
|
|
|
|
# The message event is triggered whenever a message
|
|
|
|
|
# stanza is received. Be aware that that includes
|
|
|
|
|
# MUC messages and error messages.
|
|
|
|
|
self.add_event_handler("message", self.process_message)
|
2024-04-05 17:00:19 +02:00
|
|
|
self.add_event_handler("disco_info", self.process_disco_info)
|
2024-02-28 23:54:48 +01:00
|
|
|
|
2024-04-05 17:00:19 +02:00
|
|
|
async def process_session_start(self, event):
|
2024-02-28 23:54:48 +01:00
|
|
|
"""
|
|
|
|
|
Process the session_start event.
|
|
|
|
|
|
|
|
|
|
Typical actions for the session_start event are
|
|
|
|
|
requesting the roster and broadcasting an initial
|
|
|
|
|
presence stanza.
|
|
|
|
|
|
|
|
|
|
Arguments:
|
|
|
|
|
event -- An empty dictionary. The session_start
|
|
|
|
|
event does not provide any additional
|
|
|
|
|
data.
|
|
|
|
|
"""
|
|
|
|
|
self.command_list()
|
|
|
|
|
self.send_presence()
|
|
|
|
|
# await self.get_roster()
|
|
|
|
|
await self['xep_0115'].update_caps()
|
|
|
|
|
|
2024-04-05 17:00:19 +02:00
|
|
|
async def process_disco_info(self, DiscoInfo):
|
2024-02-28 23:54:48 +01:00
|
|
|
jid = DiscoInfo['from']
|
|
|
|
|
await self['xep_0115'].update_caps(jid=jid)
|
|
|
|
|
# jid_bare = DiscoInfo['from'].bare
|
|
|
|
|
# self.bookmarks_db = buku.BukuDb(dbfile=jid_bare + '.db')
|
|
|
|
|
|
|
|
|
|
def process_message(self, message):
|
2024-04-05 17:00:19 +02:00
|
|
|
Chat.action(self, message)
|
2024-02-28 23:54:48 +01:00
|
|
|
|
|
|
|
|
def command_list(self):
|
|
|
|
|
self['xep_0050'].add_command(node='add',
|
|
|
|
|
name='🔖️ Add',
|
|
|
|
|
handler=self._handle_add)
|
|
|
|
|
self['xep_0050'].add_command(node='random',
|
|
|
|
|
name='🎲️ Random',
|
|
|
|
|
handler=self._handle_random)
|
|
|
|
|
self['xep_0050'].add_command(node='modify',
|
|
|
|
|
name='📑️ Browse',
|
|
|
|
|
handler=self._handle_browse)
|
|
|
|
|
self['xep_0050'].add_command(node='search',
|
|
|
|
|
name='🔍️ Search',
|
|
|
|
|
handler=self._handle_search)
|
|
|
|
|
# self['xep_0050'].add_command(node='tag',
|
|
|
|
|
# name='🏷️ Tags',
|
|
|
|
|
# handler=self._handle_tag)
|
|
|
|
|
# self['xep_0050'].add_command(node='statistics',
|
|
|
|
|
# name='📊️ Statistics',
|
|
|
|
|
# handler=self._handle_stats)
|
|
|
|
|
self['xep_0050'].add_command(node='help',
|
|
|
|
|
name='📔️ Help',
|
|
|
|
|
handler=self._handle_help)
|
|
|
|
|
self['xep_0050'].add_command(node='license',
|
|
|
|
|
name='✒️ License',
|
|
|
|
|
handler=self._handle_license)
|
|
|
|
|
self['xep_0050'].add_command(node='about',
|
|
|
|
|
name='📜️ About',
|
|
|
|
|
handler=self._handle_about)
|
|
|
|
|
|
|
|
|
|
def _handle_add(self, iq, session):
|
|
|
|
|
# jid = session['from'].bare
|
|
|
|
|
# jid_file = jid
|
|
|
|
|
# db_file = config.get_pathname_to_database(jid_file)
|
|
|
|
|
form = self['xep_0004'].make_form('form', 'Add')
|
|
|
|
|
form['instructions'] = 'Add a new bookmark'
|
|
|
|
|
form.add_field(desc='URL to bookmark.',
|
|
|
|
|
ftype='text-single',
|
|
|
|
|
label='URL',
|
|
|
|
|
required=True,
|
|
|
|
|
var='url')
|
|
|
|
|
form.add_field(desc='Title to add manually.',
|
|
|
|
|
ftype='text-single',
|
|
|
|
|
label='Title',
|
|
|
|
|
var='title')
|
|
|
|
|
form.add_field(desc='Description of the bookmark.',
|
|
|
|
|
ftype='text-multi',
|
|
|
|
|
label='Note',
|
|
|
|
|
var='note')
|
|
|
|
|
form.add_field(desc='Comma-separated tags.',
|
|
|
|
|
ftype='text-single',
|
|
|
|
|
label='Tags',
|
|
|
|
|
var='tag')
|
|
|
|
|
form.add_field(desc='Check to disable automatic title fetch.',
|
|
|
|
|
ftype='boolean',
|
|
|
|
|
label='Immutable',
|
|
|
|
|
var='immutable')
|
|
|
|
|
session['has_next'] = True
|
|
|
|
|
session['next'] = self._handle_edit_single
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_edit_single(self, payload, session):
|
|
|
|
|
jid_bare = session['from'].bare
|
2024-04-05 17:00:19 +02:00
|
|
|
bookmarks_db = Configuration.init_db(jid_bare)
|
2024-02-28 23:54:48 +01:00
|
|
|
form = self['xep_0004'].make_form('form', 'Edit')
|
|
|
|
|
form['instructions'] = 'Edit bookmark'
|
|
|
|
|
url = payload['values']['url']
|
|
|
|
|
idx = bookmarks_db.get_rec_id(url)
|
|
|
|
|
if not idx:
|
|
|
|
|
immu = payload['values']['immutable']
|
|
|
|
|
desc = payload['values']['note']
|
|
|
|
|
tags = payload['values']['tag']
|
|
|
|
|
name = payload['values']['title']
|
|
|
|
|
fetch = True if not name else False
|
|
|
|
|
idx = bookmarks_db.add_rec(desc=desc,
|
|
|
|
|
fetch=fetch,
|
|
|
|
|
immutable=immu,
|
|
|
|
|
tags_in=tags,
|
|
|
|
|
url=url)
|
|
|
|
|
bookmark = bookmarks_db.get_rec_by_id(idx)
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='ID #',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=str(idx))
|
2024-02-28 23:54:48 +01:00
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='Title',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[2])
|
2024-02-28 23:54:48 +01:00
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='URL',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[1])
|
2024-02-28 23:54:48 +01:00
|
|
|
form.add_field(ftype='text-multi',
|
|
|
|
|
label='Note',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[4])
|
2024-02-28 23:54:48 +01:00
|
|
|
options = form.add_field(ftype='list-multi',
|
|
|
|
|
label='Tags')
|
2024-04-05 17:00:19 +02:00
|
|
|
for tag in bookmark[3].split(','):
|
2024-02-28 23:54:48 +01:00
|
|
|
options.addOption(tag, tag)
|
|
|
|
|
form.add_field(ftype='boolean',
|
|
|
|
|
label='Immutable',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=str(bookmark[5]))
|
2024-02-28 23:54:48 +01:00
|
|
|
session['allow_complete'] = True
|
|
|
|
|
session['has_next'] = False
|
|
|
|
|
session['next'] = None
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_browse(self, iq, session):
|
|
|
|
|
form = self['xep_0004'].make_form('form', 'Browse')
|
|
|
|
|
form['instructions'] = 'Browse bookmark'
|
|
|
|
|
options = form.add_field(desc='Items per page.',
|
|
|
|
|
label='Items',
|
|
|
|
|
ftype='list-single',
|
|
|
|
|
value='20',
|
|
|
|
|
var='limit')
|
|
|
|
|
i = 10
|
|
|
|
|
while i <= 50:
|
|
|
|
|
x = str(i)
|
|
|
|
|
options.addOption(x, x)
|
|
|
|
|
i += 10
|
|
|
|
|
# options['validate']['datatype'] = 'xs:integer'
|
|
|
|
|
# options['validate']['range'] = { 'minimum': 10, 'maximum': 50 }
|
|
|
|
|
form.add_field(ftype='hidden',
|
|
|
|
|
value='0',
|
|
|
|
|
var='count')
|
|
|
|
|
session['has_next'] = True
|
|
|
|
|
session['next'] = self._handle_browse_all
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_tag(self, iq, session):
|
|
|
|
|
jid_bare = session['from'].bare
|
2024-04-05 17:00:19 +02:00
|
|
|
bookmarks_db = Configuration.init_db(jid_bare)
|
2024-02-28 23:54:48 +01:00
|
|
|
form = self['xep_0004'].make_form('form', 'Tags')
|
|
|
|
|
form['instructions'] = ('Select tags to browse')
|
|
|
|
|
options = form.add_field(desc='Select tag to list its items.',
|
|
|
|
|
ftype='list-single',
|
|
|
|
|
label='Tags',
|
|
|
|
|
var='tag')
|
|
|
|
|
tags = bookmarks_db.get_tag_all()
|
|
|
|
|
counter = 0
|
|
|
|
|
for tag in tags[0]:
|
|
|
|
|
if counter == 100: break
|
|
|
|
|
options.addOption(tag, tag)
|
|
|
|
|
counter += 1
|
|
|
|
|
session['has_next'] = True
|
|
|
|
|
session['next'] = self._handle_browse_all
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_browse_all(self, payload, session):
|
|
|
|
|
jid_bare = session['from'].bare
|
2024-04-05 17:00:19 +02:00
|
|
|
bookmarks_db = Configuration.init_db(jid_bare)
|
2024-02-28 23:54:48 +01:00
|
|
|
vals = payload['values']
|
|
|
|
|
if 'url' in vals and vals['url']:
|
|
|
|
|
act = vals['action']
|
|
|
|
|
url = vals['url']
|
|
|
|
|
form = self['xep_0004'].make_form('form', 'Edit')
|
|
|
|
|
match act:
|
|
|
|
|
case 'edit':
|
|
|
|
|
if len(url) > 1:
|
|
|
|
|
form['instructions'] = 'Modify bookmarks'
|
|
|
|
|
idxs = ''
|
|
|
|
|
tags = ''
|
|
|
|
|
for i in url:
|
|
|
|
|
idx = bookmarks_db.get_rec_id(i)
|
|
|
|
|
idxs += ',' + str(idx)
|
|
|
|
|
bookmark = bookmarks_db.get_rec_by_id(idx)
|
2024-04-05 17:00:19 +02:00
|
|
|
tags += bookmark[3]
|
2024-02-28 23:54:48 +01:00
|
|
|
form.add_field(desc='Comma-separated tags.',
|
|
|
|
|
ftype='text-single',
|
|
|
|
|
label='Tags',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[3],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='tags_new')
|
|
|
|
|
form.add_field(desc='Check to disable automatic title fetch.',
|
|
|
|
|
ftype='boolean',
|
|
|
|
|
label='Immutable',
|
|
|
|
|
value=True,
|
|
|
|
|
var='immutable')
|
|
|
|
|
form.add_field(ftype='hidden',
|
|
|
|
|
value=idxs,
|
|
|
|
|
var='ids')
|
|
|
|
|
# session['next'] = self._handle_edit_single(payload, session)
|
|
|
|
|
else:
|
|
|
|
|
form['instructions'] = 'Modify bookmark'
|
|
|
|
|
idx = bookmarks_db.get_rec_id(url[0])
|
|
|
|
|
bookmark = bookmarks_db.get_rec_by_id(idx)
|
|
|
|
|
form.add_field(ftype='fixed',
|
|
|
|
|
label='ID #',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=str(idx),
|
2024-02-28 23:54:48 +01:00
|
|
|
var='id')
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='Title',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[2],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='title')
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='URL',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[1],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='url')
|
|
|
|
|
form.add_field(ftype='text-multi',
|
|
|
|
|
label='Note',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[4],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='description')
|
|
|
|
|
form.add_field(ftype='hidden',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[3],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='tags_old')
|
|
|
|
|
form.add_field(desc='Comma-separated tags.',
|
|
|
|
|
ftype='text-single',
|
|
|
|
|
label='Tags',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[3],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='tags_new')
|
|
|
|
|
form.add_field(desc='Check to disable automatic title fetch.',
|
|
|
|
|
ftype='boolean',
|
|
|
|
|
label='Immutable',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=str(bookmark[5]),
|
2024-02-28 23:54:48 +01:00
|
|
|
var='immutable')
|
|
|
|
|
case 'remove':
|
|
|
|
|
form['instructions'] = ('The following items were deleted from '
|
|
|
|
|
'bookmarks.\nProceed to finish or '
|
|
|
|
|
'select items to restore.')
|
|
|
|
|
options = form.add_field(desc='Select items to restore',
|
|
|
|
|
ftype='list-multi',
|
|
|
|
|
label='Deleted items',
|
|
|
|
|
var='url')
|
|
|
|
|
for i in url:
|
|
|
|
|
idx = bookmarks_db.get_rec_id(i)
|
|
|
|
|
rec = bookmarks_db.get_rec_by_id(idx)
|
|
|
|
|
bookmarks_db.delete_rec(idx)
|
|
|
|
|
options.addOption(rec.title, i)
|
|
|
|
|
session['cancel'] = self._handle_cancel
|
|
|
|
|
# session['allow_complete'] = True
|
|
|
|
|
session['has_next'] = True
|
|
|
|
|
session['next'] = self._handle_action_result
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
else:
|
|
|
|
|
limit = vals['limit'] if 'limit' in vals else 0
|
|
|
|
|
if isinstance(limit, list): limit = limit[0]
|
|
|
|
|
count = vals['count'] if 'count' in vals else 0
|
|
|
|
|
if isinstance(count, list): count = count[0]
|
|
|
|
|
form = self['xep_0004'].make_form('form', 'Browse')
|
|
|
|
|
form['instructions'] = ('Select bookmarks to modify')
|
|
|
|
|
options = form.add_field(desc='Select an action',
|
|
|
|
|
ftype='list-single',
|
|
|
|
|
label='Action',
|
|
|
|
|
required=True,
|
|
|
|
|
value='edit',
|
|
|
|
|
var='action')
|
|
|
|
|
options.addOption('Edit', 'edit')
|
|
|
|
|
options.addOption('Remove', 'remove')
|
|
|
|
|
options = form.add_field(desc='Selection of several bookmarks will '
|
|
|
|
|
'only allow to modify tags.',
|
|
|
|
|
ftype='list-multi',
|
|
|
|
|
label='Bookmark',
|
|
|
|
|
var='url')
|
|
|
|
|
bookmarks = bookmarks_db.get_rec_all()
|
|
|
|
|
# bookmarks = sorted(bookmarks, key=lambda x: x.title)
|
|
|
|
|
counter = int(count)
|
|
|
|
|
limiter = counter + int(limit)
|
|
|
|
|
for bookmark in bookmarks[counter:limiter]:
|
|
|
|
|
if counter == limiter: break
|
2024-04-05 17:00:19 +02:00
|
|
|
options.addOption(bookmark[2], bookmark[1])
|
2024-02-28 23:54:48 +01:00
|
|
|
counter += 1
|
|
|
|
|
form.add_field(ftype='hidden',
|
|
|
|
|
value=str(counter),
|
|
|
|
|
var='count')
|
|
|
|
|
form.add_field(ftype='hidden',
|
|
|
|
|
value=str(limit),
|
|
|
|
|
var='limit')
|
|
|
|
|
session['has_next'] = True
|
|
|
|
|
session['next'] = self._handle_browse_all
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_action_result(self, payload, session):
|
|
|
|
|
jid_bare = session['from'].bare
|
2024-04-05 17:00:19 +02:00
|
|
|
bookmarks_db = Configuration.init_db(jid_bare)
|
2024-02-28 23:54:48 +01:00
|
|
|
vals = payload['values']
|
|
|
|
|
if 'id' in vals:
|
|
|
|
|
idx = vals['id']
|
|
|
|
|
idx = int(idx)
|
|
|
|
|
tags_new = vals['tags_new']
|
|
|
|
|
tags_new = tags_new.replace(' ,', ',')
|
|
|
|
|
tags_new = tags_new.replace(', ', ',')
|
|
|
|
|
tags_old = vals['tags_old'][0]
|
|
|
|
|
tags = tags_old.split(',')
|
|
|
|
|
tags_to_remove = '-,'
|
|
|
|
|
for tag in tags:
|
|
|
|
|
if tag not in tags_new:
|
|
|
|
|
tags_to_remove += ',' + tag
|
|
|
|
|
bookmarks_db.update_rec(idx,
|
|
|
|
|
url=vals['url'],
|
|
|
|
|
title_in=vals['title'],
|
|
|
|
|
tags_in='+,' + tags_new,
|
|
|
|
|
desc=vals['description'],
|
|
|
|
|
immutable=vals['immutable'])
|
|
|
|
|
bookmarks_db.update_rec(idx,
|
|
|
|
|
tags_in=tags_to_remove)
|
|
|
|
|
form = self['xep_0004'].make_form('result', 'Done')
|
|
|
|
|
rec = bookmarks_db.get_rec_by_id(idx)
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='Title',
|
|
|
|
|
value=rec.title)
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='URL',
|
|
|
|
|
value=rec.url)
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='Note',
|
|
|
|
|
value=rec.desc)
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='Tags',
|
|
|
|
|
value=rec.tags_raw)
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='Immutable',
|
|
|
|
|
value=str(rec.flags))
|
|
|
|
|
elif 'ids' in vals:
|
|
|
|
|
immutable = vals['immutable']
|
|
|
|
|
tags_new = vals['tags_new']
|
|
|
|
|
tags_new = tags_new.replace(' ,', ',')
|
|
|
|
|
tags_new = tags_new.replace(', ', ',')
|
|
|
|
|
idxs = vals['ids'].split(',')
|
|
|
|
|
for idx in idxs:
|
|
|
|
|
bookmarks_db.update_rec(idx,
|
|
|
|
|
tags_in='+,' + tags_new,
|
|
|
|
|
immutable=immutable)
|
|
|
|
|
form = self['xep_0004'].make_form('result', 'Done')
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='Tags',
|
|
|
|
|
value=tags_new)
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='Immutable',
|
|
|
|
|
value=immutable)
|
|
|
|
|
elif 'url' in vals:
|
|
|
|
|
url = vals['url']
|
|
|
|
|
form = self['xep_0004'].make_form('result', 'Add')
|
|
|
|
|
form['instructions'] = ('The following items have been added to '
|
|
|
|
|
'the bookmarks\nNote: You will have to '
|
|
|
|
|
'manually tag these items, if you would.')
|
|
|
|
|
for i in url:
|
|
|
|
|
idx = bookmarks_db.add_rec(url=i)
|
|
|
|
|
rec = bookmarks_db.get_rec_by_id(idx)
|
|
|
|
|
form.add_field(ftype='fixed',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=str(idx))
|
2024-02-28 23:54:48 +01:00
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
value=rec.title)
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
value=rec.url)
|
|
|
|
|
session['allow_complete'] = True
|
|
|
|
|
session['has_next'] = False
|
|
|
|
|
session['next'] = None
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_random(self, iq, session):
|
|
|
|
|
jid_bare = session['from'].bare
|
2024-04-05 17:00:19 +02:00
|
|
|
bookmarks_db = Configuration.init_db(jid_bare)
|
2024-02-28 23:54:48 +01:00
|
|
|
bookmarks = bookmarks_db.get_rec_all()
|
|
|
|
|
if bookmarks:
|
|
|
|
|
import random
|
|
|
|
|
bookmark = random.choice(bookmarks)
|
|
|
|
|
form = self['xep_0004'].make_form('form', 'Random')
|
2024-04-05 17:00:19 +02:00
|
|
|
form['instructions'] = 'Bookmark #{}'.format(bookmark[0])
|
2024-02-28 23:54:48 +01:00
|
|
|
form.add_field(ftype='fixed',
|
|
|
|
|
# ftype='text-single',
|
|
|
|
|
label='URL',
|
|
|
|
|
# required=True,
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[1],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='url')
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='Title',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[2],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='title')
|
|
|
|
|
form.add_field(ftype='text-multi',
|
|
|
|
|
label='Note',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[4],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='note')
|
|
|
|
|
form.add_field(desc='Comma-separated tags.',
|
|
|
|
|
ftype='text-single',
|
|
|
|
|
label='Tags',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[3],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='tag')
|
|
|
|
|
form.add_field(desc='Check to disable automatic title fetch.',
|
|
|
|
|
ftype='boolean',
|
|
|
|
|
label='Immutable',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=str(bookmark[5]),
|
2024-02-28 23:54:48 +01:00
|
|
|
var='immutable')
|
|
|
|
|
session['has_next'] = True
|
|
|
|
|
session['next'] = self._handle_edit_single
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
else:
|
|
|
|
|
text_note = ('There are no bookmarks, yet.')
|
|
|
|
|
session['notes'] = [['info', text_note]]
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_search(self, iq, session):
|
|
|
|
|
form = self['xep_0004'].make_form('form', 'Search')
|
|
|
|
|
form['instructions'] = 'Search for bookmarks'
|
|
|
|
|
form.add_field(desc='Enter a search term to query.',
|
|
|
|
|
ftype='text-single',
|
|
|
|
|
label='Search',
|
|
|
|
|
var='query')
|
|
|
|
|
options = form.add_field(desc='Select type of search.',
|
|
|
|
|
ftype='list-single',
|
|
|
|
|
label='Search by',
|
|
|
|
|
value='any',
|
|
|
|
|
var='type')
|
|
|
|
|
options.addOption('All keywords', 'all')
|
|
|
|
|
options.addOption('Any keyword', 'any')
|
|
|
|
|
options.addOption('Tag', 'tag')
|
|
|
|
|
form.add_field(desc='Search for matching substrings.',
|
|
|
|
|
ftype='boolean',
|
|
|
|
|
label='Deep',
|
|
|
|
|
value=True,
|
|
|
|
|
var='deep')
|
|
|
|
|
form.add_field(desc='Match a regular expression.',
|
|
|
|
|
ftype='boolean',
|
|
|
|
|
label='Regular Expression',
|
|
|
|
|
var='regex')
|
|
|
|
|
session['allow_prev'] = False
|
|
|
|
|
session['has_next'] = True
|
|
|
|
|
session['next'] = self._handle_search_result
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
session['prev'] = None
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_search_result(self, payload, session):
|
|
|
|
|
jid_bare = session['from'].bare
|
2024-04-05 17:00:19 +02:00
|
|
|
bookmarks_db = Configuration.init_db(jid_bare)
|
2024-02-28 23:54:48 +01:00
|
|
|
vals = payload['values']
|
|
|
|
|
if 'url' in vals and vals['url']:
|
|
|
|
|
act = vals['action']
|
|
|
|
|
url = vals['url']
|
|
|
|
|
form = self['xep_0004'].make_form('form', 'Edit')
|
|
|
|
|
match act:
|
|
|
|
|
case 'edit':
|
|
|
|
|
if len(url) > 1:
|
|
|
|
|
form['instructions'] = 'Modify bookmarks'
|
|
|
|
|
idxs = ''
|
|
|
|
|
tags = ''
|
|
|
|
|
for i in url:
|
|
|
|
|
idx = bookmarks_db.get_rec_id(i)
|
|
|
|
|
idxs += ',' + str(idx)
|
|
|
|
|
bookmark = bookmarks_db.get_rec_by_id(idx)
|
2024-04-05 17:00:19 +02:00
|
|
|
tags += bookmark[3]
|
2024-02-28 23:54:48 +01:00
|
|
|
form.add_field(desc='Comma-separated tags.',
|
|
|
|
|
ftype='text-single',
|
|
|
|
|
label='Tags',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[3],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='tags_new')
|
|
|
|
|
form.add_field(desc='Check to disable automatic title fetch.',
|
|
|
|
|
ftype='boolean',
|
|
|
|
|
label='Immutable',
|
|
|
|
|
value=True,
|
|
|
|
|
var='immutable')
|
|
|
|
|
form.add_field(ftype='hidden',
|
|
|
|
|
value=idxs,
|
|
|
|
|
var='ids')
|
|
|
|
|
# session['next'] = self._handle_edit_single(payload, session)
|
|
|
|
|
else:
|
|
|
|
|
form['instructions'] = 'Modify bookmark'
|
|
|
|
|
idx = bookmarks_db.get_rec_id(url[0])
|
|
|
|
|
bookmark = bookmarks_db.get_rec_by_id(idx)
|
|
|
|
|
form.add_field(ftype='fixed',
|
|
|
|
|
label='ID #',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=str(idx),
|
2024-02-28 23:54:48 +01:00
|
|
|
var='id')
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='Title',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[2],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='title')
|
|
|
|
|
form.add_field(ftype='text-single',
|
|
|
|
|
label='URL',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[1],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='url')
|
|
|
|
|
form.add_field(ftype='text-multi',
|
|
|
|
|
label='Note',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[4],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='description')
|
|
|
|
|
form.add_field(ftype='hidden',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[3],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='tags_old')
|
|
|
|
|
form.add_field(desc='Comma-separated tags.',
|
|
|
|
|
ftype='text-single',
|
|
|
|
|
label='Tags',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=bookmark[3],
|
2024-02-28 23:54:48 +01:00
|
|
|
var='tags_new')
|
|
|
|
|
form.add_field(desc='Check to disable automatic title fetch.',
|
|
|
|
|
ftype='boolean',
|
|
|
|
|
label='Immutable',
|
2024-04-05 17:00:19 +02:00
|
|
|
value=str(bookmark[5]),
|
2024-02-28 23:54:48 +01:00
|
|
|
var='immutable')
|
|
|
|
|
case 'remove':
|
|
|
|
|
form['instructions'] = ('The following items were deleted from '
|
|
|
|
|
'bookmarks.\nProceed to finish or '
|
|
|
|
|
'select items to restore.')
|
|
|
|
|
options = form.add_field(desc='Select items to restore',
|
|
|
|
|
ftype='list-multi',
|
|
|
|
|
label='Deleted items',
|
|
|
|
|
var='url')
|
|
|
|
|
for i in url:
|
|
|
|
|
idx = bookmarks_db.get_rec_id(i)
|
|
|
|
|
rec = bookmarks_db.get_rec_by_id(idx)
|
|
|
|
|
bookmarks_db.delete_rec(idx)
|
|
|
|
|
options.addOption(rec.title, i)
|
|
|
|
|
session['cancel'] = self._handle_cancel
|
|
|
|
|
# session['allow_complete'] = True
|
|
|
|
|
session['has_next'] = True
|
|
|
|
|
session['next'] = self._handle_action_result
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
else:
|
|
|
|
|
count = vals['count'] if 'count' in vals else 0
|
|
|
|
|
if isinstance(count, list): count = count[0]
|
|
|
|
|
# count = count if count else 0
|
|
|
|
|
query = vals['query']
|
|
|
|
|
if isinstance(query, list): query = query[0]
|
|
|
|
|
stype = vals['type']
|
|
|
|
|
if isinstance(stype, list): stype = stype[0]
|
|
|
|
|
deep = vals['deep']
|
|
|
|
|
if isinstance(deep, list): deep = deep[0]
|
|
|
|
|
deep = True if '1' else False
|
|
|
|
|
regex = vals['regex']
|
|
|
|
|
if isinstance(regex, list): regex = regex[0]
|
|
|
|
|
regex = True if '1' else False
|
|
|
|
|
match stype:
|
|
|
|
|
case 'all':
|
|
|
|
|
bookmarks = bookmarks_db.searchdb(query,
|
|
|
|
|
all_keywords=True,
|
|
|
|
|
deep=deep,
|
|
|
|
|
regex=regex)
|
|
|
|
|
case 'any':
|
|
|
|
|
bookmarks = bookmarks_db.searchdb(query,
|
|
|
|
|
all_keywords=False,
|
|
|
|
|
deep=deep,
|
|
|
|
|
regex=regex)
|
|
|
|
|
case 'tag':
|
|
|
|
|
bookmarks = bookmarks_db.search_by_tag(query)
|
|
|
|
|
# bookmarks = sorted(bookmarks, key=lambda x: x.title)
|
|
|
|
|
if bookmarks:
|
|
|
|
|
form = self['xep_0004'].make_form('form', 'Browse')
|
|
|
|
|
form['instructions'] = ('Select bookmarks to modify')
|
|
|
|
|
options = form.add_field(desc='Select an action',
|
|
|
|
|
ftype='list-single',
|
|
|
|
|
label='Action',
|
|
|
|
|
required=True,
|
|
|
|
|
value='edit',
|
|
|
|
|
var='action')
|
|
|
|
|
options.addOption('Edit', 'edit')
|
|
|
|
|
options.addOption('Remove', 'remove')
|
|
|
|
|
options = form.add_field(desc='Selection of several bookmarks '
|
|
|
|
|
'will only allow to modify tags.',
|
|
|
|
|
ftype='list-multi',
|
|
|
|
|
label='Bookmark',
|
|
|
|
|
var='url')
|
|
|
|
|
counter = int(count)
|
|
|
|
|
limiter = counter + 10
|
|
|
|
|
for bookmark in bookmarks[counter:limiter]:
|
|
|
|
|
if counter == limiter: break
|
2024-04-05 17:00:19 +02:00
|
|
|
options.addOption(bookmark[2], bookmark[1])
|
2024-02-28 23:54:48 +01:00
|
|
|
counter += 1
|
|
|
|
|
form.add_field(ftype='hidden',
|
|
|
|
|
value=str(counter),
|
|
|
|
|
var='count')
|
|
|
|
|
form.add_field(ftype='hidden',
|
|
|
|
|
value=query,
|
|
|
|
|
var='query')
|
|
|
|
|
form.add_field(ftype='hidden',
|
|
|
|
|
value=stype,
|
|
|
|
|
var='type')
|
|
|
|
|
deep = '1'if deep else ''
|
|
|
|
|
form.add_field(ftype='hidden',
|
|
|
|
|
value=deep,
|
|
|
|
|
var='deep')
|
|
|
|
|
regex = '1'if regex else ''
|
|
|
|
|
form.add_field(ftype='hidden',
|
|
|
|
|
value=regex,
|
|
|
|
|
var='regex')
|
|
|
|
|
session['has_next'] = True
|
|
|
|
|
session['next'] = self._handle_search_result
|
|
|
|
|
session['payload'] = form
|
|
|
|
|
else:
|
|
|
|
|
text_note = 'No results were yielded for: {}'.format(query)
|
|
|
|
|
session['notes'] = [['info', text_note]]
|
|
|
|
|
session['allow_prev'] = True
|
|
|
|
|
session['prev'] = self._handle_search
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_cancel(self, payload, session):
|
|
|
|
|
text_note = ('Operation has been cancelled.'
|
|
|
|
|
'\n\n'
|
|
|
|
|
'No action was taken.')
|
|
|
|
|
session['notes'] = [['info', text_note]]
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_help(self, iq, session):
|
|
|
|
|
text_note = Documentation.commands()
|
|
|
|
|
session['notes'] = [['info', text_note]]
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_license(self, iq, session):
|
|
|
|
|
text_note = Documentation.notice()
|
|
|
|
|
session['notes'] = [['info', text_note]]
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
def _handle_about(self, iq, session):
|
|
|
|
|
text_note = Documentation.about()
|
|
|
|
|
session['notes'] = [['info', text_note]]
|
|
|
|
|
return session
|
|
|
|
|
|
|
|
|
|
|
