togglemimepart: switch between html and plaintext.

Author: ryneeverett

alot/commands/thread.py

@@ -514,6 +514,11 @@ async def apply(self, ui):
     forced={'mimetree': 'toggle'},
     arguments=[(['query'], {'help': 'query used to filter messages to affect',
                             'nargs': '*'})])
+@registerCommand(
+    MODE, 'togglemimepart', help='switch between html and plain text message',
+    forced={'mimepart': 'toggle'},
+    arguments=[(['query'], {'help': 'query used to filter messages to affect',
+                            'nargs': '*'})])
 class ChangeDisplaymodeCommand(Command):
 
     """fold or unfold messages"""
@@ -590,7 +595,14 @@ def matches(msgt):
             all_headers = not mt.display_all_headers \
                 if self.all_headers == 'toggle' else self.all_headers
             if self.mimepart:
-                mt.set_mimepart(ui.get_deep_focus().mimepart)
+                if self.mimepart == 'toggle':
+                    message = mt.get_message()
+                    mimetype = {'plain': 'html', 'html': 'plain'}[
+                        message.mime_part.get_content_subtype()]
+                    mimepart = get_body_part(message.get_email(), mimetype)
+                elif self.mimepart is True:
+                    mimepart = ui.get_deep_focus().mimepart
+                mt.set_mimepart(mimepart)
             if self.mimetree == 'toggle':
                 tbuffer.focus_selected_message()
             mimetree = not mt.display_mimetree \

alot/completion/command.py

@@ -194,7 +194,8 @@ def f(completed, pos):
                 elif self.mode == 'thread' and cmd in ['fold', 'unfold',
                                                        'togglesource',
                                                        'toggleheaders',
-                                                       'togglemimetree']:
+                                                       'togglemimetree',
+                                                       'togglemimepart']:
                     res = self._querycompleter.complete(params, localpos)
                 elif self.mode == 'thread' and cmd in ['tag', 'retag', 'untag',
                                                        'toggletags']: