#!/bin/bash
#####################################################################
# wifiConek.sh
# Enter network name and key
# This is a script only for root
##########################################################R###########
# Failed
function usage() {
    echo -e "\n\tUsage: `basename $0` \"<essid>\" <key>" \
            "\n\t       Enter <ascii prefaces>\n"
    die
}

# Kill
function die() {
    echo -e "\n\t       `basename $0` ERROR: $1 <*>\n"
    exit 1
}

# Setup
# Get ESSID
if [[ $1 ]]
then echo -e "\n\t>>>Starting by dropping wlan0"
     ifconfig wlan0 down               # Drop wlan0
     sleep 3
     dhclient -r wlan0 -q              # Drop previous connections
     echo -e "\n\t>>>Dropped previous connection(s)"
     ifconfig wlan0 up                 # Activate wlan0
     echo -e "\n\t>>>wlan0 activated"

     echo -e "\n\tESSID set to $1"
     iwconfig wlan0 essid "$1"         # Set wlan0 to connect to $1

# Assign Key
     if [[ $2 ]]
     then echo -e "\n\tKey set to $2"
          iwconfig wlan0 key s:"$2"    # If exists, set wlan0 use key
     fi

# Connect
     echo -e "\n\t>>>Set wlan0 to Managed\n"
     iwconfig wlan0 mode Managed       # Set wlan0 for managed network
     echo -e "\n\t>>>Requesting network IP address\n"
     dhclient wlan0                    # Get IP address
# Success
     if [[ $? -eq 0 ]]
     then echo -e "\n\n\t>>>Connected to $1!"
          exit 0
     else die ">>>Failed to connect to $1"  # Failure
     fi
else usage                             # No variables
fi