From 55d13e2c5a65abc99d1be5130d7df87368f7923c Mon Sep 17 00:00:00 2001
From: Quentin Decaunes <le.storm1er@gmail.com>
Date: Sat, 17 Aug 2019 17:50:02 +0200
Subject: [PATCH] feat: #39 You can now define preset to be applied on AC
 status change.

---
 js/app.js     |   1 +
 js/methods.js |  43 ++++++++++++----
 js/preset.js  | 132 ++++++++++++++++++++++++++++++++++++++++++++------
 3 files changed, 152 insertions(+), 24 deletions(-)

diff --git a/js/app.js b/js/app.js
index 5b02c64..6f05bd7 100644
--- a/js/app.js
+++ b/js/app.js
@@ -15,6 +15,7 @@ ready(function(){
   checkForAdminRights();
   displayVersion();
   reApplyPeriodically(require('electron-settings').get('settings.reapply_periodically'));
+  handleAcStatusChanges();
   if (isWindows() && isDevMode()) {
     recreateShortcut();
   }
diff --git a/js/methods.js b/js/methods.js
index 75c6255..0920065 100644
--- a/js/methods.js
+++ b/js/methods.js
@@ -202,16 +202,16 @@ function checkForAdminRights() {
       );
     }
   } else {
-  const child = require('child_process').execFile;
-  child('NET SESSION', function(err,so,se) {
-    if (se.length !== 0) {
-      notification('warning',
-        `Warning: you should launch this app as administrator, `
-      + `ryzenadj.exe doesn't seems to work correctly without administrator rights.`
-      );
-    }
-  });
-}
+    const child = require('child_process').execFile;
+    child('NET SESSION', function(err,so,se) {
+      if (se.length !== 0) {
+        notification('warning',
+          `Warning: you should launch this app as administrator, `
+        + `ryzenadj.exe doesn't seems to work correctly without administrator rights.`
+        );
+      }
+    });
+  }
 }
 
 /**
@@ -670,3 +670,26 @@ function updateScheduledStartOnBoot(toBeEnabled) {
 function isDevMode() {
   return !require('electron').remote.app.isPackaged;
 }
+
+/**
+ * Will listen for system events and handle status for it.
+ */
+function handleAcStatusChanges() {
+  const powerMonitor = require('electron').remote.powerMonitor;
+  const settings = require('electron-settings');
+
+  let applyPresetOnAcStatusChange = function(presetName) {
+    appendLog(`applyPresetOnAcStatusChange(${presetName})`);
+    if (!presetName) {
+      return;
+    }
+    preset_apply(presetName);
+  };
+
+  powerMonitor.on('on-ac', () => {
+    applyPresetOnAcStatusChange(settings.get(`auto-apply.update-ac-plugged-in`));
+  });
+  powerMonitor.on('on-battery', () => {
+    applyPresetOnAcStatusChange(settings.get(`auto-apply.update-ac-plugged-out`));
+  });
+}
diff --git a/js/preset.js b/js/preset.js
index 1b62f21..cd26b61 100644
--- a/js/preset.js
+++ b/js/preset.js
@@ -85,6 +85,33 @@ function preset_findUnusedPresetName(newPresetName, suffix = 1) {
   return `${newPresetName}${suffix}`;
 }
 
+/**
+ * Will save the preset to be enabled on AC plugged out.
+ */
+function preset_enableAutoApplyOnAcStatusChange(statusName, presetName) {
+  const settings = require('electron-settings');
+  const status = {
+    "update-ac-plugged-in": `will be applied on AC plugged in.`,
+    "update-ac-plugged-out": `will be applied on AC plugged out.`,
+  };
+
+  if (typeof status[statusName] === "undefined") {
+    let message = `Error while updating auto apply on AC status change.`;
+    notification('danger', message);
+    console.log(`preset_enableAutoApplyOnAcStatusChange(statusName:"${statusName}", presetName:"${presetName}")`);
+    Sentry.captureException(new Error(message));
+    return;
+  }
+
+  settings.set(`auto-apply.${statusName}`, presetName);
+
+  if (presetName) {
+    notification('primary', `Preset "${presetName}" ${status[statusName]}`);
+  } else {
+    notification('primary', `No preset ${status[statusName]}`);
+  }
+}
+
 /**
  * This will update the preset tab based on saved presets.
  */
