use get/getall/getpwd

user.sh

@@ -24,26 +24,55 @@ mkdir -p $DIR
 
 # dispatch
 case "$1" in
+
     add)
-        echo "USERNAME: "
+        read -p "USERNAME: " USERNAME
-        read USERNAME
+        read -p "PASSWORD: " PASSWORD
-        echo "PASSWORD: "
+        echo $PASSWORD > "$DIR/${USERNAME}"
-        read PASSWORD
-        echo $PASSWORD > "$DIR/$USERNAME"
         ;;
+
     edit)
-        echo "USERNAME: "
+        read -p "USERNAME: " USERNAME
-        read USERNAME
+        read -p "PASSWORD: " PASSWORD
-        echo "PASSWORD: "
+        echo $PASSWORD > "$DIR/${USERNAME}"
-        read PASSWORD
-        echo $PASSWORD > "$DIR/$USERNAME"
         ;;
+
     delete)
-        echo "USERNAME: "
+        read -p "USERNAME: " USERNAME
-        read USERNAME
+        rm -i "$DIR/${USERNAME}"
-        rm -i "$DIR/$USERNAME"
         ;;
+
     list)
-        ls $DIR
+        ls -1A $DIR
+        ;;
+
+    # Get a user
+    # @private
+    get)
+        COUNT=$(ls -1A $DIR | wc -l)
+        if [ "${COUNT}" -eq "0" ]; then
+            echo "No user found. Use 'zjunet user add' to add a user."
+            exit 1
+        else
+            if [ "${COUNT}" -gt "1" ]; then
+                USERS=$(ls -1A $DIR | xargs | tr "\n" " ")
+                read -p "Choose User [ ${USERS}]: " USERNAME
+            else
+                USERNAME=$(ls -1 $DIR | head -n1)
+            fi
+            echo $USERNAME
+        fi
+        ;;
+
+    # Get all users
+    # @private
+    getall)
+        ls -1A $DIR | xargs | tr "\n" " "
+        ;;
+
+    # @private
+    getpwd)
+        USERNAME=$2
+        cat "$DIR/${USERNAME}"
         ;;
 esac

wlan.sh

@@ -1,4 +1,6 @@
-# wlan.sh
+#!/bin/sh
+
+# wlan.sh -- login/logout for ZJUWLAN
 #
 # Copyright (C) 2014 Zhang Hai <Dreaming.in.Code.ZH@Gmail.com>
 # Copyright (C) 2014 Zeno Zeng <zenoofzeng@gmail.com>
@@ -45,9 +47,18 @@ login() {
     fi    
 }
 
+BASEDIR=$(dirname $0)
+USER="${BASEDIR}/user.sh"
+
+USERNAME=$($USER get)
+
+PASSWORD=$($USER getpwd $USERNAME)
+
 case "$1" in
     login)
+        login $USERNAME $PASSWORD
         ;;
     logout)
+        logout $USERNAME $PASSWORD
         ;;
 esac

zjunet.sh

@@ -22,18 +22,18 @@ BASEDIR=$(dirname $0)
 
 case "$1" in
     user)
-        "$BASEDIR/user.sh" $2
+        "${BASEDIR}/user.sh" $2
         ;;
     ip)
         ip route show 0/0 | cut -d " " -f 3
         ;;
     wlan)
-        "$BASEDIR/wlan.sh"
+        "${BASEDIR}/wlan.sh"
         ;;
     vpn)
-        "$BASEDIR/vpn.sh"
+        "${BASEDIR}/vpn.sh"
         ;;
     dns)
-        "$BASEDIR/dns.sh"
+        "${BASEDIR}/dns.sh"
         ;;
 esac