|
3137
|
1 #!/usr/bin/env python3 |
|
|
2 |
|
986
|
3 |
|
|
4 # jp: a SAT command line tool |
|
3136
|
5 # Copyright (C) 2009-2020 Jérôme Poisson (goffi@goffi.org) |
|
986
|
6 |
|
|
7 # This program is free software: you can redistribute it and/or modify |
|
|
8 # it under the terms of the GNU Affero General Public License as published by |
|
|
9 # the Free Software Foundation, either version 3 of the License, or |
|
|
10 # (at your option) any later version. |
|
|
11 |
|
|
12 # This program is distributed in the hope that it will be useful, |
|
|
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
|
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
|
15 # GNU Affero General Public License for more details. |
|
|
16 |
|
|
17 # You should have received a copy of the GNU Affero General Public License |
|
|
18 # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
|
19 |
|
3028
|
20 from . import base |
|
986
|
21 from sat.core.i18n import _ |
|
3040
|
22 from sat_frontends.jp.constants import Const as C |
|
986
|
23 |
|
|
24 __commands__ = ["Bookmarks"] |
|
|
25 |
|
|
26 STORAGE_LOCATIONS = ('local', 'private', 'pubsub') |
|
|
27 TYPES = ('muc', 'url') |
|
|
28 |
|
|
29 class BookmarksCommon(base.CommandBase): |
|
|
30 """Class used to group common options of bookmarks subcommands""" |
|
|
31 |
|
|
32 def add_parser_options(self, location_default='all'): |
|
|
33 self.parser.add_argument('-l', '--location', type=str, choices=(location_default,) + STORAGE_LOCATIONS, default=location_default, help=_("storage location (default: %(default)s)")) |
|
|
34 self.parser.add_argument('-t', '--type', type=str, choices=TYPES, default=TYPES[0], help=_("bookmarks type (default: %(default)s)")) |
|
|
35 |
|
|
36 class BookmarksList(BookmarksCommon): |
|
|
37 |
|
|
38 def __init__(self, host): |
|
|
39 super(BookmarksList, self).__init__(host, 'list', help=_('list bookmarks')) |
|
|
40 |
|
3040
|
41 async def start(self): |
|
|
42 try: |
|
|
43 data = await self.host.bridge.bookmarksList( |
|
|
44 self.args.type, self.args.location, self.host.profile) |
|
|
45 except Exception as e: |
|
|
46 self.disp(f"can't get bookmarks list: {e}", error=True) |
|
|
47 self.host.quit(C.EXIT_BRIDGE_ERRBACK) |
|
|
48 |
|
986
|
49 mess = [] |
|
|
50 for location in STORAGE_LOCATIONS: |
|
|
51 if not data[location]: |
|
|
52 continue |
|
|
53 loc_mess = [] |
|
3040
|
54 loc_mess.append(f"{location}:") |
|
986
|
55 book_mess = [] |
|
3028
|
56 for book_link, book_data in list(data[location].items()): |
|
986
|
57 name = book_data.get('name') |
|
|
58 autojoin = book_data.get('autojoin', 'false') == 'true' |
|
|
59 nick = book_data.get('nick') |
|
3028
|
60 book_mess.append("\t%s[%s%s]%s" % ((name+' ') if name else '', |
|
986
|
61 book_link, |
|
3028
|
62 ' (%s)' % nick if nick else '', |
|
|
63 ' (*)' if autojoin else '')) |
|
|
64 loc_mess.append('\n'.join(book_mess)) |
|
|
65 mess.append('\n'.join(loc_mess)) |
|
986
|
66 |
|
3028
|
67 print('\n\n'.join(mess)) |
|
3040
|
68 self.host.quit() |
|
986
|
69 |
|
|
70 |
|
|
71 class BookmarksRemove(BookmarksCommon): |
|
|
72 |
|
|
73 def __init__(self, host): |
|
|
74 super(BookmarksRemove, self).__init__(host, 'remove', help=_('remove a bookmark')) |
|
|
75 |
|
|
76 def add_parser_options(self): |
|
|
77 super(BookmarksRemove, self).add_parser_options() |
|
3040
|
78 self.parser.add_argument( |
|
|
79 'bookmark', help=_('jid (for muc bookmark) or url of to remove')) |
|
|
80 self.parser.add_argument( |
|
|
81 "-f", "--force", action="store_true", |
|
|
82 help=_("delete bookmark without confirmation"),) |
|
|
83 |
|
|
84 async def start(self): |
|
|
85 if not self.args.force: |
|
|
86 await self.host.confirmOrQuit(_("Are you sure to delete this bookmark?")) |
|
986
|
87 |
|
3040
|
88 try: |
|
|
89 await self.host.bridge.bookmarksRemove( |
|
|
90 self.args.type, self.args.bookmark, self.args.location, self.host.profile) |
|
|
91 except Exception as e: |
|
|
92 self.disp(_(f"can't delete bookmark: {e}"), error=True) |
|
|
93 self.host.quit(C.EXIT_BRIDGE_ERRBACK) |
|
|
94 else: |
|
|
95 self.disp(_('bookmark deleted')) |
|
|
96 self.host.quit() |
|
986
|
97 |
|
|
98 |
|
|
99 class BookmarksAdd(BookmarksCommon): |
|
|
100 |
|
|
101 def __init__(self, host): |
|
|
102 super(BookmarksAdd, self).__init__(host, 'add', help=_('add a bookmark')) |
|
|
103 |
|
|
104 def add_parser_options(self): |
|
|
105 super(BookmarksAdd, self).add_parser_options(location_default='auto') |
|
3040
|
106 self.parser.add_argument( |
|
|
107 'bookmark', help=_('jid (for muc bookmark) or url of to remove')) |
|
3028
|
108 self.parser.add_argument('-n', '--name', help=_("bookmark name")) |
|
986
|
109 muc_group = self.parser.add_argument_group(_('MUC specific options')) |
|
3028
|
110 muc_group.add_argument('-N', '--nick', help=_('nickname')) |
|
3040
|
111 muc_group.add_argument( |
|
|
112 '-a', '--autojoin', action='store_true', |
|
|
113 help=_('join room on profile connection')) |
|
986
|
114 |
|
3040
|
115 async def start(self): |
|
986
|
116 if self.args.type == 'url' and (self.args.autojoin or self.args.nick is not None): |
|
3040
|
117 self.parser.error(_("You can't use --autojoin or --nick with --type url")) |
|
986
|
118 data = {} |
|
|
119 if self.args.autojoin: |
|
|
120 data['autojoin'] = 'true' |
|
|
121 if self.args.nick is not None: |
|
|
122 data['nick'] = self.args.nick |
|
|
123 if self.args.name is not None: |
|
|
124 data['name'] = self.args.name |
|
3040
|
125 try: |
|
|
126 await self.host.bridge.bookmarksAdd( |
|
|
127 self.args.type, self.args.bookmark, data, self.args.location, |
|
|
128 self.host.profile) |
|
|
129 except Exception as e: |
|
|
130 self.disp(f"can't add bookmark: {e}", error=True) |
|
|
131 self.host.quit(C.EXIT_BRIDGE_ERRBACK) |
|
|
132 else: |
|
|
133 self.disp(_('bookmark successfully added')) |
|
|
134 self.host.quit() |
|
986
|
135 |
|
|
136 |
|
|
137 class Bookmarks(base.CommandBase): |
|
|
138 subcommands = (BookmarksList, BookmarksRemove, BookmarksAdd) |
|
|
139 |
|
|
140 def __init__(self, host): |
|
|
141 super(Bookmarks, self).__init__(host, 'bookmarks', use_profile=False, help=_('manage bookmarks')) |
