Can not make it work on a Nativescript-Vue project #6

Author: EddyVerbruggen

insomnia.android.js

@@ -1,11 +1,26 @@
-var appModule = require("application");
+var app = require("tns-core-modules/application");
 
 exports.keepAwake = function () {
+
+  var keepScreenOn = function() {
+    var activity = app.android.foregroundActivity || app.android.startActivity;
+    var window = activity.getWindow();
+    window.addFlags(android.view.WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON);
+  };
+
   return new Promise(function (resolve, reject) {
     try {
-      var window = appModule.android.currentContext.getWindow();
-      window.addFlags(android.view.WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON);
-      resolve();
+
+      if (app.android.foregroundActivity || app.android.startActivity) {
+        keepScreenOn();
+        resolve();
+      } else {
+        app.android.on("activityStarted", function (args) {
+          keepScreenOn();
+          resolve();
+        });
+      }
+
     } catch (ex) {
       console.log("Error in insomnia.keepAwake: " + ex);
       reject(ex);
@@ -16,12 +31,13 @@ exports.keepAwake = function () {
 exports.allowSleepAgain = function () {
   return new Promise(function (resolve, reject) {
     try {
-      var window = appModule.android.currentContext.getWindow();
+      var activity = app.android.foregroundActivity || app.android.startActivity;
+      var window = activity.getWindow();
       window.clearFlags(android.view.WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON);
       resolve();
     } catch (ex) {
       console.log("Error in insomnia.allowSleepAgain: " + ex);
       reject(ex);
     }
   });
-};
+};

insomnia.ios.js

@@ -1,4 +1,4 @@
-var utils = require("utils/utils");
+var utils = require("tns-core-modules/utils/utils");
 
 exports.keepAwake = function () {
   return new Promise(function (resolve, reject) {
@@ -28,4 +28,4 @@ exports.allowSleepAgain = function () {
       reject(ex);
     }
   });
-};
+};

package.json

@@ -1,6 +1,6 @@
 {
   "name": "nativescript-insomnia",
-  "version": "1.2.1",
+  "version": "1.2.2",
   "description": "Make the screen not dim (and eventually lock the device) while Insomnia is active",
   "main": "insomnia",
   "typings": "index.d.ts",