From 9739d2da0bb280b5fd6b95c77a7baa3853850c9f Mon Sep 17 00:00:00 2001
From: Leonardo Gama <leonardo.reis.gama@gmail.com>
Date: Wed, 30 Dec 2020 00:26:36 -0300
Subject: [PATCH] Make :Iawriter command accept an optional 'dimension'
 parameter and pass it to :Goyo command

---
 plugin/vim_iawriter.vim | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/plugin/vim_iawriter.vim b/plugin/vim_iawriter.vim
index 47aacb4..9810048 100644
--- a/plugin/vim_iawriter.vim
+++ b/plugin/vim_iawriter.vim
@@ -191,21 +191,25 @@ fun! s:leave()	" {{{1
 	silent! doautocmd <nomodeline> User IawriterPostLeave
 endfun
 
-fun! s:toggle()	" {{{1
+fun! s:toggle(dimension)	" {{{1
 	silent! doautocmd <nomodeline> User IawriterToggleTriggered
 	if exists('#goyo') && s:vim_iawriter_enabled
 		Goyo!
 	else
 		call <SID>pre_enter()
-		Goyo
+		if a:dimension is v:none
+			Goyo
+		else
+			execute 'Goyo' a:dimension
+		endif
 	endif
 	silent! doautocmd <nomodeline> User IawriterToggleFinished
 endfun	" }}}
 
-fun! vim_iawriter#toggle()	" {{{1
-	call <SID>toggle()
+fun! vim_iawriter#toggle(dimension=v:none)	" {{{1
+	call <SID>toggle(a:dimension)
 endfun	" }}}
-command! Iawriter call vim_iawriter#toggle()
+command! -nargs=? Iawriter call vim_iawriter#toggle(<f-args>)
 
 " Close vim when only window open is iAwriter	" {{{1
 " Taken from https://github.com/junegunn/goyo.vim/wiki/Customization#ensure-q-to-quit-even-when-goyo-is-active