Allow printing colored text

Author: Serious-senpai

src/commands/eval.hpp

@@ -26,7 +26,10 @@ class EvalCommand : public liteshell::BaseCommand
         auto input = context.get("expression");
         if (context.present.count("-p"))
         {
-            input = context.client->get_stream()->getline(input, liteshell::InputStream::FORCE_STDOUT | liteshell::InputStream::FORCE_STDIN);
+            input = context.client->get_stream()->getline(
+                [&input]()
+                { std::cout << input << std::flush; },
+                liteshell::InputStream::FORCE_STDOUT | liteshell::InputStream::FORCE_STDIN);
         }
 
         auto result = context.present.count("-m") ? std::to_string(context.client->get_environment()->eval_ll(input)) : input;

src/commands/for.hpp

@@ -34,7 +34,10 @@ class ForCommand : public liteshell::BaseCommand
         bool force_stream = !context.client->get_stream()->exhaust();
         while (true)
         {
-            auto input = utils::strip(context.client->get_stream()->getline("for>", force_stream ? liteshell::InputStream::FORCE_STREAM : 0));
+            auto input = utils::strip(context.client->get_stream()->getline(
+                []()
+                { std::cout << "for>" << std::flush; },
+                force_stream ? liteshell::InputStream::FORCE_STREAM : 0));
             if (utils::startswith(input, "for "))
             {
                 counter++;

src/commands/if.hpp

@@ -31,7 +31,10 @@ class IfCommand : public liteshell::BaseCommand
         bool has_else = false;
         while (true)
         {
-            auto input = utils::strip(context.client->get_stream()->getline("if_true>", force_stream ? liteshell::InputStream::FORCE_STREAM : 0));
+            auto input = utils::strip(context.client->get_stream()->getline(
+                []()
+                { std::cout << "if_true>" << std::flush; },
+                force_stream ? liteshell::InputStream::FORCE_STREAM : 0));
             if (utils::startswith(input, "if "))
             {
                 counter++;
@@ -63,7 +66,10 @@ class IfCommand : public liteshell::BaseCommand
         {
             while (true)
             {
-                auto input = utils::strip(context.client->get_stream()->getline("if_false>", force_stream ? liteshell::InputStream::FORCE_STREAM : 0));
+                auto input = utils::strip(context.client->get_stream()->getline(
+                    []()
+                    { std::cout << "if_false>" << std::flush; },
+                    force_stream ? liteshell::InputStream::FORCE_STREAM : 0));
                 if (utils::startswith(input, "if "))
                 {
                     counter++;

src/include/all.hpp

@@ -16,6 +16,7 @@
 #include "standard.hpp"
 #include "stream.hpp"
 #include "strip.hpp"
+#include "style.hpp"
 #include "subprocess.hpp"
 #include "tables.hpp"
 #include "units.hpp"

src/include/client.hpp

@@ -6,6 +6,7 @@
 #include "fuzzy_search.hpp"
 #include "maps.hpp"
 #include "stream.hpp"
+#include "style.hpp"
 #include "subprocess.hpp"
 #include "wrapper.hpp"
 
@@ -392,16 +393,16 @@ namespace liteshell
             utils::set_ignore_ctrl_c(true);
             while (true)
             {
-                SYSTEMTIME time;
-                GetLocalTime(&time);
                 process_command(
                     _stream->getline(
-                        utils::format(
-                            "\n[%d:%d:%d]liteshell~%s>",
-                            time.wHour,
-                            time.wMinute,
-                            time.wSecond,
-                            utils::get_working_directory().c_str()),
+                        []()
+                        {
+                            SYSTEMTIME time;
+                            GetLocalTime(&time);
+                            std::cout << utils::format("\n[%d:%d:%d]", time.wHour, time.wMinute, time.wSecond);
+                            utils::style_print("liteshell~", FOREGROUND_BLUE | FOREGROUND_INTENSITY);
+                            std::cout << utils::get_working_directory() << ">";
+                        },
                         0));
             }
         }

src/include/stream.hpp

@@ -82,11 +82,11 @@ namespace liteshell
         /**
          * @brief Read the next command
          *
-         * @param prompt The prompt to display before reading
+         * @param prompt The function to display the prompt string before reading
          * @param flags The flags to use when reading the command
          * @return The next command in the input stream
          */
-        std::string getline(const std::string &prompt, const int flags)
+        std::string getline(const std::function<void()> &prompt, const int flags)
         {
 #ifdef DEBUG
             std::cout << "Received getline request, flags = " << flags << std::endl;
@@ -111,7 +111,7 @@ namespace liteshell
 
             if ((flags & FORCE_STDOUT) || (echo && peek_echo()))
             {
-                std::cout << prompt << std::flush;
+                prompt();
             }
 
             bool from_stdin = (flags & FORCE_STDIN) || exhaust();

src/include/style.hpp

@@ -0,0 +1,23 @@
+#pragma once
+
+#include "standard.hpp"
+
+namespace utils
+{
+    template <typename T>
+    void style_print(const T &message, const WORD &attributes)
+    {
+        const auto console = GetStdHandle(STD_OUTPUT_HANDLE);
+
+        CONSOLE_SCREEN_BUFFER_INFO current;
+        const auto has_old_attr = GetConsoleScreenBufferInfo(console, &current);
+
+        SetConsoleTextAttribute(console, attributes); // Ignore failure
+        std::cout << message;
+
+        if (has_old_attr)
+        {
+            SetConsoleTextAttribute(console, current.wAttributes);
+        }
+    }
+}