AfflictionNetworks/ooo-user
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added 'create' and 'make_admin' commands for db bootstrap.

Browse Source 
When setting up a local deployment, it's nice to be able to create a user
without having to go to all the trouble of setting up the register webapp.
This commit is contained in:
Michael Bayne
2016-12-18 09:24:41 -08:00
parent 218a333f15
commit ec027a560e
1 changed files with 66 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/threerings/user/tools/UserTool.java
+66 -5
View File
@@ -4,8 +4,11 @@
package com.threerings.user.tools;
import com.samskivert.jdbc.StaticConnectionProvider;
import com.samskivert.servlet.user.Password;
import com.samskivert.servlet.user.Username;
import com.samskivert.util.Config;
import com.threerings.user.OOOUser;
import com.threerings.user.OOOUserRepository;
/**
@@ -13,20 +16,36 @@ import com.threerings.user.OOOUserRepository;
*/
public class UserTool
{
public static final String PROPS_NAME = System.getProperty("ooo.propfile", "threerings");
public static void main (String[] args)
{
if (args.length == 0) {
System.err.println(USAGE);
System.exit(255);
failWithUsage();
}
Config config = new Config("threerings");
Config config = new Config(PROPS_NAME);
try {
OOOUserRepository repo = new OOOUserRepository(
new StaticConnectionProvider(config.getSubProperties("db")));
if (args[0].equals("prune")) {
switch (args[0]) {
case "prune":
repo.pruneUsers(PRUNE_DAYS);
break;
case "create":
createUser(repo, args);
break;
case "make_admin":
makeAdmin(repo, args);
break;
default:
System.err.println("Unknown command: " + args[0]);
failWithUsage();
break;
}
} catch (Exception e) {
@@ -34,7 +53,49 @@ public class UserTool
}
}
protected static final String USAGE = "Usage: UserTool [prune]";
protected static void failWithUsage () {
for (String usage : USAGE) {
System.err.println(usage);
}
System.exit(255);
}
protected static void createUser(OOOUserRepository repo, String[] args) throws Exception {
if (args.length != 4) {
System.err.println("Invalid 'create' args. Usage:");
System.err.println(USAGE[2]);
System.exit(255);
}
Username username = new Username(args[1]);
Password password = Password.makeFromClear(args[2]);
int id = repo.createUser(username, password, args[3], OOOUser.DEFAULT_SITE_ID, 0);
System.out.println("Created user '" + username + "' with id " + id);
}
protected static void makeAdmin(OOOUserRepository repo, String[] args) throws Exception {
if (args.length != 2) {
System.err.println("Invalid 'make_admin' args. Usage:");
System.err.println(USAGE[3]);
System.exit(255);
}
String username = args[1];
OOOUser user = repo.loadUser(username, false);
if (user.holdsToken(OOOUser.ADMIN)) {
System.err.println("User " + username + " (id: " + user.userId + ") already admin.");
System.exit(255);
}
user.addToken(OOOUser.ADMIN);
repo.updateUser(user);
System.out.println("Marked user " + username + " (id: " + user.userId + ") as admin.");
}
protected static final String[] USAGE = {
"Usage: UserTool [prune|create]",
" prune - prunes expired users",
" create username email password - create a user account",
" make_admin username - marks a user account as admin"
};
/** Users that have been purged from all other games for 30 days are toast. */
protected static final int PRUNE_DAYS = 30;