From 2399a040bbd8cb1345b0c4b2dcc6f9ef0d79ff09 Mon Sep 17 00:00:00 2001
From: Tomas Babej <tomasbabej@gmail.com>
Date: Sun, 22 Mar 2015 19:22:26 +0100
Subject: [PATCH] taskwiki: Add TaskWikiCalendar command

---
 ftplugin/vimwiki.vim |  1 +
 taskwiki/taskwiki.py | 13 +++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/ftplugin/vimwiki.vim b/ftplugin/vimwiki.vim
index 607384b..94dc170 100644
--- a/ftplugin/vimwiki.vim
+++ b/ftplugin/vimwiki.vim
@@ -10,6 +10,7 @@ augroup END
 command! -nargs=* TaskWikiProjects :py SplitProjects(<q-args>).execute()
 command! -nargs=* TaskWikiProjectsSummary :py SplitSummary(<q-args>).execute()
 command! -nargs=* TaskWikiBurndown :py SplitBurndown(<q-args>).execute()
+command! -nargs=* TaskWikiCalendar :py SplitCalendar(<q-args>).execute()
 
 command! -range TaskWikiInfo :<line1>,<line2>py SelectedTasks().info()
 command! -range TaskWikiLink :<line1>,<line2>py SelectedTasks().link()
diff --git a/taskwiki/taskwiki.py b/taskwiki/taskwiki.py
index 82613ba..3c89e20 100644
--- a/taskwiki/taskwiki.py
+++ b/taskwiki/taskwiki.py
@@ -116,6 +116,19 @@ class SplitBurndown(Split):
     maxwidth = True
 
 
+class SplitCalendar(Split):
+    command = 'calendar'
+    colorful = True
+    maxwidth = True
+
+    # Task calendar does not take fitler and in general uses
+    # command-suffix syntax
+    def __init__(self, args):
+        self.args = []
+        self.tw_extra_args = util.tw_modstring_to_args(args)
+        self.split_name = self.split_name or self.command
+
+
 class SelectedTasks(object):
     def __init__(self):
         self.tw = tw