121f97dc35b449dbf4014a7d19c6d51e0f876cc4
1 package be
.nikiroo
.fanfix
.supported
;
3 import java
.io
.IOException
;
4 import java
.io
.InputStream
;
5 import java
.net
.MalformedURLException
;
7 import java
.util
.AbstractMap
;
8 import java
.util
.ArrayList
;
9 import java
.util
.HashMap
;
10 import java
.util
.List
;
12 import java
.util
.Map
.Entry
;
13 import java
.util
.Scanner
;
15 import be
.nikiroo
.fanfix
.Instance
;
16 import be
.nikiroo
.fanfix
.bundles
.Config
;
17 import be
.nikiroo
.fanfix
.data
.MetaData
;
18 import be
.nikiroo
.utils
.Image
;
19 import be
.nikiroo
.utils
.Progress
;
20 import be
.nikiroo
.utils
.StringUtils
;
23 * Support class for <a href="https://sofurry.com/">SoFurry.com</a>, a Furry
24 * website supporting images and stories (we only retrieve the stories).
28 class YiffStar
extends BasicSupport_Deprecated
{
30 public String
getSourceName() {
35 protected MetaData
getMeta(URL source
, InputStream in
) throws IOException
{
36 MetaData meta
= new MetaData();
38 meta
.setTitle(getTitle(reset(in
)));
39 meta
.setAuthor(getAuthor(reset(in
)));
41 meta
.setTags(getTags(reset(in
)));
42 meta
.setSource(getSourceName());
43 meta
.setUrl(source
.toString());
44 meta
.setPublisher(getSourceName());
45 meta
.setUuid(source
.toString());
48 meta
.setSubject("Furry");
49 meta
.setType(getType().toString());
50 meta
.setImageDocument(false);
51 meta
.setCover(getCover(source
, reset(in
)));
57 protected boolean supports(URL url
) {
58 String host
= url
.getHost();
59 if (host
.startsWith("www.")) {
60 host
= host
.substring("www.".length());
63 return "sofurry.com".equals(host
);
67 protected boolean isHtml() {
72 public void login() throws IOException
{
73 // Note: this should not be necessary anymore
74 // (the "/guest" trick is enough)
75 String login
= Instance
.getConfig().getString(
76 Config
.LOGIN_YIFFSTAR_USER
);
77 String password
= Instance
.getConfig().getString(
78 Config
.LOGIN_YIFFSTAR_PASS
);
80 if (login
!= null && !login
.isEmpty() && password
!= null
81 && !password
.isEmpty()) {
83 Map
<String
, String
> post
= new HashMap
<String
, String
>();
84 post
.put("LoginForm[sfLoginUsername]", login
);
85 post
.put("LoginForm[sfLoginPassword]", password
);
86 post
.put("YII_CSRF_TOKEN", "");
87 post
.put("yt1", "Login");
88 post
.put("returnUrl", "/");
90 // Cookies will actually be retained by the cache manager once
93 .openNoCache(new URL("https://www.sofurry.com/user/login"),
94 this, post
, null, null).close();
99 public URL
getCanonicalUrl(URL source
) {
101 if (source
.getPath().startsWith("/view")) {
102 source
= guest(source
.toString());
103 // NO CACHE because we don't want the NotLoggedIn message later
104 InputStream in
= Instance
.getCache().openNoCache(source
, this,
106 String line
= getLine(in
, "/browse/folder/", 0);
108 String
[] tab
= line
.split("\"");
109 if (tab
.length
> 1) {
110 String groupUrl
= source
.getProtocol() + "://"
111 + source
.getHost() + tab
[1];
112 return guest(groupUrl
);
116 } catch (Exception e
) {
117 Instance
.getTraceHandler().error(e
);
120 return super.getCanonicalUrl(source
);
123 private List
<String
> getTags(InputStream in
) {
124 List
<String
> tags
= new ArrayList
<String
>();
126 String line
= getLine(in
, "class=\"sf-story-big-tags", 0);
128 String
[] tab
= StringUtils
.unhtml(line
).split(",");
129 for (String possibleTag
: tab
) {
130 String tag
= possibleTag
.trim();
131 if (!tag
.isEmpty() && !tag
.equals("...") && !tags
.contains(tag
)) {
140 private Image
getCover(URL source
, InputStream in
) throws IOException
{
142 List
<Entry
<String
, URL
>> chaps
= getChapters(source
, in
, null);
143 if (!chaps
.isEmpty()) {
144 in
= Instance
.getCache().open(chaps
.get(0).getValue(), this, true);
145 String line
= getLine(in
, " name=\"og:image\"", 0);
148 for (int i
= 0; i
< 3; i
++) {
149 pos
= line
.indexOf('"', pos
+ 1);
153 line
= line
.substring(pos
+ 1);
154 pos
= line
.indexOf('"');
156 line
= line
.substring(0, pos
);
157 if (line
.contains("/thumb?")) {
158 line
= line
.replace("/thumb?",
159 "/auxiliaryContent?type=25&");
160 return getImage(this, null, line
);
170 private String
getAuthor(InputStream in
) {
171 String author
= getLine(in
, "class=\"onlinestatus", 0);
172 if (author
!= null) {
173 return StringUtils
.unhtml(author
).trim();
179 private String
getTitle(InputStream in
) {
180 String title
= getLine(in
, "class=\"sflabel pagetitle", 0);
182 if (title
.contains("(series)")) {
183 title
= title
.replace("(series)", "");
185 return StringUtils
.unhtml(title
).trim();
192 protected String
getDesc(URL source
, InputStream in
) throws IOException
{
193 return null; // TODO: no description at all? Cannot find one...
197 protected List
<Entry
<String
, URL
>> getChapters(URL source
, InputStream in
,
198 Progress pg
) throws IOException
{
199 List
<Entry
<String
, URL
>> urls
= new ArrayList
<Entry
<String
, URL
>>();
201 @SuppressWarnings("resource")
202 Scanner scan
= new Scanner(in
, "UTF-8");
203 scan
.useDelimiter("\\n");
204 while (scan
.hasNext()) {
205 String line
= scan
.next();
206 if (line
.contains("\"/view/") && line
.contains("title=")) {
207 String
[] tab
= line
.split("\"");
208 if (tab
.length
> 5) {
209 String link
= tab
[5];
210 if (link
.startsWith("/")) {
211 link
= source
.getProtocol() + "://" + source
.getHost()
214 urls
.add(new AbstractMap
.SimpleEntry
<String
, URL
>(
215 StringUtils
.unhtml(line
).trim(), guest(link
)));
224 protected String
getChapterContent(URL source
, InputStream in
, int number
,
225 Progress pg
) throws IOException
{
226 StringBuilder builder
= new StringBuilder();
228 String startAt
= "id=\"sfContentBody";
229 String endAt
= "id=\"recommendationArea";
232 @SuppressWarnings("resource")
233 Scanner scan
= new Scanner(in
, "UTF-8");
234 scan
.useDelimiter("\\n");
235 while (scan
.hasNext()) {
236 String line
= scan
.next();
237 if (!ok
&& line
.contains(startAt
)) {
239 } else if (ok
&& line
.contains(endAt
)) {
245 builder
.append(line
);
250 return builder
.toString();
254 * Return a {@link URL} from the given link, but add the "/guest" part to it
255 * to make sure we don't need to be logged-in to see it.
260 * @return the {@link URL}
262 * @throws MalformedURLException
263 * in case of data error
265 private URL
guest(String link
) throws MalformedURLException
{
266 if (link
.contains("?")) {
267 if (link
.contains("/?")) {
268 return new URL(link
.replace("?", "guest?"));
271 return new URL(link
.replace("?", "/guest?"));
274 return new URL(link
+ "/guest");