From 5831a68e114b227adecaf7156b22135478aace9a Mon Sep 17 00:00:00 2001
From: Aviral Dasgupta <me@aviraldg.com>
Date: Mon, 13 Jun 2016 23:25:31 +0530
Subject: [PATCH] Labs improvements

---
 src/UserSettingsStore.js                  |  2 --
 src/components/structures/UserSettings.js | 26 +++++++++++++----------
 2 files changed, 15 insertions(+), 13 deletions(-)

diff --git a/src/UserSettingsStore.js b/src/UserSettingsStore.js
index f9c8a3ed2d..305994aa0e 100644
--- a/src/UserSettingsStore.js
+++ b/src/UserSettingsStore.js
@@ -114,12 +114,10 @@ module.exports = {
     },
 
     isFeatureEnabled: function(feature: string): boolean {
-        feature = feature.match(/\w+/g).join('_').toLowerCase();
         return localStorage.getItem(`mx_labs_feature_${feature}`) === 'true';
     },
 
     setFeatureEnabled: function(feature: string, enabled: boolean) {
-        feature = feature.match(/\w+/g).join('_').toLowerCase();
         localStorage.setItem(`mx_labs_feature_${feature}`, enabled);
     }
 };
diff --git a/src/components/structures/UserSettings.js b/src/components/structures/UserSettings.js
index 2023082422..a320a02da8 100644
--- a/src/components/structures/UserSettings.js
+++ b/src/components/structures/UserSettings.js
@@ -27,8 +27,14 @@ var Email = require('../../email');
 var AddThreepid = require('../../AddThreepid');
 
 const LABS_FEATURES = [
-    'Rich Text Editor',
-    'End-to-End Encryption'
+    {
+        name: 'Rich Text Editor',
+        id: 'rich_text_editor'
+    },
+    {
+        name: 'End-to-End Encryption',
+        id: 'e2e_encryption'
+    }
 ];
 
 module.exports = React.createClass({
@@ -364,14 +370,14 @@ module.exports = React.createClass({
 
         this._renderLabs = function () {
             let features = LABS_FEATURES.map(feature => (
-                <div key={feature}>
+                <div key={feature.id}>
                     <input
                            type="checkbox"
-                           id={feature}
-                           name={feature}
-                           defaultChecked={UserSettingsStore.isFeatureEnabled(feature)}
-                           onChange={e => UserSettingsStore.setFeatureEnabled(feature, e.target.checked)} />
-                    <label htmlFor={feature}>{feature}</label>
+                           id={feature.id}
+                           name={feature.id}
+                           defaultChecked={UserSettingsStore.isFeatureEnabled(feature.id)}
+                           onChange={e => UserSettingsStore.setFeatureEnabled(feature.id, e.target.checked)} />
+                    <label htmlFor={feature.id}>{feature.name}</label>
                 </div>
             ));
             return (
@@ -379,10 +385,8 @@ module.exports = React.createClass({
                     <h3>Labs</h3>
 
                     <div className="mx_UserSettings_section">
-                        <p>These experimental features may change, break or disappear at any time. We make absolutely no guarantees about what may happen if you turn one of these experiments on, and your client may even spontaneously combust. Jokes aside, your client may delete all your data or your security and privacy could be compromised in unexpected ways. Please proceed with caution.</p>
+                        <p>These are experimental features that may break in unexpected ways. Use with caution.</p>
                         {features}
-                        <br />
-                        <div className="mx_UserSettings_button" onClick={() => window.location.reload()}>Restart Vector</div>
                     </div>
                 </div>
             )