最近ずっと Denite をいじっております。こいつは Unite の後継だけあって大した時間泥棒プラグインですね！ いくらでもカスタム可能なので際限なく設定に凝ることが出来ます。Unite と違って Python で拡張が書けるのでなおさらです。
今回はいつも利用させていただいてます、mattn/memoの source を書いてみました。これはリンク先を見るとわかるとおり、メモを取るためのツールなのですが、本来はターミナルから起動してメモを作成／検索するためのものです。ターミナルからメモを作成／検索するのは簡単なんですが、一度作成したメモを Vim 上から検索したりするのが難しかったんですよね。これを簡単にしてみました。
こんな感じです。インストール方法などは README を参照してください。以下の機能が利用できます。
" List up all memos:Denite memo" List up all memos or create a new one:Denite memo memo:new" Search memo:Denite memo/grep" Search memo interactively:Denite memo/grep::!特に最後の、「memo のインクリメンタル検索」が欲しくて作ったようなものです。同じ事は peco の --exec オプションを使えばできるのですが、Vim 上でやることに意味があるのです！多分！
以下、vim-go に unite/denite source を作って contribute した話 - Qiitaの続きのような内容ですが、開発中に気をつけた点についてまとめておきます。
memo 一覧を非同期に読み込む
grep sourceが非常に参考になりました。通常は一度だけ呼ばれて候補を返す gather_candidatesメソッドですが、context['is_async'] = Trueにしておくと Denite が繰り返し呼んでくれるようになります。
内部では Popen.communicate メソッドを使って 0.01 秒ごと（async_timeoutオプションで変更可能）に結果を返すようにしています。gather_candidatesから処理が戻る度に逐次表示も更新されます。
気をつけなければいけないのは、候補を出し尽くした後は context['is_async'] = Falseにしないと永遠に待ち続けてしまうことです。
# 省略しています。詳しくはソースを見てください。defgather_candidates(self,context):ifcontext['__proc']:returnself.__async_gather_candidates(context,0.01)# process.Process は subprocess.Popen のラッパーです。context['__proc']=process.Process(['memo','list'],context,context['path'])returnself.__async_gather_candidates(context,0.01)def__async_gather_candidates(self,context,timeout):# エラー処理は省略outs,errs=context['__proc'].communicate(timeout=timeout)# eof に到達したら完了context['is_async']=notcontext['__proc'].eof()# ここで出力結果をごにょごにょして候補を作るreturncandidatesタイトルを全角○文字に切り詰める
memo コマンドはターミナルで起動したとき「ファイル名 : タイトル」のように一覧を表示してくれるのですが、長すぎるファイル名は適当にちょん切ってくれます。これを Denite 上で再現するのは結構骨でした。どんな言語でも環境でもそうですが、文字列の「全角○文字詰め」って厄介なんですよねー。
Vim の場合さらに 'ambiwidth'というオプションがありまして、設定によっては同じ文字が半角だったり全角だったりします。strwidth()という関数を使うと「'ambiwidth'オプションを考慮した文字幅」が得られます。
ホントは「'ambiwidth'オプションを考慮した文字数にカットしてくれる関数」が欲しかったのですが、そういうものは Vim には無いようです。しょうがないので地道にループ廻して一文字ずつカウントしています。
# なんか Python っぽくないコードになった……。i=0whileTrue:char=string[i:i+1]next_r=result+char# Denite の便利メソッドを使うと Vim の関数が簡単に呼べますnlen=self.vim.funcs.strwidth(next_r)ifnlen>col-3:rlen=self.vim.funcs.strwidth(result)returnresult+('....'ifrlen<col-3else'...')elifnlen==col-3:returnnext_r+'...'result=next_ri+=1他の source を継承して source を作る
memo/grep source は grep source を継承して、ロジックをそのまま使っています。継承のためにモジュールを import するには相対パスを使うと簡単です。
from.grepimportSourceasGrep# grep source を継承して必要な情報だけ上書きするclassSource(Grep):def__init__(self,vim):super().__init__(vim)self.name='memo/grep'最初同様のロジックを見たとき、「同じディレクトリに grep.pyなんて存在しないのに、なんでこれでいいの？」って思ったのですが、これは Denite のコードに秘密がありました。denite.load_sourcesというメソッドで 'runtimepath'内にある全ての source を動的に読み込んでいるんですね。grep.pyは標準添付の source ですので import 出来た訳です。
さらに memo_grep.pyでは同じディレクトリにある memo.pyをも import しています。これは memo.pyが memo_grep.pyより先に読み込まれていることを期待しているので少々 hacky です。
# memo.py は先に読み込まれているはずfrom.memoimportMemo,CommandNotFoundError他の kind を継承して kind を作る
「候補を列挙する」ものが source なら「選択した候補で何かする」のが kind です。今回の memo source による各の候補は単なるファイルと異なりませんから file kind を継承して作れば良さそうです。
from.fileimportKindasFileclassKind(File):def__init__(self,vim):super().__init__(vim)self.name='memo'source の時と一緒ですね。今回は memo を開くための open action だけを上書きしています。ホントは memo を削除する action も追加しようと思ったのですが、それは file kind の方に追加した方が良さそうです。
以上です。特に is_asyncオプションで簡単に候補の非同期読み込みができるのがいい感じです。Vim だって job/channel による非同期実行は可能になりましたが、やっぱり Python 使えると楽ですねえ。