From dc82a4a2af4c8c2e34cdbd56e07f097dff5c9a78 Mon Sep 17 00:00:00 2001
From: Eric Wong <e@80x24.org>
Date: Thu, 29 Apr 2021 09:46:20 +0000
Subject: lei import: support shell completion of known folders

This also fixes completion of "lei up" for IMAP folders.
---
 lib/PublicInbox/LeiImport.pm | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'lib/PublicInbox/LeiImport.pm')

diff --git a/lib/PublicInbox/LeiImport.pm b/lib/PublicInbox/LeiImport.pm
index 277f4f95..def121ab 100644
--- a/lib/PublicInbox/LeiImport.pm
+++ b/lib/PublicInbox/LeiImport.pm
@@ -89,6 +89,14 @@ sub lei_import { # the main "lei import" method
 	$op_c->op_wait_event($ops);
 }
 
+sub _complete_import {
+	my ($lei, @argv) = @_;
+	my $sto = $lei->_lei_store or return;
+	my $lms = $sto->search->lms or return;
+	my $match_cb = $lei->complete_url_prepare(\@argv);
+	map { $match_cb->($_) } $lms->folders;
+}
+
 no warnings 'once';
 *ipc_atfork_child = \&PublicInbox::LeiInput::input_only_atfork_child;
 
-- 
cgit v1.2.3-24-ge0c7