001/** 002 * Copyright (c) 2012, The University of Southampton and the individual contributors. 003 * All rights reserved. 004 * 005 * Redistribution and use in source and binary forms, with or without modification, 006 * are permitted provided that the following conditions are met: 007 * 008 * * Redistributions of source code must retain the above copyright notice, 009 * this list of conditions and the following disclaimer. 010 * 011 * * Redistributions in binary form must reproduce the above copyright notice, 012 * this list of conditions and the following disclaimer in the documentation 013 * and/or other materials provided with the distribution. 014 * 015 * * Neither the name of the University of Southampton nor the names of its 016 * contributors may be used to endorse or promote products derived from this 017 * software without specific prior written permission. 018 * 019 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 020 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 021 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 022 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR 023 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 024 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 025 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON 026 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 027 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 028 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 029 */ 030package org.openimaj.tools.reddit; 031 032import java.io.File; 033import java.io.IOException; 034 035import org.kohsuke.args4j.CmdLineException; 036import org.kohsuke.args4j.CmdLineParser; 037import org.kohsuke.args4j.Option; 038import org.kohsuke.args4j.ProxyOptionHandler; 039import org.openimaj.tools.FileToolsUtil; 040import org.openimaj.tools.InOutToolOptions; 041 042/** 043 * 044 * @author Jon Hare (jsh2@ecs.soton.ac.uk), Sina Samangooei (ss@ecs.soton.ac.uk) 045 * 046 */ 047public class RedditHarvesterOptions extends InOutToolOptions{ 048 049 /** 050 * the root reddit url 051 */ 052 public static final String REDDIT_BASE_URL = "http://www.reddit.com"; 053 054 @Option(name="--split-mode", aliases="-sm", required=false, usage="How to split the reddit inputs.", handler=ProxyOptionHandler.class) 055 SplitModeOption mode = SplitModeOption.TIME; 056 SplitMode modeOp = mode.getOptions(); 057 058 private String[] args; 059 /** 060 * @param args the arguments, prepared using the prepare method 061 * @param prepare whether prepare should be called now or later 062 */ 063 public RedditHarvesterOptions(String[] args, boolean prepare) { 064 this.args = args; 065 if(prepare) this.prepare(); 066 } 067 068 /** 069 * @param args the arguments, prepared using the prepare method 070 */ 071 public RedditHarvesterOptions(String[] args) { 072 this(args,true); 073 } 074 075 /** 076 * prepare the tool for running 077 */ 078 public void prepare(){ 079 CmdLineParser parser = new CmdLineParser(this); 080 try { 081 parser.parseArgument(args); 082 this.validate(); 083 } catch (CmdLineException e) { 084 System.err.println(e.getMessage()); 085 System.err.println("Usage: java -jar RedditHarvester.jar [options...] [files...]"); 086 parser.printUsage(System.err); 087 System.err.println(this.getExtractUsageInfo()); 088 System.exit(1); 089 } 090 091 } 092 093 private String getExtractUsageInfo() { 094 return ""; 095 } 096 097 private void validate() throws CmdLineException { 098 if(this.getInput() == null || this.getInput() == ""){ 099 this.setInput("/all.json"); 100 } 101 else if(!this.getInput().endsWith(".json")){ 102 this.setInput(this.getInput()+".json"); 103 } 104 this.setOutput(this.getOutput() + File.separator + this.getInput()); 105 try { 106 File fileOut = FileToolsUtil.validateLocalOutput(this); 107 fileOut.mkdirs(); 108 109 } catch (IOException e) { 110 throw new CmdLineException(null,e.getMessage()); 111 } 112 113 } 114}