130 lines
4.5 KiB
Java
130 lines
4.5 KiB
Java
package ctbrec.sites.streamate;
|
|
|
|
import java.io.IOException;
|
|
import java.util.Collections;
|
|
import java.util.NoSuchElementException;
|
|
|
|
import org.json.JSONObject;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import ctbrec.Config;
|
|
import ctbrec.io.HttpClient;
|
|
import okhttp3.Cookie;
|
|
import okhttp3.HttpUrl;
|
|
import okhttp3.MediaType;
|
|
import okhttp3.Request;
|
|
import okhttp3.RequestBody;
|
|
import okhttp3.Response;
|
|
|
|
public class StreamateHttpClient extends HttpClient {
|
|
|
|
private static final transient Logger LOG = LoggerFactory.getLogger(StreamateHttpClient.class);
|
|
|
|
private Long userId;
|
|
private String saKey = "";
|
|
|
|
public StreamateHttpClient() {
|
|
super("streamate");
|
|
|
|
// this cookie is needed for the search
|
|
Cookie searchCookie = new Cookie.Builder()
|
|
.domain("streamate.com")
|
|
.name("Xld_rct")
|
|
.value("1")
|
|
.build();
|
|
getCookieJar().saveFromResponse(HttpUrl.parse(Streamate.BASE_URL), Collections.singletonList(searchCookie));
|
|
|
|
// try to load sakey from cookie
|
|
try {
|
|
Cookie cookie = getCookieJar().getCookie(HttpUrl.parse("https://www.streamate.com"), "sakey");
|
|
saKey = cookie.value();
|
|
} catch (NoSuchElementException e) {
|
|
// ignore
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public synchronized boolean login() throws IOException {
|
|
if(loggedIn) {
|
|
return true;
|
|
}
|
|
|
|
boolean cookiesWorked = checkLoginSuccess();
|
|
if(cookiesWorked) {
|
|
loggedIn = true;
|
|
LOG.debug("Logged in with cookies");
|
|
return true;
|
|
}
|
|
|
|
JSONObject loginRequest = new JSONObject();
|
|
loginRequest.put("email", Config.getInstance().getSettings().streamateUsername);
|
|
loginRequest.put("password", Config.getInstance().getSettings().streamatePassword);
|
|
loginRequest.put("referrerId", 0);
|
|
loginRequest.put("siteId", 1);
|
|
RequestBody body = RequestBody.create(MediaType.parse("application/json"), loginRequest.toString());
|
|
Request login = new Request.Builder()
|
|
.url(Streamate.BASE_URL + "/api/member/login")
|
|
.addHeader("User-Agent", Config.getInstance().getSettings().httpUserAgent)
|
|
.addHeader("Accept", "application/json, text/javascript, */*")
|
|
.addHeader("Accept-Language", "en")
|
|
.addHeader("Referer", Streamate.BASE_URL)
|
|
.addHeader("X-Requested-With", "XMLHttpRequest")
|
|
.post(body)
|
|
.build();
|
|
try (Response response = client.newCall(login).execute()) {
|
|
String content = response.body().string();
|
|
if(response.isSuccessful()) {
|
|
JSONObject json = new JSONObject(content);
|
|
loggedIn = json.has("sakey");
|
|
saKey = json.optString("sakey");
|
|
JSONObject account = json.getJSONObject("account");
|
|
userId = account.getLong("userid");
|
|
} else {
|
|
throw new IOException("Login failed: " + response.code() + " " + response.message());
|
|
}
|
|
response.close();
|
|
}
|
|
|
|
return loggedIn;
|
|
}
|
|
|
|
/**
|
|
* Check, if the login worked by loading the favorites
|
|
*/
|
|
public boolean checkLoginSuccess() {
|
|
String url = Streamate.BASE_URL + "/api/search/v1/favorites?host=streamate.com&domain=streamate.com";
|
|
url = url + "&page_number=1&results_per_page=48&sakey=" + saKey + "&userid=" + userId;
|
|
Request request = new Request.Builder()
|
|
.url(url)
|
|
.addHeader("User-Agent", Config.getInstance().getSettings().httpUserAgent)
|
|
.addHeader("Accept", "application/json, */*")
|
|
.addHeader("Accept-Language", "en")
|
|
.addHeader("Referer", Streamate.BASE_URL)
|
|
.build();
|
|
try(Response response = execute(request)) {
|
|
if (response.isSuccessful()) {
|
|
String content = response.body().string();
|
|
JSONObject json = new JSONObject(content);
|
|
if(json.optString("status").equals("SM_OK")) {
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
} else {
|
|
return false;
|
|
}
|
|
} catch(Exception e) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public String getSaKey() {
|
|
return saKey;
|
|
}
|
|
|
|
public Long getUserId() {
|
|
return userId;
|
|
}
|
|
}
|