#!/bin/sh

# wlan.sh -- login/logout for ZJUWLAN
#
# Requirements: curl
#
# Copyright (C) 2014 Zhang Hai <Dreaming.in.Code.ZH@Gmail.com>
# Copyright (C) 2014 Zeno Zeng <zenoofzeng@gmail.com>
# Copyright (C) 2017 Wu Fan <wfwf1997@gmail.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see
# <http://www.gnu.org/licenses/>.

BASEDIR=$(dirname $0)

USER="${BASEDIR}/user.sh"

USERNAME=$($USER get)
PASSWORD=$($USER getpwd $USERNAME)

logout() {
    USERNAME=$1
    PASSWORD=$2

    echo "Logout: ${USERNAME}"
    RESPONSE=$(curl "https://net.zju.edu.cn/include/auth_action.php" -H "Content-Type: application/x-www-form-urlencoded" -d "action=logout&username=${USERNAME}&password=${PASSWORD}&ajax=1" -s)

    if [[ $? -eq 60 ]]; then
        CHOOSE=''
        read -p "There's an issue with ZJUNET's CA certificates, do you want to do it anyway? [yes/NO] " CHOOSE
        if [[ $CHOOSE == "YES" ]] || [[ $CHOOSE == "yes" ]]; then
            RESPONSE=$(curl "https://net.zju.edu.cn/include/auth_action.php" -H "Content-Type: application/x-www-form-urlencoded" -d "action=logout&username=${USERNAME}&password=${PASSWORD}&ajax=1" -s -k)
        fi
    fi
        
    case "${RESPONSE}" in
        *ok*)
            echo "Logout: success."
            ;;
        *已断开*)
            echo "Logout: success."
            ;;
        *)
            echo "Logout: unsuccess."
            echo "${RESPONSE}"
            exit 1;
            ;;
    esac
}

login() {
    USERNAME=$1
    PASSWORD=$2

    STATUS=$(curl http://g.cn/generate_204 -I -s | grep HTTP | awk {'print $2'})
    if [[ STATUS -eq 204 ]]; then
        echo "You have already logged in."
        exit 0
    fi

    echo "Login: ${USERNAME}"
    RESPONSE=$(curl "https://net.zju.edu.cn/include/auth_action.php" -H "Content-Type: application/x-www-form-urlencoded" -d "action=login&username=${USERNAME}&password=${PASSWORD}&ac_id=3&user_ip=&nas_ip=&user_mac=&save_me=1&ajax=1" -s)

    if [[ $? -eq 60 ]]; then
        CHOOSE=''
        read -p "There's an issue with ZJUNET's CA certificates, do you want to do it anyway? [yes/NO] " CHOOSE
        if [[ $CHOOSE == "YES" ]] || [[ $CHOOSE == "yes" ]]; then
            RESPONSE=$(curl "https://net.zju.edu.cn/include/auth_action.php" -H "Content-Type: application/x-www-form-urlencoded" -d "action=login&username=${USERNAME}&password=${PASSWORD}&ac_id=3&user_ip=&nas_ip=&user_mac=&save_me=1&ajax=1" -s -k)
        fi
    fi

    case "${RESPONSE}" in
        *help.html*)
            echo "Login: success."
            ;;
        *login_ok*)
            echo "Login: success."
            ;;
        *E2532*)
            echo "Login: failed. Please retry after 10s." >&2
            echo "Login: ${RESPONSE}" >&2
            exit 1
            ;;
        *)
            echo "Login: failed." >&2
            echo "Login: ${RESPONSE}" >&2
            exit 1
            ;;
    esac
}


case "$1" in
    disconnect)
        logout $USERNAME $PASSWORD
        ;;
    -d)
        logout $USERNAME $PASSWORD
        ;;
    *)
        login $USERNAME $PASSWORD
        ;;
esac

"${BASEDIR}/sudo.sh" "${BASEDIR}/route.sh"