private void preAwardCredits(DirectoryHandle preDir, HashMap<User, Long> awards) {
try {
for (InodeHandle file : preDir.getInodeHandlesUnchecked()) {
if (file.isFile()) {
User owner;
try {
owner =
GlobalContext.getGlobalContext()
.getUserManager()
.getUserByNameUnchecked(file.getUsername());
} catch (NoSuchUserException e) {
logger.warn("PRE: Cannot award credits to non-existing user", e);
continue;
} catch (UserFileException e) {
logger.warn("", e);
continue;
} catch (FileNotFoundException e) {
logger.warn("", e);
continue;
}
Long total = awards.get(owner);
if (total == null) {
total = 0L;
}
total =
(total
+ (long)
(file.getSize() * owner.getKeyedMap().getObjectFloat(UserManagement.RATIO)));
awards.put(owner, total);
}
if (file.isDirectory()) {
preAwardCredits((DirectoryHandle) file, awards);
}
}
} catch (FileNotFoundException e) {
logger.warn("FileNotFoundException on preAwardCredits()", e);
}
}
/**
* USAGE: site nuke <directory> <multiplier> <message> Nuke a directory
*
* <p>ex. site nuke shit 2 CRAP
*
* <p>This will nuke the directory 'shit' and remove x2 credits with the comment 'CRAP'.
*
* <p>NOTE: You can enclose the directory in braces if you have spaces in the name ex. site NUKE
* {My directory name} 1 because_i_dont_like_it
*
* <p>Q) What does the multiplier in 'site nuke' do? A) Multiplier is a penalty measure. If it is
* 0, the user doesn't lose any credits for the stuff being nuked. If it is 1, user only loses the
* amount of credits he gained by uploading the files (which is calculated by multiplying total
* size of file by his/her ratio). If multiplier is more than 1, the user loses the credits he/she
* gained by uploading, PLUS some extra credits. The formula is this: size * ratio + size *
* (multiplier - 1). This way, multiplier of 2 causes user to lose size * ratio + size * 1, so the
* additional penalty in this case is the size of nuked files. If the multiplier is 3, user loses
* size * ratio + size * 2, etc.
*
* @throws ImproperUsageException
*/
public CommandResponse doSITE_NUKE(CommandRequest request) throws ImproperUsageException {
if (!request.hasArgument()) {
throw new ImproperUsageException();
}
StringTokenizer st = new StringTokenizer(request.getArgument());
if (!st.hasMoreTokens()) {
throw new ImproperUsageException();
}
Session session = request.getSession();
int multiplier;
DirectoryHandle currentDir = request.getCurrentDirectory();
String nukeDirName = st.nextToken();
User requestUser = session.getUserNull(request.getUser());
String nukeDirPath = VirtualFileSystem.fixPath(nukeDirName);
if (!(nukeDirPath.startsWith(VirtualFileSystem.separator))) {
// Not a full path, let's make it one
if (request.getCurrentDirectory().isRoot()) {
boolean searchIndex =
request.getProperties().getProperty("search", "true").equalsIgnoreCase("true");
if (searchIndex) {
// Get dirs from index system
ArrayList<DirectoryHandle> dirsToNuke;
try {
dirsToNuke = NukeUtils.findNukeDirs(currentDir, requestUser, nukeDirPath);
} catch (FileNotFoundException e) {
logger.warn(e);
return new CommandResponse(550, e.getMessage());
}
ReplacerEnvironment env = new ReplacerEnvironment();
if (dirsToNuke.isEmpty()) {
env.add("searchstr", nukeDirPath);
return new CommandResponse(
550, session.jprintf(_bundle, _keyPrefix + "nuke.search.empty", env, requestUser));
} else if (dirsToNuke.size() == 1) {
nukeDirPath = dirsToNuke.get(0).getPath();
} else {
CommandResponse response = new CommandResponse(200);
for (DirectoryHandle nukeDir : dirsToNuke) {
try {
env.add("name", nukeDir.getName());
env.add("path", nukeDir.getPath());
env.add("owner", nukeDir.getUsername());
env.add("group", nukeDir.getGroup());
env.add("size", Bytes.formatBytes(nukeDir.getSize()));
response.addComment(
session.jprintf(_bundle, _keyPrefix + "nuke.search.item", env, requestUser));
} catch (FileNotFoundException e) {
logger.warn(
"Dir deleted after index search?, skip and continue: " + nukeDir.getPath());
}
}
response.addComment(
session.jprintf(_bundle, _keyPrefix + "nuke.search.end", env, requestUser));
// Return matching dirs and let user decide what to nuke
return response;
}
} else {
nukeDirPath = VirtualFileSystem.separator + nukeDirPath;
}
} else {
nukeDirPath =
request.getCurrentDirectory().getPath() + VirtualFileSystem.separator + nukeDirPath;
}
}
DirectoryHandle nukeDir;
try {
nukeDir = request.getCurrentDirectory().getDirectory(nukeDirPath, requestUser);
} catch (FileNotFoundException e) {
return StandardCommandManager.genericResponse("RESPONSE_550_REQUESTED_ACTION_NOT_TAKEN");
} catch (ObjectNotValidException e) {
return new CommandResponse(550, nukeDirPath + " is not a directory");
}
nukeDirName = nukeDir.getName();
if (!st.hasMoreTokens()) {
throw new ImproperUsageException();
}
try {
multiplier = Integer.parseInt(st.nextToken());
} catch (NumberFormatException ex) {
logger.warn(ex, ex);
return new CommandResponse(501, "Invalid multiplier: " + ex.getMessage());
}
String reason = "";
if (st.hasMoreTokens()) {
reason = st.nextToken("").trim();
}
NukeData nd;
try {
nd = NukeUtils.nuke(nukeDir, multiplier, reason, requestUser);
} catch (NukeException e) {
return new CommandResponse(500, "Nuke failed: " + e.getMessage());
}
CommandResponse response = new CommandResponse(200, "Nuke succeeded");
GlobalContext.getEventService().publishAsync(new NukeEvent(requestUser, "NUKE", nd));
ReplacerEnvironment env = new ReplacerEnvironment();
String section = GlobalContext.getGlobalContext().getSectionManager().lookup(nukeDir).getName();
env.add("section", section);
env.add("dir", nukeDirName);
env.add("path", nukeDirPath);
env.add("relpath", nukeDirPath.replaceAll("/.*?" + section + "/", ""));
env.add("multiplier", multiplier);
env.add("nukedamount", Bytes.formatBytes(nd.getAmount()));
env.add("reason", reason);
env.add("size", Bytes.formatBytes(nd.getSize()));
if (session instanceof BaseFtpConnection) {
response.addComment(session.jprintf(_bundle, _keyPrefix + "nuke", env, requestUser));
for (NukedUser nukeeObj : NukeBeans.getNukeeList(nd)) {
ReplacerEnvironment nukeeenv = new ReplacerEnvironment();
User nukee;
try {
nukee =
GlobalContext.getGlobalContext()
.getUserManager()
.getUserByName(nukeeObj.getUsername());
} catch (NoSuchUserException e1) {
// Unable to get user, does not exist.. skip announce for this user
continue;
} catch (UserFileException e1) {
// Error in user file.. skip announce for this user
continue;
}
long debt =
NukeUtils.calculateNukedAmount(
nukeeObj.getAmount(),
nukee.getKeyedMap().getObjectFloat(UserManagement.RATIO),
multiplier);
nukeeenv.add("nukedamount", Bytes.formatBytes(debt));
response.addComment(session.jprintf(_bundle, _keyPrefix + "nuke.nukees", nukeeenv, nukee));
}
}
return response;
}
/**
* USAGE: site unnuke <directory> <message> Unnuke a directory.
*
* <p>ex. site unnuke shit NOT CRAP
*
* <p>This will unnuke the directory 'shit' with the comment 'NOT CRAP'.
*
* <p>NOTE: You can enclose the directory in braces if you have spaces in the name ex. site unnuke
* {My directory name} justcause
*
* <p>You need to configure glftpd to keep nuked files if you want to unnuke. See the section
* about glftpd.conf.
*
* @throws ImproperUsageException
*/
public CommandResponse doSITE_UNNUKE(CommandRequest request) throws ImproperUsageException {
if (!request.hasArgument()) {
throw new ImproperUsageException();
}
StringTokenizer st = new StringTokenizer(request.getArgument());
Session session = request.getSession();
DirectoryHandle currentDir = request.getCurrentDirectory();
User user = session.getUserNull(request.getUser());
String toName = st.nextToken();
String toDir;
String nukeName;
if (!toName.startsWith(VirtualFileSystem.separator)) {
// Not a full path, let's make it one and append [NUKED]- if needed.
if (toName.startsWith("[NUKED]-")) {
nukeName = toName;
toName = toName.substring(8);
} else {
nukeName = "[NUKED]-" + toName;
}
if (request.getCurrentDirectory().isRoot()) {
boolean searchIndex =
request.getProperties().getProperty("search", "true").equalsIgnoreCase("true");
if (searchIndex) {
// Get dirs from index system
ArrayList<DirectoryHandle> dirsToUnNuke;
try {
dirsToUnNuke = NukeUtils.findNukeDirs(currentDir, user, nukeName);
} catch (FileNotFoundException e) {
logger.warn(e);
return new CommandResponse(550, e.getMessage());
}
ReplacerEnvironment env = new ReplacerEnvironment();
if (dirsToUnNuke.isEmpty()) {
env.add("searchstr", nukeName);
return new CommandResponse(
550, session.jprintf(_bundle, _keyPrefix + "unnuke.search.empty", env, user));
} else if (dirsToUnNuke.size() == 1) {
toDir = dirsToUnNuke.get(0).getParent().getPath() + VirtualFileSystem.separator;
} else {
CommandResponse response = new CommandResponse(200);
for (DirectoryHandle nukeDir : dirsToUnNuke) {
try {
env.add("name", nukeDir.getName());
env.add("path", nukeDir.getPath());
env.add("owner", nukeDir.getUsername());
env.add("group", nukeDir.getGroup());
env.add("size", Bytes.formatBytes(nukeDir.getSize()));
response.addComment(
session.jprintf(_bundle, _keyPrefix + "unnuke.search.item", env, user));
} catch (FileNotFoundException e) {
logger.warn(
"Dir deleted after index search?, skip and continue: " + nukeDir.getPath());
}
}
response.addComment(
session.jprintf(_bundle, _keyPrefix + "unnuke.search.end", env, user));
// Return matching dirs and let user decide what to unnuke
return response;
}
} else {
toDir = VirtualFileSystem.separator;
}
} else {
toDir = currentDir.getPath() + VirtualFileSystem.separator;
}
} else {
// Full path to Nuked dir provided, append [NUKED]- if needed.
toDir = VirtualFileSystem.fixPath(toName);
toName = toDir.substring(toDir.lastIndexOf(VirtualFileSystem.separator) + 1);
toDir = toDir.substring(0, toDir.lastIndexOf(VirtualFileSystem.separator) + 1);
if (toName.startsWith("[NUKED]-")) {
nukeName = toName;
toName = toName.substring(8);
} else {
nukeName = "[NUKED]-" + toName;
}
}
String reason;
if (st.hasMoreTokens()) {
reason = st.nextToken("").trim();
} else {
reason = "";
}
DirectoryHandle nukeDir;
try {
nukeDir = currentDir.getDirectory(toDir + nukeName, user);
} catch (FileNotFoundException e) {
// Maybe dir was deleted/wiped, lets remove it from nukelog.
try {
NukeBeans.getNukeBeans().remove(toDir + toName);
} catch (ObjectNotFoundException ex) {
return new CommandResponse(
500, toDir + nukeName + " doesnt exist and no nukelog for this path was found.");
}
return new CommandResponse(
200, toDir + nukeName + " doesnt exist, removed nuke from nukelog.");
} catch (ObjectNotValidException e) {
return new CommandResponse(550, toDir + nukeName + " is not a directory");
}
NukeData nd;
try {
nd = NukeUtils.unnuke(nukeDir, reason);
} catch (NukeException e) {
return new CommandResponse(500, "Unnuke failed: " + e.getMessage());
}
CommandResponse response = new CommandResponse(200, "Unnuke succeeded");
NukeEvent nukeEvent = new NukeEvent(session.getUserNull(request.getUser()), "UNNUKE", nd);
GlobalContext.getEventService().publishAsync(nukeEvent);
ReplacerEnvironment env = new ReplacerEnvironment();
String section = GlobalContext.getGlobalContext().getSectionManager().lookup(nukeDir).getName();
env.add("section", section);
env.add("dir", nukeDir.getName());
env.add("path", nukeDir.getPath());
env.add("relpath", nukeDir.getPath().replaceAll("/" + section + "/", ""));
env.add("multiplier", nd.getMultiplier());
env.add("nukedamount", Bytes.formatBytes(nd.getAmount()));
env.add("reason", reason);
env.add("size", Bytes.formatBytes(nd.getSize()));
if (session instanceof BaseFtpConnection) {
response.addComment(session.jprintf(_bundle, _keyPrefix + "unnuke", env, user));
for (NukedUser nukeeObj : NukeBeans.getNukeeList(nd)) {
ReplacerEnvironment nukeeenv = new ReplacerEnvironment();
User nukee;
try {
nukee =
GlobalContext.getGlobalContext()
.getUserManager()
.getUserByName(nukeeObj.getUsername());
} catch (NoSuchUserException e1) {
// Unable to get user, does not exist.. skip announce for this user
continue;
} catch (UserFileException e1) {
// Error in user file.. skip announce for this user
continue;
}
long debt =
NukeUtils.calculateNukedAmount(
nukeeObj.getAmount(),
nukee.getKeyedMap().getObjectFloat(UserManagement.RATIO),
nd.getMultiplier());
nukeeenv.add("nukedamount", Bytes.formatBytes(debt));
response.addComment(
session.jprintf(_bundle, _keyPrefix + "unnuke.nukees", nukeeenv, nukee));
}
}
return response;
}