From 020f1402b966077f18e44ddb2ff551d7b7a54de9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jan=20Tr=C3=A1vn=C3=AD=C4=8Dek?= <travnja3@fit.cvut.cz>
Date: Thu, 11 Mar 2021 20:10:57 +0100
Subject: [PATCH] aql2: allow completion on print command

---
 aql2/src/prompt/ReadlinePromptCompletion.cpp | 2 +-
 aql2/src/prompt/ReadlinePromptCompletion.h   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/aql2/src/prompt/ReadlinePromptCompletion.cpp b/aql2/src/prompt/ReadlinePromptCompletion.cpp
index 3d213c6dc2..ac54dcfa4d 100644
--- a/aql2/src/prompt/ReadlinePromptCompletion.cpp
+++ b/aql2/src/prompt/ReadlinePromptCompletion.cpp
@@ -95,7 +95,7 @@ ReadlinePromptCompletion::CompletionContext ReadlinePromptCompletion::context (
 	if ( p >= rl_line_buffer && ( *p == '|' || *p == '(' ) )
 		return CompletionContext::ALGORITHM;
 
-	if ( masterCommandCompletionTest ( line, start, "execute" ) )
+	if ( masterCommandCompletionTest ( line, start, "execute" ) || masterCommandCompletionTest ( line, start, "print" ) )
 		return CompletionContext::ALGORITHM;
 
 	/* undecided, fallback to filepath */
diff --git a/aql2/src/prompt/ReadlinePromptCompletion.h b/aql2/src/prompt/ReadlinePromptCompletion.h
index afb26f7ee3..778ea683a7 100644
--- a/aql2/src/prompt/ReadlinePromptCompletion.h
+++ b/aql2/src/prompt/ReadlinePromptCompletion.h
@@ -64,7 +64,7 @@ class ReadlinePromptCompletion {
 	}
 
 	static std::set < std::string > fetchCommands ( const char *text ) {
-		return filter_completions ( { "execute", "introspect", "quit", "help", "set" }, text );
+		return filter_completions ( { "print", "execute", "introspect", "quit", "help", "set" }, text );
 	}
 
 	static std::set < std::string > fetchCommandsIntrospect ( const char *text ) {
-- 
GitLab