make shortid stores in localStorage

istutus · Jan 4, 2016 · f36682a · f36682a
1 parent e85fd2e
commit f36682a
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 6 deletions.
diff --git a/modules/io.js b/modules/io.js
@@ -23,9 +23,12 @@ module.exports = function(io) {
   });
 
   io.on('connection', socket => {
-    socket.shortid = shortid.generate().toLowerCase(); // generate shortid for an incomming connection
-    onlines.set(socket.shortid, socket); // add incomming connection to online table
-    socket.emit('shortid', socket.shortid);
+    socket.on('request shortid', function() {
+      onlines.delete(socket.shortid);
+      socket.shortid = shortid.generate().toLowerCase(); // generate shortid for a request
+      onlines.set(socket.shortid, socket); // add incomming connection to online table
+      socket.emit('shortid', socket.shortid);
+    });
   });
 
   io.on('disconnect', socket => {

diff --git a/public/index.html b/public/index.html
@@ -14,7 +14,13 @@
       <img class="logo" src="http://semantic-ui.com/examples/assets/images/logo.png" alt="logo">
       Forsaken Mail
     </div>
-    <div class="ui label item right floated mailaddress"><i class="mail icon copyable"></i><span id="shortid">请等待分配临时邮箱</span></div>
+    <div class="ui label item right floated mailaddress">
+      <i class="mail icon copyable"></i>
+      <div class="ui icon input">
+        <input id="shortid" type="text" placeholder="请等待分配临时邮箱" disabled>
+        <i id="refreshShortid" class="circular refresh link icon"></i>
+      </div>
+    </div>
   </div>
 </div>
 <div class="ui main container">

diff --git a/public/js/app.js b/public/js/app.js
@@ -24,9 +24,29 @@ $(function(){
   });
 
   var socket = io();
-  socket.on('shortid', function(id) {
+
+  var setMailAddress = function(id) {
+    localStorage.setItem('shortid', id);
     var mailaddress = id + '@' + location.hostname;
-    $('#shortid').text(mailaddress).siblings('i').attr('data-clipboard-text', mailaddress);
+    $('#shortid').val(mailaddress).parent().siblings('i').attr('data-clipboard-text', mailaddress);
+  };
+
+  if(('localStorage' in window)) {
+    var shortid = localStorage.getItem('shortid');
+    if(!shortid) {
+      socket.emit('request shortid', true);
+    }
+    else {
+      setMailAddress(shortid);
+    }
+  }
+
+  $('#refreshShortid').click(function() {
+    socket.emit('request shortid', true);
+  });
+
+  socket.on('shortid', function(id) {
+    setMailAddress(id);
   });
 
   socket.on('mail', function(mail) {