From 9d5cf7e33a5cfe3b4a6b1414b56d9598e2f9422e Mon Sep 17 00:00:00 2001
From: Alex David <alex.david@originate.com>
Date: Sat, 18 Apr 2020 10:09:38 -0700
Subject: [PATCH] Support list or string for default_command config

---
 AUTHORS.rst       |  1 +
 tests/test_cli.py | 21 ++++++++-------------
 todoman/cli.py    | 16 +++++-----------
 3 files changed, 14 insertions(+), 24 deletions(-)

diff --git a/AUTHORS.rst b/AUTHORS.rst
index 53799d13..4275421b 100644
--- a/AUTHORS.rst
+++ b/AUTHORS.rst
@@ -11,6 +11,7 @@ when copying code from elsewhere).
 Authors are listed in alphabetical order.
 
 * Anubha Agrawal <anubhakannu@gmail.com>
+* Alex David <git@alexd.me>
 * Ben Moran <github@benmoran.net>
 * Benjamin Frank <ben+gitlab@pipsfrank.de>
 * Christian Geier <geier@lostpackets.de>
diff --git a/tests/test_cli.py b/tests/test_cli.py
index c294436a..7a03a928 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -847,28 +847,23 @@ def test_status_filtering(runner, todo_factory):
     assert 'two' in result.output
 
 
-def test_invoke_command(runner, tmpdir):
+def test_default_command_string(runner, tmpdir):
     path = tmpdir.join('config')
     path.write('default_command = flush\n', 'a')
 
-    flush = mock.MagicMock()
-    with patch.dict(cli.commands, values=dict(flush=flush)):
-        result = runner.invoke(cli, catch_exceptions=False)
-
-    assert not result.exception
-    assert not result.output.strip()
-    assert flush.call_count == 1
+    result = runner.invoke(cli, catch_exceptions=False)
+    assert (
+        'Are you sure you want to delete all done tasks?' in result.output
+    )
 
 
-def test_invoke_invalid_command(runner, tmpdir):
+def test_default_command_list(runner, tmpdir):
     path = tmpdir.join('config')
-    path.write('default_command = DoTheRobot\n', 'a')
+    path.write('default_command = show, 1234\n', 'a')
 
     result = runner.invoke(cli, catch_exceptions=False)
-
-    assert result.exception
     assert (
-        'Error: Invalid setting for [main][default_command]' in result.output
+        'No todo with id 1234.' in result.output
     )
 
 
diff --git a/todoman/cli.py b/todoman/cli.py
index 06e64f69..ab7c465e 100644
--- a/todoman/cli.py
+++ b/todoman/cli.py
@@ -307,19 +307,13 @@ def cli(click_ctx, colour, porcelain, humanize, config):
     locale.setlocale(locale.LC_TIME, "")
 
     if not click_ctx.invoked_subcommand:
-        invoke_command(
-            click_ctx,
-            ctx.config['main']['default_command'],
-        )
+        invoke_command(click_ctx, ctx.config['main']['default_command'])
 
 
-def invoke_command(click_ctx, command):
-    name, *args = command.split(' ')
-    if name not in cli.commands:
-        raise click.ClickException(
-            'Invalid setting for [main][default_command]'
-        )
-    click_ctx.invoke(cli.commands[command], args)
+def invoke_command(click_ctx, argv):
+    if not isinstance(argv, type([])):
+        argv = [argv]
+    cli.invoke(cli.make_context('default_command', argv, click_ctx))
 
 
 try:  # pragma: no cover