mirror of https://github.com/ThmmyNoLife/mTHMMY
Ezerous
8 years ago
6 changed files with 352 additions and 291 deletions
@ -0,0 +1,291 @@ |
|||||
|
package gr.thmmy.mthmmy.session; |
||||
|
|
||||
|
import android.content.SharedPreferences; |
||||
|
import android.util.Log; |
||||
|
|
||||
|
import com.franmontiel.persistentcookiejar.PersistentCookieJar; |
||||
|
import com.franmontiel.persistentcookiejar.persistence.SharedPrefsCookiePersistor; |
||||
|
|
||||
|
import org.jsoup.Jsoup; |
||||
|
import org.jsoup.nodes.Document; |
||||
|
import org.jsoup.nodes.Element; |
||||
|
import org.jsoup.select.Elements; |
||||
|
|
||||
|
import java.util.List; |
||||
|
import java.util.regex.Matcher; |
||||
|
import java.util.regex.Pattern; |
||||
|
|
||||
|
import okhttp3.Cookie; |
||||
|
import okhttp3.FormBody; |
||||
|
import okhttp3.HttpUrl; |
||||
|
import okhttp3.OkHttpClient; |
||||
|
import okhttp3.Request; |
||||
|
import okhttp3.RequestBody; |
||||
|
import okhttp3.Response; |
||||
|
|
||||
|
/** |
||||
|
This class handles all session related operations (e.g. login, logout) |
||||
|
and stores data to SharedPreferences (session informarion and cookies). |
||||
|
*/ |
||||
|
public class SessionManager |
||||
|
{ |
||||
|
//Class TAG
|
||||
|
private static final String TAG = "SessionManager"; |
||||
|
|
||||
|
//Generic constants
|
||||
|
public static final HttpUrl indexUrl = HttpUrl.parse("https://www.thmmy.gr/smf/index.php"); |
||||
|
private static final HttpUrl loginUrl = HttpUrl.parse("https://www.thmmy.gr/smf/index.php?action=login2"); |
||||
|
private static final String guestName = "Guest"; |
||||
|
|
||||
|
//Response Codes
|
||||
|
public static final int SUCCESS = 0; |
||||
|
public static final int FAILURE = 1; //Generic Error
|
||||
|
public static final int WRONG_USER = 2; |
||||
|
public static final int WRONG_PASSWORD = 3; |
||||
|
public static final int EXCEPTION = 4; |
||||
|
|
||||
|
//Login status codes
|
||||
|
public static final int LOGGED_OUT = 0; |
||||
|
public static final int LOGGED_IN = 1; //Logged in (as a normal user)
|
||||
|
public static final int AS_GUEST = 2; //User chose to continue as guest
|
||||
|
|
||||
|
// Client & Cookies
|
||||
|
private OkHttpClient client; |
||||
|
private PersistentCookieJar cookieJar; |
||||
|
private SharedPrefsCookiePersistor cookiePersistor; //Used to explicitly edit cookies in cookieJar
|
||||
|
|
||||
|
//Shared Preferences & its keys
|
||||
|
private SharedPreferences sharedPrefs; |
||||
|
private static final String USERNAME = "Username"; |
||||
|
private static final String LOGOUT_LINK = "LogoutLink"; |
||||
|
private static final String LOGIN_STATUS = "IsLoggedIn"; |
||||
|
|
||||
|
//Constructor
|
||||
|
public SessionManager(OkHttpClient client, PersistentCookieJar cookieJar, |
||||
|
SharedPrefsCookiePersistor cookiePersistor, SharedPreferences sharedPrefs) |
||||
|
{ |
||||
|
this.client = client; |
||||
|
this.cookiePersistor=cookiePersistor; |
||||
|
this.cookieJar = cookieJar; |
||||
|
this.sharedPrefs = sharedPrefs; |
||||
|
} |
||||
|
|
||||
|
//------------------------------------AUTH BEGINS----------------------------------------------
|
||||
|
/** |
||||
|
* Login function with two options: (username, password) or nothing (using saved cookies). |
||||
|
* Always call it in a separate thread. |
||||
|
*/ |
||||
|
public int login(String... strings) |
||||
|
{ |
||||
|
Log.i(TAG, "Logging in..."); |
||||
|
|
||||
|
//Build the login request for each case
|
||||
|
Request request; |
||||
|
if (strings.length == 2) |
||||
|
{ |
||||
|
clearSessionData(); |
||||
|
|
||||
|
String loginName = strings[0]; |
||||
|
String password = strings[1]; |
||||
|
|
||||
|
RequestBody formBody = new FormBody.Builder() |
||||
|
.add("user", loginName) |
||||
|
.add("passwrd", password) |
||||
|
.add("cookielength", "-1") //-1 is forever
|
||||
|
.build(); |
||||
|
request = new Request.Builder() |
||||
|
.url(loginUrl) |
||||
|
.post(formBody) |
||||
|
.build(); |
||||
|
} |
||||
|
else |
||||
|
{ |
||||
|
request = new Request.Builder() |
||||
|
.url(loginUrl) |
||||
|
.build(); |
||||
|
} |
||||
|
|
||||
|
try { |
||||
|
//Make request & handle response
|
||||
|
Response response = client.newCall(request).execute(); |
||||
|
Document document = Jsoup.parse(response.body().string()); |
||||
|
|
||||
|
Element logoutButton = document.getElementById("logoutbtn"); //Attempt to find logout button
|
||||
|
if (logoutButton != null) //If logout button exists, login was successful
|
||||
|
{ |
||||
|
Log.i(TAG, "Login successful!"); |
||||
|
setPersistentCookieSession(); //Store cookies
|
||||
|
|
||||
|
//Edit SharedPreferences, save session's data
|
||||
|
sharedPrefs.edit().putInt(LOGIN_STATUS, LOGGED_IN).apply(); |
||||
|
sharedPrefs.edit().putString(USERNAME, extractUserName(document)).apply(); |
||||
|
sharedPrefs.edit().putString(LOGOUT_LINK, HttpUrl.parse(logoutButton.attr("href")).toString()).apply(); |
||||
|
|
||||
|
return SUCCESS; |
||||
|
} |
||||
|
else |
||||
|
{ |
||||
|
Log.i(TAG, "Login failed."); |
||||
|
|
||||
|
//Investigate login failure
|
||||
|
Elements error = document.select("b:contains(That username does not exist.)"); |
||||
|
if (error.size() == 1) { //Wrong username
|
||||
|
Log.i(TAG, "Wrong Username"); |
||||
|
return WRONG_USER; |
||||
|
} |
||||
|
|
||||
|
error = document.select("body:contains(Password incorrect)"); |
||||
|
if (error.size() == 1) { //Wrong password
|
||||
|
Log.i(TAG, "Wrong Password"); |
||||
|
return WRONG_PASSWORD; |
||||
|
} |
||||
|
|
||||
|
//Other error e.g. session was reset server-side
|
||||
|
clearSessionData(); //Clear invalid saved data
|
||||
|
return FAILURE; |
||||
|
} |
||||
|
//Handle exception
|
||||
|
} catch (Exception e) { |
||||
|
Log.w(TAG, "Login Exception: "+ e.getMessage(), e); |
||||
|
return EXCEPTION; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* A function that checks the validity of the current saved session (if it exists). |
||||
|
* If LOGIN_STATUS is true, it will call login() with cookies. This can only return |
||||
|
* the codes {SUCCESS, FAILURE, EXCEPTION}. EXCEPTION is considered a SUCCESS (e.g. no internet |
||||
|
* connection), at least until a more thorough handling of different exceptions is implemented. |
||||
|
* Always call it in a separate thread. |
||||
|
*/ |
||||
|
public void validateSession() |
||||
|
{ |
||||
|
Log.i(TAG, "Validating session..."); |
||||
|
|
||||
|
//Check if user is currently logged in
|
||||
|
int status = sharedPrefs.getInt(LOGIN_STATUS,LOGGED_OUT); |
||||
|
if(status==LOGGED_IN) |
||||
|
{ |
||||
|
int loginResult = login(); |
||||
|
if(loginResult == SUCCESS || loginResult == EXCEPTION) |
||||
|
return; |
||||
|
} |
||||
|
else if(status==AS_GUEST) |
||||
|
return; |
||||
|
|
||||
|
clearSessionData(); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Call this function when user explicitly chooses to continue as a guest (UI thread). |
||||
|
*/ |
||||
|
public void guestLogin() |
||||
|
{ |
||||
|
Log.i("TAG", "Continuing as a guest, as chosen by the user."); |
||||
|
clearSessionData(); |
||||
|
sharedPrefs.edit().putInt(LOGIN_STATUS, AS_GUEST).apply(); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
/** |
||||
|
* Logout function. Always call it in a separate thread. |
||||
|
*/ |
||||
|
public int logout() |
||||
|
{ |
||||
|
Log.i(TAG, "Logging out..."); |
||||
|
|
||||
|
Request request = new Request.Builder() |
||||
|
.url(sharedPrefs.getString(LOGOUT_LINK,"LogoutLink")) |
||||
|
.build(); |
||||
|
|
||||
|
try { |
||||
|
//Make request & handle response
|
||||
|
Response response = client.newCall(request).execute(); |
||||
|
Document document = Jsoup.parse(response.body().string()); |
||||
|
|
||||
|
Elements loginButton = document.select("[value=Login]"); //Attempt to find login button
|
||||
|
if (!loginButton.isEmpty()) //If login button exists, logout was successful
|
||||
|
{ |
||||
|
Log.i("Logout", "Logout successful!"); |
||||
|
return SUCCESS; |
||||
|
} |
||||
|
else |
||||
|
{ |
||||
|
Log.i(TAG, "Logout failed."); |
||||
|
return FAILURE; |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
Log.w(TAG, "Logout Exception: "+ e.getMessage(), e); |
||||
|
return EXCEPTION; |
||||
|
} finally { |
||||
|
//All data should always be cleared from device regardless the result of logout
|
||||
|
clearSessionData(); |
||||
|
Log.i(TAG,"Session data cleared."); |
||||
|
} |
||||
|
} |
||||
|
//--------------------------------------AUTH ENDS-----------------------------------------------
|
||||
|
|
||||
|
//---------------------------------------GETTERS------------------------------------------------
|
||||
|
public String getUsername() { |
||||
|
return sharedPrefs.getString(USERNAME, "Username"); |
||||
|
} |
||||
|
|
||||
|
public String getLogoutLink() { |
||||
|
return sharedPrefs.getString(LOGOUT_LINK, "LogoutLink"); |
||||
|
} |
||||
|
|
||||
|
public int getLogStatus() { |
||||
|
return sharedPrefs.getInt(LOGIN_STATUS, LOGGED_OUT); |
||||
|
} |
||||
|
|
||||
|
//--------------------------------------GETTERS END---------------------------------------------
|
||||
|
|
||||
|
//------------------------------------OTHER FUNCTIONS-------------------------------------------
|
||||
|
private void setPersistentCookieSession() |
||||
|
{ |
||||
|
List<Cookie> cookieList = cookieJar.loadForRequest(indexUrl); |
||||
|
|
||||
|
if (cookieList.size() == 2) |
||||
|
{ |
||||
|
if ((cookieList.get(0).name().equals("THMMYgrC00ki3")) |
||||
|
&& (cookieList.get(1).name().equals("PHPSESSID"))) { |
||||
|
Cookie.Builder builder = new Cookie.Builder(); |
||||
|
builder.name(cookieList.get(1).name()) |
||||
|
.value(cookieList.get(1).value()) |
||||
|
.domain(cookieList.get(1).domain()) |
||||
|
.expiresAt(cookieList.get(0).expiresAt()); |
||||
|
cookieList.remove(1); |
||||
|
cookieList.add(builder.build()); |
||||
|
cookiePersistor.clear(); |
||||
|
cookiePersistor.saveAll(cookieList); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void clearSessionData() |
||||
|
{ |
||||
|
cookieJar.clear(); |
||||
|
sharedPrefs.edit().clear().apply(); //Clear session data
|
||||
|
sharedPrefs.edit().putString(USERNAME, guestName).apply(); //User becomes guest
|
||||
|
sharedPrefs.edit().putInt(LOGIN_STATUS, LOGGED_OUT).apply(); |
||||
|
} |
||||
|
|
||||
|
private String extractUserName(Document doc) |
||||
|
{ |
||||
|
if (doc != null) { |
||||
|
Elements user = doc.select("div[id=myuser] > h3"); |
||||
|
|
||||
|
if (user.size() == 1) { |
||||
|
String txt = user.first().ownText(); |
||||
|
|
||||
|
Pattern pattern = Pattern.compile(", (.*?),"); |
||||
|
Matcher matcher = pattern.matcher(txt); |
||||
|
if (matcher.find()) |
||||
|
return matcher.group(1); |
||||
|
} |
||||
|
} |
||||
|
return null; |
||||
|
} |
||||
|
//----------------------------------OTHER FUNCTIONS END-----------------------------------------
|
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue