forked from j62/ctbrec
1
0
Fork 0

Implement login and token balance

This commit is contained in:
0xboobface 2019-04-15 19:17:41 +02:00
parent 50c38d16a7
commit 8f5c7ac9c4
1 changed files with 29 additions and 31 deletions

View File

@ -44,7 +44,7 @@ public class Flirt4Free extends AbstractSite {
@Override
public String getBuyTokensLink() {
return BASE_URI;
return getAffiliateLink();
}
@Override
@ -58,28 +58,26 @@ public class Flirt4Free extends AbstractSite {
@Override
public Double getTokenBalance() throws IOException {
return 0d;
// if (!credentialsAvailable()) {
// throw new IOException("Account settings not available");
// }
//
// String username = Config.getInstance().getSettings().camsodaUsername;
// String url = BASE_URI + "/api/v1/user/" + username;
// Request request = new Request.Builder().url(url).build();
// try(Response response = getHttpClient().execute(request)) {
// if(response.isSuccessful()) {
// JSONObject json = new JSONObject(response.body().string());
// if(json.has("user")) {
// JSONObject user = json.getJSONObject("user");
// if(user.has("tokens")) {
// return (double) user.getInt("tokens");
// }
// }
// } else {
// throw new HttpException(response.code(), response.message());
// }
// }
// throw new RuntimeException("Tokens not found in response");
if (!credentialsAvailable()) {
throw new IOException("Account settings not available");
}
login();
String url = Flirt4Free.BASE_URI + "/my-account/";
Request request = new Request.Builder()
.url(url)
.header("User-Agent", Config.getInstance().getSettings().httpUserAgent)
.build();
try(Response response = getHttpClient().execute(request)) {
if(response.isSuccessful()) {
String body = response.body().string();
String text = HtmlParser.getText(body, "span[class~=user-credit-amount] strong");
return Double.parseDouble(text);
} else {
throw new RuntimeException("Tokens not found in response");
}
}
}
@Override
@ -108,7 +106,7 @@ public class Flirt4Free extends AbstractSite {
@Override
public boolean supportsTips() {
return false;
return true;
}
@Override
@ -126,7 +124,7 @@ public class Flirt4Free extends AbstractSite {
String url = BASE_URI + "/search/?query=" + URLEncoder.encode(q, "utf-8");
Request req = new Request.Builder()
.url(url)
.addHeader("User-Agent", Config.getInstance().getSettings().httpUserAgent)
.header("User-Agent", Config.getInstance().getSettings().httpUserAgent)
.build();
try (Response response = getHttpClient().execute(req)) {
if (response.isSuccessful()) {
@ -157,7 +155,7 @@ public class Flirt4Free extends AbstractSite {
@Override
public boolean credentialsAvailable() {
String username = Config.getInstance().getSettings().camsodaUsername;
String username = Config.getInstance().getSettings().flirt4freeUsername;
return username != null && !username.trim().isEmpty();
}