TheaninovOS/desktops/hyprland/ags/modules/sessionscreen.js

// This is for the cool memory indicator on the sidebar
// For the right pill of the bar, see system.js
const {Gdk, Gtk} = imports.gi
const GObject = imports.gi.GObject
const Lang = imports.lang
import {App, Service, Utils, Widget} from "../imports.js"
const {exec, execAsync} = Utils

const SessionButton = (name, icon, command, props = {}) => {
  const buttonDescription = Widget.Revealer({
    valign: "end",
    transitionDuration: 200,
    transition: "slide_down",
    revealChild: false,
    child: Widget.Label({
      className: "txt-smaller session-button-desc",
      label: name,
    }),
  })
  return Widget.Button({
    onClicked: command,
    className: "session-button",
    child: Widget.Overlay({
      className: "session-button-box",
      child: Widget.Label({
        vexpand: true,
        className: "icon-material",
        label: icon,
      }),
      overlays: [buttonDescription],
    }),
    onHover: button => {
      const display = Gdk.Display.get_default()
      const cursor = Gdk.Cursor.new_from_name(display, "pointer")
      button.get_window().set_cursor(cursor)
      buttonDescription.revealChild = true
    },
    onHoverLost: button => {
      const display = Gdk.Display.get_default()
      const cursor = Gdk.Cursor.new_from_name(display, "default")
      button.get_window().set_cursor(cursor)
      buttonDescription.revealChild = false
    },
    connections: [
      [
        "focus-in-event",
        self => {
          buttonDescription.revealChild = true
          self.toggleClassName("session-button-focused", true)
        },
      ],
      [
        "focus-out-event",
        self => {
          buttonDescription.revealChild = false
          self.toggleClassName("session-button-focused", false)
        },
      ],
    ],
    ...props,
  })
}

export default () => {
  // lock, logout, sleep
  const lockButton = SessionButton("Lock", "lock", () => {
    App.closeWindow("session")
    execAsync("gtklock")
  })
  const logoutButton = SessionButton("Logout", "logout", () => {
    App.closeWindow("session")
    execAsync(["bash", "-c", "loginctl terminate-user $USER"])
  })
  const sleepButton = SessionButton("Sleep", "sleep", () => {
    App.closeWindow("session")
    execAsync("systemctl suspend")
  })
  // hibernate, shutdown, reboot
  const hibernateButton = SessionButton("Hibernate", "downloading", () => {
    App.closeWindow("session")
    execAsync("systemctl hibernate")
  })
  const shutdownButton = SessionButton("Shutdown", "power_settings_new", () => {
    App.closeWindow("session")
    execAsync("systemctl poweroff")
  })
  const rebootButton = SessionButton("Reboot", "restart_alt", () => {
    App.closeWindow("session")
    execAsync("systemctl reboot")
  })
  const cancelButton = SessionButton("Cancel", "close", () => App.closeWindow("session"), {
    className: "session-button-cancel",
  })
  return Widget.Box({
    className: "session-bg",
    style: `
        min-width: ${SCREEN_WIDTH * 2}px; 
        min-height: ${SCREEN_HEIGHT * 2}px;
        `, // Hack to draw over reserved bar space
    vertical: true,
    children: [
      Widget.EventBox({
        onPrimaryClick: () => App.closeWindow("session"),
        onSecondaryClick: () => App.closeWindow("session"),
        onMiddleClick: () => App.closeWindow("session"),
      }),
      Widget.Box({
        halign: "center",
        vexpand: true,
        vertical: true,
        children: [
          Widget.Box({
            valign: "center",
            vertical: true,
            className: "spacing-v-15",
            children: [
              Widget.Box({
                vertical: true,
                style: "margin-bottom: 0.682rem;",
                children: [
                  Widget.Label({
                    className: "txt-title txt",
                    label: "Session",
                  }),
                  Widget.Label({
                    justify: Gtk.Justification.CENTER,
                    className: "txt-small txt",
                    label: "Use arrow keys to navigate.\nEnter to select, Esc to cancel.",
                  }),
                ],
              }),
              Widget.Box({
                halign: "center",
                className: "spacing-h-15",
                children: [
                  // lock, logout, sleep
                  lockButton,
                  logoutButton,
                  sleepButton,
                ],
              }),
              Widget.Box({
                halign: "center",
                className: "spacing-h-15",
                children: [
                  // hibernate, shutdown, reboot
                  hibernateButton,
                  shutdownButton,
                  rebootButton,
                ],
              }),
              Widget.Box({
                halign: "center",
                className: "spacing-h-15",
                children: [
                  // hibernate, shutdown, reboot
                  cancelButton,
                ],
              }),
            ],
          }),
        ],
      }),
    ],
    connections: [
      [
        App,
        (_b, name, visible) => {
          if (visible) lockButton.grab_focus() // Lock is the default option
        },
      ],
    ],
  })
}