mirror of
https://github.com/Alfresco/alfresco-community-repo.git
synced 2025-07-24 17:32:48 +00:00
Moved a bunch of things to more logical locations.
git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@4497 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
This commit is contained in:
201
source/java/org/alfresco/repo/clt/CltBase.java
Normal file
201
source/java/org/alfresco/repo/clt/CltBase.java
Normal file
@@ -0,0 +1,201 @@
|
||||
/**
|
||||
*
|
||||
*/
|
||||
package org.alfresco.repo.clt;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.io.OutputStream;
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import org.alfresco.repo.remote.ClientTicketHolder;
|
||||
import org.alfresco.service.cmr.avmsync.AVMSyncService;
|
||||
import org.alfresco.service.cmr.remote.AVMRemote;
|
||||
import org.alfresco.service.cmr.security.AuthenticationService;
|
||||
import org.springframework.context.ConfigurableApplicationContext;
|
||||
import org.springframework.context.support.ClassPathXmlApplicationContext;
|
||||
|
||||
/**
|
||||
* This is the base class for AVM clts.
|
||||
* @author britt
|
||||
*/
|
||||
public abstract class CltBase
|
||||
{
|
||||
/**
|
||||
* The instance of the remote interface.
|
||||
*/
|
||||
protected AVMRemote fAVMRemote;
|
||||
|
||||
/**
|
||||
* The instance of the remote sync service interface.
|
||||
*/
|
||||
protected AVMSyncService fAVMSyncService;
|
||||
|
||||
/**
|
||||
* The ApplicationContext.
|
||||
*/
|
||||
protected ConfigurableApplicationContext fContext;
|
||||
|
||||
/**
|
||||
* The Authentication Service.
|
||||
*/
|
||||
protected AuthenticationService fAuthenticationService;
|
||||
|
||||
/**
|
||||
* Construct a new one. This takes care of instantiating
|
||||
* the application context and grabs references to the
|
||||
* services.
|
||||
* @param args The program arguments.
|
||||
*/
|
||||
protected CltBase()
|
||||
{
|
||||
fContext = new ClassPathXmlApplicationContext("alfresco/avm-clt-context.xml");
|
||||
fAVMRemote = (AVMRemote)fContext.getBean("avmRemote");
|
||||
fAVMSyncService = (AVMSyncService)fContext.getBean("avmSyncService");
|
||||
fAuthenticationService = (AuthenticationService)fContext.getBean("authenticationService");
|
||||
fAuthenticationService.authenticate("admin", "admin".toCharArray());
|
||||
String ticket = fAuthenticationService.getCurrentTicket();
|
||||
ClientTicketHolder.SetTicket(ticket);
|
||||
}
|
||||
|
||||
/**
|
||||
* All clts go through this call. This parses the arguments, exits if
|
||||
* there are any errors and then passes the broken flags and arguments
|
||||
* to the run method of the derived clt.
|
||||
* @param args The raw command line arguments.
|
||||
* @param flagDefs The definition of what flags to accept and their
|
||||
* arities.
|
||||
* @param minArgs The minimum number of actual arguments expected.
|
||||
* @param usageMessage The message that should be printed if there is a
|
||||
* syntax error.
|
||||
*/
|
||||
public void exec(String [] args,
|
||||
Object [] flagDefs,
|
||||
int minArgs,
|
||||
String usageMessage)
|
||||
{
|
||||
Map<String, Integer> flagArgs = new HashMap<String, Integer>();
|
||||
Map<String, List<String>> flagValues = new HashMap<String, List<String>>();
|
||||
List<String> actualArgs = new ArrayList<String>();
|
||||
// Convert the flag definitions into a convenient form.
|
||||
for (int i = 0; i < flagDefs.length / 2; i++)
|
||||
{
|
||||
flagArgs.put((String)flagDefs[i * 2], (Integer)flagDefs[i * 2 + 1]);
|
||||
}
|
||||
// Walk through the raw command line arguments.
|
||||
int pos = 0;
|
||||
while (pos < args.length)
|
||||
{
|
||||
if (args[pos].equals("-h"))
|
||||
{
|
||||
usage(usageMessage);
|
||||
}
|
||||
// If the argument is one of the accepted flags then it's
|
||||
// a flag.
|
||||
if (flagArgs.containsKey(args[pos]))
|
||||
{
|
||||
String flag = args[pos];
|
||||
pos++;
|
||||
int count = flagArgs.get(flag);
|
||||
// Check for too few arguments
|
||||
if (args.length - pos < count)
|
||||
{
|
||||
usage(usageMessage);
|
||||
}
|
||||
// Stuff the parsed flag away.
|
||||
List<String> flArgs = new ArrayList<String>();
|
||||
for (int i = 0; i < count; i++)
|
||||
{
|
||||
flArgs.add(args[pos + i]);
|
||||
}
|
||||
flagValues.put(flag, flArgs);
|
||||
pos += count;
|
||||
continue;
|
||||
}
|
||||
// Otherwise its just a plain old arg.
|
||||
actualArgs.add(args[pos]);
|
||||
pos++;
|
||||
}
|
||||
// Check for too few arguments.
|
||||
if (actualArgs.size() < minArgs)
|
||||
{
|
||||
usage(usageMessage);
|
||||
}
|
||||
// Do the work.
|
||||
run(flagValues, actualArgs);
|
||||
// Cleanup.
|
||||
fContext.close();
|
||||
}
|
||||
|
||||
/**
|
||||
* Handle syntax error by exiting.
|
||||
* @param usageMessage The message to print.
|
||||
*/
|
||||
protected void usage(String usageMessage)
|
||||
{
|
||||
System.err.println(usageMessage);
|
||||
fContext.close();
|
||||
System.exit(1);
|
||||
}
|
||||
|
||||
/**
|
||||
* Utility to split an AVM path into a parent path and a
|
||||
* base name.
|
||||
* @param path The path to split.
|
||||
* @return An array of 1 or 2 Strings representing the parent path
|
||||
* and the base name, or just the path if the path given is a root path.
|
||||
*/
|
||||
protected String[] splitPath(String path)
|
||||
{
|
||||
if (path.endsWith(":/"))
|
||||
{
|
||||
String [] ret = { path };
|
||||
return ret;
|
||||
}
|
||||
int lastSlash = path.lastIndexOf("/");
|
||||
if (lastSlash == -1)
|
||||
{
|
||||
System.err.println("Malformed path: " + path);
|
||||
fContext.close();
|
||||
System.exit(1);
|
||||
}
|
||||
String name = path.substring(lastSlash + 1);
|
||||
String parent = path.substring(0, lastSlash);
|
||||
if (parent.endsWith(":"))
|
||||
{
|
||||
parent = parent + "/";
|
||||
}
|
||||
while (parent.endsWith("/") && !parent.endsWith(":/"))
|
||||
{
|
||||
parent = parent.substring(0, parent.length() - 1);
|
||||
}
|
||||
String [] ret = { parent, name };
|
||||
return ret;
|
||||
}
|
||||
|
||||
protected void copyStream(InputStream in, OutputStream out)
|
||||
{
|
||||
try
|
||||
{
|
||||
byte [] buff = new byte[8192];
|
||||
int read = 0;
|
||||
while ((read = in.read(buff)) != -1)
|
||||
{
|
||||
out.write(buff, 0, read);
|
||||
}
|
||||
in.close();
|
||||
out.close();
|
||||
}
|
||||
catch (IOException e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
fContext.close();
|
||||
System.exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
protected abstract void run(Map<String, List<String>> flags, List<String> args);
|
||||
}
|
Reference in New Issue
Block a user