From 290dd037774e2b3837a2aada5bc4d877ee25b68d Mon Sep 17 00:00:00 2001
From: jhenrique09 <jhenrique09.mcz@hotmail.com>
Date: Wed, 25 Oct 2017 07:55:12 -0200
Subject: [PATCH] sanders: KeyHandler: Fix haptic on power action

---
 .../src/org/lineageos/settings/device/KeyHandler.java  | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/LineageActions/src/org/lineageos/settings/device/KeyHandler.java b/LineageActions/src/org/lineageos/settings/device/KeyHandler.java
index 3fbd786..bb98733 100644
--- a/LineageActions/src/org/lineageos/settings/device/KeyHandler.java
+++ b/LineageActions/src/org/lineageos/settings/device/KeyHandler.java
@@ -511,6 +511,9 @@ public class KeyHandler implements DeviceKeyHandler {
         if (isHapticFeedbackEnabledOnFP && (action == ACTION_CAMERA || action == ACTION_FLASHLIGHT)) {
             vibrate(action == ACTION_CAMERA ? 500 : 250);
         }
+        if (isHapticFeedbackEnabledOnFP && action == ACTION_POWER) {
+            doHapticFeedbackFP(false);
+        }
         switch (action) {
             case ACTION_HOME:
                 if (!mKeyguardManager.inKeyguardRestrictedInputMode()) {
@@ -563,7 +566,7 @@ public class KeyHandler implements DeviceKeyHandler {
                 }
                 break;
         }
-        if (isHapticFeedbackEnabledOnFP && action != ACTION_VOICE_ASSISTANT && action != ACTION_CAMERA && action != ACTION_FLASHLIGHT) { // prevent double vibration
+        if (isHapticFeedbackEnabledOnFP && action != ACTION_VOICE_ASSISTANT && action != ACTION_CAMERA && action != ACTION_FLASHLIGHT && action != ACTION_POWER) { // prevent double vibration
             doHapticFeedbackFP(false);
         }
     }
@@ -766,6 +769,9 @@ public class KeyHandler implements DeviceKeyHandler {
         if (haptic && (action == ACTION_CAMERA || action == ACTION_FLASHLIGHT)) {
             vibrate(action == ACTION_CAMERA ? 500 : 250);
         }
+        if (haptic && action == ACTION_POWER){
+            doHapticFeedbackScreenOff();
+        }
         switch (action) {
             case ACTION_POWER:
                 toggleScreenState();
@@ -798,7 +804,7 @@ public class KeyHandler implements DeviceKeyHandler {
                 openMessages();
                 break;
         }
-        if (action != ACTION_FLASHLIGHT && action != ACTION_CAMERA) {
+        if (action != ACTION_FLASHLIGHT && action != ACTION_CAMERA && action != ACTION_POWER) {
             doHapticFeedbackScreenOff();
         }
     }