@@ -93,10 +120,20 @@ function preset_updateList() {
   const currentPresets = require('electron-settings').get('presets') || {};
 
   var content = '';
-  content += '<ul class="uk-list">';
+  content += /*html*/`
+    <table class="uk-table uk-table-striped uk-table-hover uk-table-middle">
+      <thead>
+        <tr>
+          <th>Name</th>
+          <th colspan="2">Apply on</th>
+          <th>Action</th>
+        </tr>
+      </thead>
+      <tbody>
+  `;
 
   if (Object.keys(currentPresets).length === 0) {
-    content += `<li>No preset has been created yet, import them or use the "Save to preset" button on Controller tab to create one.</li>`;
+    content += /*html*/`<li>No preset has been created yet, import them or use the "Save to preset" button on Controller tab to create one.</li>`;
   }
 
   for (const presetName in currentPresets) {
@@ -112,22 +149,89 @@ function preset_updateList() {
       }
       valueSummary.join(', ');
 
-      content += `
-        <li class="uk-margin">
-          <span class="uk-text-lead">${presetName}</span>
-          <button class="uk-button uk-button-danger uk-align-right" type="button" onClick="preset_deletion('${presetName}')">
-            Delete
-          </button>
-          <button class="uk-button uk-button-primary uk-align-right" type="button" onClick="preset_apply('${presetName}')">
-            Apply
-          </button>
-          <i class="uk-text-small">${valueSummary}</i>
-        </li>
+      content += /*html*/`
+        <tr class="uk-margin">
+          <td>
+            <span class="uk-text-lead">${presetName}</span>
+            <i class="uk-text-small">${valueSummary}</i>
+          </td>
+          <td class="uk-table-expand">
+            <label style="cursor: pointer;">
+              <input
+                value="${presetName}"
+                class="uk-radio onAcStatusChange"
+                type="radio"
+                name="update-ac-plugged-in"
+                ${presetName === require('electron-settings').get('auto-apply.update-ac-plugged-in') ? 'checked' : ''}
+              />
+              AC&nbsp;plugged&nbsp;in
+            </label>
+          </td>
+          <td class="uk-table-expand">
+            <label style="cursor: pointer;">
+              <input
+                value="${presetName}"
+                class="uk-radio onAcStatusChange"
+                type="radio"
+                name="update-ac-plugged-out"
+                ${presetName === require('electron-settings').get('auto-apply.update-ac-plugged-out') ? 'checked' : ''}
+              />
+              AC&nbsp;plugged&nbsp;out
+            </label>
+          </td>
+          <td>
+            <button class="uk-button uk-margin uk-button-danger" type="button" onClick="preset_deletion('${presetName}')">
+              Delete
+            </button>
+            <button class="uk-button uk-button-primary" type="button" onClick="preset_apply('${presetName}')">
+              Apply
+            </button>
+          </td>
+        </tr>
       `;
     }
   }
-  content += '</ul>';
+  content += /*html*/`
+        <tr>
+          <td><span class="uk-align-right">Disable auto apply</span></td>
+          <td>
+            <label style="cursor: pointer;">
+              <input
+                value=""
+                class="uk-radio onAcStatusChange"
+                type="radio"
+                name="update-ac-plugged-in"
+                ${!require('electron-settings').get('auto-apply.update-ac-plugged-in') ? 'checked' : ''}
+              />
+              On&nbsp;charging
+            </label>
+          </td>
+          <td>
+            <label style="cursor: pointer;">
+              <input
+                value=""
+                class="uk-radio onAcStatusChange"
+                type="radio"
+                name="update-ac-plugged-out"
+                ${!require('electron-settings').get('auto-apply.update-ac-plugged-out') ? 'checked' : ''}
+              />
+              On&nbsp;discharging
+            </label>
+          </td>
+          <td></td>
+        </tr>
+      <tbody>
+    </table>
+  `;
   presetTab.innerHTML = content;
+  var acStatusChangeRadios = document.querySelectorAll('.onAcStatusChange');
+
+  Array.from(acStatusChangeRadios).forEach(radio => {
+    radio.addEventListener('click', function(event) {
+      preset_enableAutoApplyOnAcStatusChange(this.name, this.value);
+    });
+  });
+
 }
 
 /**