#9056 Refactor YDB CLI by extracting highlight and complete modules

ydb/public/lib/ydb_cli/commands/interactive/complete/ya.make

@@ -0,0 +1,13 @@
+LIBRARY()
+
+SRCS(
+    yql_completer.cpp
+)
+
+PEERDIR(
+    contrib/restricted/patched/replxx
+    yql/essentials/sql/v1/complete
+)
+
+END()
+

ydb/public/lib/ydb_cli/commands/interactive/complete/yql_completer.cpp

@@ -0,0 +1,45 @@
+#include "yql_completer.h"
+
+#include <yql/essentials/sql/v1/complete/sql_complete.h>
+
+namespace NYdb::NConsoleClient {
+
+    class TYQLCompleter: public IYQLCompleter {
+    public:
+        using TPtr = THolder<IYQLCompleter>;
+
+        explicit TYQLCompleter(NSQLComplete::ISqlCompletionEngine::TPtr engine)
+            : Engine(std::move(engine))
+        {
+        }
+
+        TCompletions Apply(const std::string& prefix, int& /* contextLen */) override {
+            auto completion = Engine->Complete({
+                .Text = prefix,
+                .CursorPosition = prefix.length(),
+            });
+
+            replxx::Replxx::completions_t entries;
+            for (auto& candidate : completion.Candidates) {
+                candidate.Content += ' ';
+                entries.emplace_back(
+                    std::move(candidate.Content),
+                    ReplxxColorOf(candidate.Kind));
+            }
+            return entries;
+        }
+
+    private:
+        static replxx::Replxx::Color ReplxxColorOf(NSQLComplete::ECandidateKind /* kind */) {
+            return replxx::Replxx::Color::DEFAULT;
+        }
+
+        NSQLComplete::ISqlCompletionEngine::TPtr Engine;
+    };
+
+    IYQLCompleter::TPtr MakeYQLCompleter() {
+        return IYQLCompleter::TPtr(
+            new TYQLCompleter(NSQLComplete::MakeSqlCompletionEngine()));
+    }
+
+} // namespace NYdb::NConsoleClient

ydb/public/lib/ydb_cli/commands/interactive/complete/yql_completer.h

@@ -0,0 +1,21 @@
+#pragma once
+
+#include <contrib/restricted/patched/replxx/include/replxx.hxx>
+
+#include <util/generic/ptr.h>
+
+namespace NYdb::NConsoleClient {
+
+    using TCompletions = replxx::Replxx::completions_t;
+
+    class IYQLCompleter {
+    public:
+        using TPtr = THolder<IYQLCompleter>;
+
+        virtual TCompletions Apply(const std::string& prefix, int& contextLen) = 0;
+        virtual ~IYQLCompleter() = default;
+    };
+
+    IYQLCompleter::TPtr MakeYQLCompleter();
+
+} // namespace NYdb::NConsoleClient

ydb/public/lib/ydb_cli/commands/interactive/highlight/ut/ya.make

@@ -0,0 +1,7 @@
+UNITTEST_FOR(ydb/public/lib/ydb_cli/commands/interactive/highlight)
+
+SRCS(
+    yql_highlighter_ut.cpp
+)
+
+END()

ydb/public/lib/ydb_cli/commands/interactive/highlight/ya.make

@@ -0,0 +1,20 @@
+LIBRARY()
+
+SRCS(
+    yql_highlighter.cpp
+    yql_position.cpp
+)
+
+PEERDIR(
+    contrib/restricted/patched/replxx
+    yql/essentials/sql/v1/lexer
+    yql/essentials/sql/v1/lexer/antlr4
+    yql/essentials/sql/v1/lexer/antlr4_ansi
+    yql/essentials/sql/settings
+)
+
+END()
+
+RECURSE_FOR_TESTS(
+    ut
+)