1 files changed, 144 insertions, 0 deletions
diff --git a/Servers/WindowServer/WSMenuManager.cpp b/Servers/WindowServer/WSMenuManager.cpp
new file mode 100644
index 0000000000..50a88f274c
--- /dev/null
+++ b/Servers/WindowServer/WSMenuManager.cpp
@@ -0,0 +1,144 @@
+#include <LibCore/CTimer.h>
+#include <SharedGraphics/Font.h>
+#include <SharedGraphics/Painter.h>
+#include <WindowServer/WSMenuBarKeeper.h>
+#include <WindowServer/WSWindowManager.h>
+#include <time.h>
+#include <unistd.h>
+
+WSMenuManager::WSMenuManager()
+{
+    m_username = getlogin();
+
+    new CTimer(300, [this] {
+        static time_t last_update_time;
+        time_t now = time(nullptr);
+        if (now != last_update_time || m_cpu_monitor.is_dirty()) {
+            tick_clock();
+            last_update_time = now;
+            m_cpu_monitor.set_dirty(false);
+        }
+    });
+}
+
+WSMenuManager::~WSMenuManager()
+{
+}
+
+void WSMenuManager::setup()
+{
+    m_window = make<WSWindow>(*this, WSWindowType::Menubar);
+    m_window->set_rect(WSWindowManager::the().menubar_rect());
+}
+
+void WSMenuManager::draw()
+{
+    auto& wm = WSWindowManager::the();
+    auto menubar_rect = wm.menubar_rect();
+
+    Painter painter(*window().backing_store());
+
+    painter.fill_rect(menubar_rect, Color::WarmGray);
+    painter.draw_line({ 0, menubar_rect.bottom() }, { menubar_rect.right(), menubar_rect.bottom() }, Color::MidGray);
+    int index = 0;
+    wm.for_each_active_menubar_menu([&](WSMenu& menu) {
+        Color text_color = Color::Black;
+        if (&menu == wm.current_menu()) {
+            painter.fill_rect(menu.rect_in_menubar(), wm.menu_selection_color());
+            text_color = Color::White;
+        }
+        painter.draw_text(
+            menu.text_rect_in_menubar(),
+            menu.name(),
+            index == 1 ? wm.app_menu_font() : wm.menu_font(),
+            TextAlignment::CenterLeft,
+            text_color);
+        ++index;
+        return true;
+    });
+
+    int username_width = Font::default_bold_font().width(m_username);
+    Rect username_rect {
+        menubar_rect.right() - wm.menubar_menu_margin() / 2 - Font::default_bold_font().width(m_username),
+        menubar_rect.y(),
+        username_width,
+        menubar_rect.height()
+    };
+    painter.draw_text(username_rect, m_username, Font::default_bold_font(), TextAlignment::CenterRight, Color::Black);
+
+    time_t now = time(nullptr);
+    auto* tm = localtime(&now);
+    auto time_text = String::format("%4u-%02u-%02u %02u:%02u:%02u",
+        tm->tm_year + 1900,
+        tm->tm_mon + 1,
+        tm->tm_mday,
+        tm->tm_hour,
+        tm->tm_min,
+        tm->tm_sec);
+    int time_width = wm.font().width(time_text);
+    Rect time_rect {
+        username_rect.left() - wm.menubar_menu_margin() / 2 - time_width,
+        menubar_rect.y(),
+        time_width,
+        menubar_rect.height()
+    };
+
+    painter.draw_text(time_rect, time_text, wm.font(), TextAlignment::CenterRight, Color::Black);
+
+    Rect cpu_rect { time_rect.right() - wm.font().width(time_text) - m_cpu_monitor.capacity() - 10, time_rect.y() + 1, m_cpu_monitor.capacity(), time_rect.height() - 2 };
+    m_cpu_monitor.paint(painter, cpu_rect);
+}
+
+void WSMenuManager::tick_clock()
+{
+    refresh();
+}
+
+void WSMenuManager::refresh()
+{
+    if (!m_window)
+        return;
+    draw();
+    window().invalidate();
+}
+
+void WSMenuManager::event(CEvent& event)
+{
+    if (event.type() == WSEvent::MouseMove || event.type() == WSEvent::MouseUp || event.type() == WSEvent::MouseDown || event.type() == WSEvent::MouseWheel) {
+        auto& mouse_event = static_cast<WSMouseEvent&>(event);
+        WSWindowManager::the().for_each_active_menubar_menu([&](WSMenu& menu) {
+            if (menu.rect_in_menubar().contains(mouse_event.position())) {
+                handle_menu_mouse_event(menu, mouse_event);
+                return false;
+            }
+            return true;
+        });
+    }
+    return CObject::event(event);
+}
+
+void WSMenuManager::handle_menu_mouse_event(WSMenu& menu, const WSMouseEvent& event)
+{
+    auto& wm = WSWindowManager::the();
+    bool is_hover_with_any_menu_open = event.type() == WSMouseEvent::MouseMove && wm.current_menu() && (wm.current_menu()->menubar() || wm.current_menu() == wm.system_menu());
+    bool is_mousedown_with_left_button = event.type() == WSMouseEvent::MouseDown && event.button() == MouseButton::Left;
+    bool should_open_menu = &menu != wm.current_menu() && (is_hover_with_any_menu_open || is_mousedown_with_left_button);
+
+    if (should_open_menu) {
+        if (wm.current_menu() == &menu)
+            return;
+        wm.close_current_menu();
+        if (!menu.is_empty()) {
+            auto& menu_window = menu.ensure_menu_window();
+            menu_window.move_to({ menu.rect_in_menubar().x(), menu.rect_in_menubar().bottom() + 2 });
+            menu_window.set_visible(true);
+        }
+        wm.set_current_menu(&menu);
+        refresh();
+        return;
+    }
+    if (event.type() == WSMouseEvent::MouseDown && event.button() == MouseButton::Left) {
+        wm.close_current_menu();
+        return;
+    }
+}