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 }