Allow specifying file name and format as cmdline arguments for tmuxp freeze, deduce format from filename

Author: dnaaun

tmuxp/cli.py

@@ -911,8 +911,25 @@ def command_shell(
 @click.argument('session_name', nargs=1, required=False)
 @click.option('-S', 'socket_path', help='pass-through for tmux -S')
 @click.option('-L', 'socket_name', help='pass-through for tmux -L')
+@click.option(
+    '-f',
+    '--config-format',
+    type=click.Choice(['yaml', 'json']),
+    help='format to save in',
+)
+@click.option('-o', '--save-to', type=click.Path(exists=False), help='file to save to')
+@click.option(
+    '-y',
+    '--yes',
+    type=bool,
+    is_flag=True,
+    default=False,
+    help="Don't prompt for confirmation",
+)
 @click.option('--force', 'force', help='overwrite the config file', is_flag=True)
-def command_freeze(session_name, socket_name, socket_path, force):
+def command_freeze(
+    session_name, socket_name, config_format, save_to, socket_path, yes, force
+):
     """Snapshot a session into a config.
 
     If SESSION_NAME is provided, snapshot that session. Otherwise, use the
@@ -936,62 +953,81 @@ def command_freeze(session_name, socket_name, socket_path, force):
     configparser = kaptan.Kaptan()
     newconfig = config.inline(sconf)
     configparser.import_config(newconfig)
-    config_format = click.prompt(
-        'Convert to', value_proc=_validate_choices(['yaml', 'json']), default='yaml'
-    )
-
-    if config_format == 'yaml':
-        newconfig = configparser.export(
-            'yaml', indent=2, default_flow_style=False, safe=True
-        )
-    elif config_format == 'json':
-        newconfig = configparser.export('json', indent=2)
-    else:
-        sys.exit('Unknown config format.')
 
     print(
         '---------------------------------------------------------------'
         '\n'
         'Freeze does its best to snapshot live tmux sessions.\n'
     )
-    if click.confirm(
-        'The new config *WILL* require adjusting afterwards. Save config?'
+    if not (
+        yes
+        or click.confirm(
+            'The new config *WILL* require adjusting afterwards. Save config?'
+        )
     ):
-        dest = None
-        while not dest:
-            save_to = os.path.abspath(
-                os.path.join(
-                    get_config_dir(),
-                    '%s.%s' % (sconf.get('session_name'), config_format),
-                )
-            )
-            dest_prompt = click.prompt(
-                'Path to new config:', value_proc=get_abs_path, default=save_to
-            )
-            if not force and os.path.exists(dest_prompt):
-                print('%s exists. Pick a new filename.' % dest_prompt)
-                continue
-
-            dest = dest_prompt
-
-        dest = os.path.abspath(os.path.relpath(os.path.expanduser(dest)))
-        if click.confirm('Save to %s?' % dest):
-            destdir = os.path.dirname(dest)
-            if not os.path.isdir(destdir):
-                os.makedirs(destdir)
-            buf = open(dest, 'w')
-            buf.write(newconfig)
-            buf.close()
-
-            print('Saved to %s.' % dest)
-    else:
         print(
             'tmuxp has examples in JSON and YAML format at '
             '<http://tmuxp.git-pull.com/examples.html>\n'
             'View tmuxp docs at <http://tmuxp.git-pull.com/>.'
         )
         sys.exit()
 
+    dest = save_to
+    while not dest:
+        save_to = os.path.abspath(
+            os.path.join(
+                get_config_dir(),
+                '%s.%s' % (sconf.get('session_name'), config_format or 'yaml'),
+            )
+        )
+        dest_prompt = click.prompt(
+            'Save to: %s' % save_to, value_proc=get_abs_path, default=save_to
+        )
+        if not force and os.path.exists(dest_prompt):
+            print('%s exists. Pick a new filename.' % dest_prompt)
+            continue
+
+        dest = dest_prompt
+    dest = os.path.abspath(os.path.relpath(os.path.expanduser(dest)))
+
+    if config_format is None:
+        valid_config_formats = ['json', 'yaml']
+        _, config_format = os.path.splitext(dest)
+        config_format = config_format[1:].lower()
+        if config_format not in valid_config_formats:
+            config_format = click.prompt(
+                "Couldn't ascertain one of [%s] from file name. Convert to"
+                % ", ".join(valid_config_formats),
+                value_proc=_validate_choices(['yaml', 'json']),
+                default='yaml',
+            )
+
+    if config_format == 'yaml':
+        newconfig = configparser.export(
+            'yaml', indent=2, default_flow_style=False, safe=True
+        )
+    elif config_format == 'json':
+        newconfig = configparser.export('json', indent=2)
+
+    if yes or click.confirm('Save to %s?' % dest):
+        destdir = os.path.dirname(dest)
+        if not os.path.isdir(destdir):
+            os.makedirs(destdir)
+        buf = open(dest, 'w')
+        buf.write(newconfig)
+        buf.close()
+
+        print('Saved to %s.' % dest)
+
+    if config_format == 'yaml':
+        newconfig = configparser.export(
+            'yaml', indent=2, default_flow_style=False, safe=True
+        )
+    elif config_format == 'json':
+        newconfig = configparser.export('json', indent=2)
+    else:
+        sys.exit('Unknown config format.')
+
 
 @cli.command(name='load', short_help='Load tmuxp workspaces.')
 @click.pass_context