first draft of sequence generator

Author: drallieiv

KinanCity-core/src/main/resources/accounts.example.csv KinanCity-core/accounts.example.csv

@@ -11,9 +11,13 @@ public enum CliOptions {
 
 	CK("ck", "captchaKey", true, "2 captcha API Key"),
 
-	SINGLE_EMAIL("m", "mail", true, "account email address"),
+	EMAIL("m", "mail", true, "account email address"),
 	SINGLE_USERNAME("u", "username", true, "account username/login"),
-	SINGLE_PASSWORD("p", "password", true, "account password"),
+	PASSWORD("p", "password", true, "account password"),
+	
+	SEQ_ACCOUNTS_COUNT("c", "count", true, "number of accounts to generate"),
+	SEQ_ACCOUNTS_START("s", "startnum", true, "number of the first one"),
+	SEQ_ACCOUNTS_FORMAT("f", "format", true, "format of the username : prefix*****suffix with enough *"),
 
 	MULTIPLE_ACCOUNTS("a", "accounts", true, "name of an account file");
 

KinanCity-core/src/main/resources/config.example.properties KinanCity-core/config.example.properties

@@ -1,6 +1,8 @@
 package com.kinancity.core;
 
 import java.io.BufferedWriter;
+import java.io.File;
+import java.io.FileInputStream;
 import java.io.FileWriter;
 import java.io.IOException;
 import java.io.InputStream;
@@ -32,19 +34,19 @@ public class Configuration {
 	private String twoCaptchaApiKey;
 
 	private String mailHost;
-	
+
 	private int nbThreads = 5;
-	
+
 	private CaptchaProvider captchaProvider;
-	
+
 	private ProxyManager proxyManager;
-	
+
 	private String resultLogFilename = "result.csv";
-	
+
 	private PrintWriter resultLogWriter;
-	
+
 	private boolean initDone = false;
-	
+
 	// If true, everything will be mocked
 	private boolean dryRun = false;
 
@@ -57,62 +59,65 @@ public static Configuration getInstance() {
 		return instance;
 	}
 
-	public void init() throws ConfigurationException{
-		
+	public void init() throws ConfigurationException {
+
 		try {
-			if(captchaProvider == null){
+			if (captchaProvider == null) {
 				captchaProvider = new TwoCaptchaService(twoCaptchaApiKey, dryRun);
 			}
-			
-			if(proxyManager == null){
+
+			if (proxyManager == null) {
 				proxyManager = new ProxyManager();
 				// Add Direct connection
 				proxyManager.addProxy(new ProxyInfo(new NintendoTimeLimitPolicy(), new NoProxy()));
 			}
-			
+
 			resultLogWriter = new PrintWriter(new FileWriter(resultLogFilename, true));
 		} catch (IOException e) {
 			throw new ConfigurationException(e);
 		}
-		
+
 		initDone = true;
 	}
-	
+
 	/**
 	 * Check if all config are OK
 	 * 
 	 * @return
 	 */
 	public boolean checkConfiguration() {
-		if(!initDone){
+		if (!initDone) {
 			try {
 				init();
 			} catch (ConfigurationException e) {
 				logger.error("Configuration Init Failed", e);
 				return false;
 			}
 		}
-		
+
 		if (captchaProvider == null) {
 			return false;
 		}
 
 		return true;
 	}
-	
+
 	/**
 	 * Load config from prop file
 	 * 
-	 * @param configFile
+	 * @param configFilePath
 	 */
-	private void load(String configFile) {
+	private void load(String configFilePath) {
 		try {
 			Properties prop = new Properties();
-			InputStream in = getClass().getResourceAsStream("/" + CONFIG_FILE);
-			if (in == null) {
+			File configFile = new File(CONFIG_FILE);
+
+			if (!configFile.exists()) {
 				logger.warn("Skipping loading configuration file, you may copy config.example.properties as config.properties and edit your configuration");
 				return;
 			}
+
+			InputStream in = new FileInputStream(configFile);
 			prop.load(in);
 			in.close();
 

KinanCity-core/src/main/java/com/kinancity/core/CliOptions.java

@@ -2,6 +2,8 @@
 
 import java.io.PrintWriter;
 import java.io.StringWriter;
+import java.util.ArrayList;
+import java.util.List;
 
 import org.apache.commons.cli.CommandLine;
 import org.apache.commons.cli.CommandLineParser;
@@ -17,6 +19,7 @@
 import com.kinancity.core.creation.PtcCreationSummary;
 import com.kinancity.core.data.AccountData;
 import com.kinancity.core.errors.AccountCreationException;
+import com.kinancity.core.generator.impl.SequenceAccountGenerator;
 
 public class KinanCityCli {
 
@@ -26,7 +29,7 @@ public static void main(String[] args) {
 
 		Thread t = Thread.currentThread();
 		t.setName("Kinan City");
-		
+
 		try {
 			LOGGER.info(" -- Start Kinan City CLI -- ");
 
@@ -35,42 +38,73 @@ public static void main(String[] args) {
 
 			// Dry Run Mode
 			options.addOption(CliOptions.DRY_RUN.asOption());
-			
+
 			// Creates only 1 account
-			options.addOption(CliOptions.SINGLE_EMAIL.asOption());
+			options.addOption(CliOptions.EMAIL.asOption());
 			options.addOption(CliOptions.SINGLE_USERNAME.asOption());
-			options.addOption(CliOptions.SINGLE_PASSWORD.asOption());
+			options.addOption(CliOptions.PASSWORD.asOption());
 
 			// Create Multiple accounts
 			options.addOption(CliOptions.MULTIPLE_ACCOUNTS.asOption());
 
+			// Create a Sequence of accounts
+			options.addOption(CliOptions.SEQ_ACCOUNTS_COUNT.asOption());
+			options.addOption(CliOptions.SEQ_ACCOUNTS_FORMAT.asOption());
+			options.addOption(CliOptions.SEQ_ACCOUNTS_START.asOption());
+
 			// Captcha key given at commandLine
 			options.addOption(CliOptions.CK.asOption());
 
 			CommandLineParser parser = new DefaultParser();
 			CommandLine cmd = parser.parse(options, args);
 
 			Configuration config = Configuration.getInstance();
-			
-			if(cmd.hasOption(CliOptions.DRY_RUN.shortName)){
+
+			if (cmd.hasOption(CliOptions.DRY_RUN.shortName)) {
 				config.setDryRun(true);
-			}			
+			}
 
 			if (cmd.hasOption(CliOptions.CK.shortName)) {
 				config.setTwoCaptchaApiKey(cmd.getOptionValue(CliOptions.CK.shortName));
 			}
 
 			if (config.checkConfiguration()) {
-				
+
 				PtcAccountCreator creator = new PtcAccountCreator(config);
 
-				if (cmd.hasOption(CliOptions.SINGLE_EMAIL.shortName) && cmd.hasOption(CliOptions.SINGLE_USERNAME.shortName) && cmd.hasOption(CliOptions.SINGLE_PASSWORD.shortName)) {
+				if (cmd.hasOption(CliOptions.SEQ_ACCOUNTS_FORMAT.shortName) && cmd.hasOption(CliOptions.SEQ_ACCOUNTS_COUNT.shortName) && cmd.hasOption(CliOptions.EMAIL.shortName) && cmd.hasOption(CliOptions.PASSWORD.shortName)) {
+
+					SequenceAccountGenerator generator = new SequenceAccountGenerator();
+					generator.setBaseEmail(cmd.getOptionValue(CliOptions.EMAIL.shortName));
+					generator.setStaticPassword(cmd.getOptionValue(CliOptions.PASSWORD.shortName));
+					
+					generator.setUsernamePattern(cmd.getOptionValue(CliOptions.SEQ_ACCOUNTS_FORMAT.shortName));
+					generator.setNbAccounts(Integer.parseInt(cmd.getOptionValue(CliOptions.SEQ_ACCOUNTS_COUNT.shortName)));
+					generator.setStartFrom(Integer.parseInt(cmd.getOptionValue(CliOptions.SEQ_ACCOUNTS_START.shortName, "0")));
+					
+					List<AccountData> accountsToCreate = new ArrayList<>();
+					AccountData account;
+					while ((account = generator.nextAccountData()) != null) {
+						accountsToCreate.add(account);
+					}
+
+					try {
+						PtcCreationSummary summary = creator.createAccounts(accountsToCreate);
+
+						LOGGER.info(" All creations DONE : {}", summary);
+						System.exit(0);
+					} catch (AccountCreationException e) {
+						LOGGER.error("\n Account Creation Error : {}", e.getMessage());
+						System.exit(1);
+					}
+
+				} else if (cmd.hasOption(CliOptions.EMAIL.shortName) && cmd.hasOption(CliOptions.SINGLE_USERNAME.shortName) && cmd.hasOption(CliOptions.PASSWORD.shortName)) {
 					LOGGER.info("Create a single account");
 
 					AccountData account = new AccountData();
-					account.setEmail(cmd.getOptionValue(CliOptions.SINGLE_EMAIL.shortName));
+					account.setEmail(cmd.getOptionValue(CliOptions.EMAIL.shortName));
 					account.setUsername(cmd.getOptionValue(CliOptions.SINGLE_USERNAME.shortName));
-					account.setPassword(cmd.getOptionValue(CliOptions.SINGLE_PASSWORD.shortName));
+					account.setPassword(cmd.getOptionValue(CliOptions.PASSWORD.shortName));
 
 					try {
 						PtcCreationResult result = creator.createAccount(account);
@@ -84,23 +118,26 @@ public static void main(String[] args) {
 					String accountFileName = cmd.getOptionValue(CliOptions.MULTIPLE_ACCOUNTS.shortName);
 
 					PtcCreationSummary summary = creator.createAccounts(accountFileName);
-					
-					LOGGER.info(" All creations DONE : {}",summary);
+
+					LOGGER.info(" All creations DONE : {}", summary);
 					System.exit(0);
 
 				} else {
 
-					LOGGER.error("\ninvalid arguments\n");
+					LOGGER.error("invalid arguments\n");
 
 					HelpFormatter formatter = new HelpFormatter();
 					StringWriter out = new StringWriter();
 					PrintWriter writer = new PrintWriter(out);
 
-					String cmdLineSyntax = " (-m <email>  -u <username> -p <password>) (-a <accounts.csv> ) [-ck <captchakey>]";
+					String cmdLineSyntax = " one of \n"
+							+ "  -m <email>  -u <username> -p <password> \n"
+							+ "  -a <accounts.csv> \n"
+							+ "  -m <email> -c <#ofAccounts> -f <format**> -p <password> (-s <first#>)\n"
+							+ " and optional : -ck <captchakey> \n\n";
+					formatter.setWidth(180);
 					formatter.printHelp(cmdLineSyntax, options);
 
-					String usage = out.toString();
-					LOGGER.error(usage);
 					System.exit(0);
 				}
 

KinanCity-core/src/main/java/com/kinancity/core/Configuration.java

@@ -104,7 +104,7 @@ public PtcCreationSummary createAccounts(String accountFileName) {
 	 * @return 
 	 * @throws AccountCreationException
 	 */
-	private PtcCreationSummary createAccounts(List<AccountData> accountsToCreate) throws AccountCreationException {
+	public PtcCreationSummary createAccounts(List<AccountData> accountsToCreate) throws AccountCreationException {
 		logger.info("Start creating {} account in batch loaded from csv");
 
 		// add all accounts to pool

KinanCity-core/src/main/java/com/kinancity/core/KinanCityCli.java

@@ -0,0 +1,7 @@
+package com.kinancity.core.generator;
+
+import com.kinancity.core.data.AccountData;
+
+public interface AccountGenerator {
+	AccountData nextAccountData();
+}

KinanCity-core/src/main/java/com/kinancity/core/creation/PtcAccountCreator.java

@@ -0,0 +1,5 @@
+package com.kinancity.core.generator;
+
+public interface EmailGenerator {
+	String generateEmail(String... args);
+}

KinanCity-core/src/main/java/com/kinancity/core/generator/AccountGenerator.java

@@ -0,0 +1,5 @@
+package com.kinancity.core.generator;
+
+public interface PasswordGenerator {
+	String generatePassword();
+}

KinanCity-core/src/main/java/com/kinancity/core/generator/EmailGenerator.java

@@ -0,0 +1,5 @@
+package com.kinancity.core.generator;
+
+public interface UsernameGenerator {
+	String generateUsername();
+}

KinanCity-core/src/main/java/com/kinancity/core/generator/PasswordGenerator.java

@@ -0,0 +1,31 @@
+package com.kinancity.core.generator.impl;
+
+import com.kinancity.core.generator.EmailGenerator;
+
+/**
+ * Generate and email using the Plus Trick
+ * 
+ * Output will be [email protected]
+ * 
+ * @author drallieiv
+ *
+ */
+public class PlusTrickEmailGenerator implements EmailGenerator {
+
+	@Override
+	public String generateEmail(String... args) {
+		if(args.length < 2){
+			throw new IllegalArgumentException("Plus Trick needs the base email and the plus part");
+		}
+		
+		String baseMail = args[0];
+		String username = args[1];
+		
+		return generateEmail(baseMail, username);
+	}
+	
+	public String generateEmail(String baseMail, String username) {	
+		return baseMail.replaceFirst("@", "+"+username+"@");
+	}
+
+}