1 package us.codecraft.webmagic.scripts;
2
3 import java.util.HashMap;
4 import java.util.List;
5 import java.util.Map;
6 import java.util.Set;
7
8 import lombok.Getter;
9 import lombok.Setter;
10 import us.codecraft.webmagic.scripts.languages.JRuby;
11 import us.codecraft.webmagic.scripts.languages.Javascript;
12 import us.codecraft.webmagic.scripts.languages.Language;
13 import us.codecraft.webmagic.utils.WMCollections;
14
15 public class Params {
16 @Getter
17 Language language = new Javascript();
18
19 @Getter @Setter
20 String scriptFileName;
21
22 @Getter @Setter
23 List<String> urls;
24
25 @Getter @Setter
26 int thread = 1;
27
28 @Getter @Setter
29 int sleepTime = 1000;
30
31 private static Map<Language, Set<String>> alias;
32
33 public Params() {
34 alias = new HashMap<Language, Set<String>>();
35 alias.put(new Javascript(), WMCollections.<String>newHashSet("js", "javascript", "JavaScript", "JS"));
36 alias.put(new JRuby(), WMCollections.<String>newHashSet("ruby", "jruby", "Ruby", "JRuby"));
37 }
38
39 public void setLanguagefromArg(String arg) {
40 for (Map.Entry<Language, Set<String>> languageSetEntry : alias.entrySet()) {
41 if (languageSetEntry.getValue().contains(arg)) {
42 this.language = languageSetEntry.getKey();
43 return;
44 }
45 }
46 }
47 }