From 5e2792019f0f400e7b19291da9087fe9722728ba Mon Sep 17 00:00:00 2001
From: Linus <linus@cosmos-ink.net>
Date: Mon, 3 Aug 2020 19:33:54 +0200
Subject: [PATCH] Add xochitl support

---
 src/app.rs         | 20 +++++++++++++++++++-
 src/view/common.rs |  5 ++++-
 src/view/mod.rs    |  1 +
 3 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/src/app.rs b/src/app.rs
index 19062702..1ad75d5f 100644
--- a/src/app.rs
+++ b/src/app.rs
@@ -80,6 +80,7 @@ pub struct Context {
     pub covered: bool,
     pub shared: bool,
     pub online: bool,
+    pub killed_xochitl: bool,
 }
 
 impl Context {
@@ -94,7 +95,7 @@ impl Context {
                   keyboard_layouts: BTreeMap::new(), input_history: FxHashMap::default(),
                   battery, frontlight, lightsensor, notification_index: 0,
                   kb_rect: Rectangle::default(), rng, plugged: false, covered: false,
-                  shared: false, online: false }
+                  shared: false, online: false, killed_xochitl: false }
     }
 
     pub fn batch_import(&mut self) {
@@ -335,6 +336,7 @@ fn set_wifi(enable: bool, context: &mut Context) {
 
 enum ExitStatus {
     Quit,
+    QuitToXochitl,
     Reboot,
     PowerOff,
 }
@@ -435,6 +437,14 @@ pub fn run() -> Result<(), Error> {
     schedule_task(TaskId::CheckBattery, Event::CheckBattery,
                   BATTERY_REFRESH_INTERVAL, &tx, &mut tasks);
 
+    if let Ok(status) = Command::new("pidof").arg("xochitl").status() {
+        if status.code().unwrap() == 0 {
+            Command::new("systemctl").arg("stop").arg("xochitl").status().ok();
+            context.killed_xochitl = true;
+            println!("Xochitl was found and killed. You may only exit by starting Xochitl again.")
+        }
+    }
+
     while let Ok(evt) = rx.recv() {
         match evt {
             Event::Device(de) => {
@@ -1047,6 +1057,10 @@ pub fn run() -> Result<(), Error> {
             Event::Select(EntryId::Quit) => {
                 break;
             },
+            Event::Select(EntryId::QuitToXochitl) => {
+                exit_status = ExitStatus::QuitToXochitl;
+                break;
+            },
             Event::Select(EntryId::RebootInNickel) => {
                 fs::remove_file("bootlock").map_err(|e| {
                     eprintln!("Couldn't remove the bootlock file: {}", e);
@@ -1101,6 +1115,10 @@ pub fn run() -> Result<(), Error> {
             Command::new("sync").status().ok();
             Command::new("poweroff").arg("-f").status().ok();
         },
+        ExitStatus::QuitToXochitl => {
+            Command::new("sync").status().ok();
+            Command::new("systemctl").arg("start").arg("xochitl").status().ok();
+        },
         _ => (),
     }
 
diff --git a/src/view/common.rs b/src/view/common.rs
index d581f0a1..bcd97865 100644
--- a/src/view/common.rs
+++ b/src/view/common.rs
@@ -136,7 +136,10 @@ pub fn toggle_main_menu(view: &mut dyn View, rect: Rectangle, enable: Option<boo
             entries.push(EntryKind::Command("Reboot".to_string(), EntryId::Reboot));
         } else {
             entries.push(EntryKind::Command("Reboot".to_string(), EntryId::Reboot));
-            entries.push(EntryKind::Command("Quit".to_string(), EntryId::Quit));
+            if ! context.killed_xochitl {
+                entries.push(EntryKind::Command("Quit".to_string(), EntryId::Quit));
+            }
+            entries.push(EntryKind::Command("Quit to Xochitl".to_string(), EntryId::QuitToXochitl));
         }
 
         if CURRENT_DEVICE.has_page_turn_buttons() {
diff --git a/src/view/mod.rs b/src/view/mod.rs
index aad91755..bce5832e 100644
--- a/src/view/mod.rs
+++ b/src/view/mod.rs
@@ -513,6 +513,7 @@ pub enum EntryId {
     Reboot,
     RebootInNickel,
     Quit,
+    QuitToXochitl,
 }
 
 impl EntryKind {