1 files changed, 92 insertions, 0 deletions
diff --git a/services/java/com/android/server/status/widget/LockScreenButton.java b/services/java/com/android/server/status/widget/LockScreenButton.java
new file mode 100755
index 0000000..e1b205d
--- /dev/null
+++ b/services/java/com/android/server/status/widget/LockScreenButton.java
@@ -0,0 +1,92 @@
+package com.android.server.status.widget;
+
+import com.android.internal.R;
+import com.android.server.status.widget.PowerButton;
+
+import android.app.Activity;
+import android.app.KeyguardManager;
+import android.app.KeyguardManager.KeyguardLock;
+import android.content.Context;
+import android.content.SharedPreferences;
+import android.view.Gravity;
+import android.widget.Toast;
+
+public class LockScreenButton extends PowerButton {
+
+    static Boolean lockScreen = null;
+
+    public static final String LOCK_SCREEN = "lockScreen";
+    static LockScreenButton ownButton = null;
+    KeyguardLock lock;
+
+    private KeyguardLock getLock(Context context) {
+        if (lock == null) {
+            KeyguardManager keyguardManager = (KeyguardManager)context.
+                    getSystemService(Activity.KEYGUARD_SERVICE);
+            lock = keyguardManager.newKeyguardLock(Context.KEYGUARD_SERVICE);
+        }
+        return lock;
+    }
+
+    public void updateState(Context context) {
+        getState(context);
+        if (lockScreen == null) {
+            currentIcon = R.drawable.stat_lock_screen_off;
+            currentState = PowerButton.STATE_INTERMEDIATE;
+        } else if (lockScreen) {
+            currentIcon = R.drawable.stat_lock_screen_on;
+            currentState = PowerButton.STATE_ENABLED;
+        } else {
+            currentIcon = R.drawable.stat_lock_screen_off;
+            currentState = PowerButton.STATE_DISABLED;
+        }
+    }
+
+    /**
+     * Toggles the state of GPS.
+     *
+     * @param context
+     */
+    public void toggleState(Context context) {
+        getState(context);
+        if(lockScreen == null) {
+            Toast msg = Toast.makeText(context, "Not yet initialized", Toast.LENGTH_LONG);
+            msg.setGravity(Gravity.CENTER, msg.getXOffset() / 2, msg.getYOffset() / 2);
+            msg.show();
+        } else {
+            getLock(context);
+            if (lockScreen && lock != null) {
+                lock.disableKeyguard();
+                lockScreen = false;
+            } else if (lock != null) {
+                lock.reenableKeyguard();
+                lockScreen = true;
+            }
+        }
+    }
+
+    /**
+     * Gets the state of GPS location.
+     *
+     * @param context
+     * @return true if enabled.
+     */
+    private static boolean getState(Context context) {
+        if (lockScreen == null) {
+            lockScreen = true;
+        }
+        return lockScreen;
+    }
+
+
+    public static LockScreenButton getInstance() {
+        if (ownButton==null) ownButton = new LockScreenButton();
+
+        return ownButton;
+    }
+
+    @Override
+    void initButton(int position) {
+    }
+}
+