com.aoindustries.aoserv.client.AOSH Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of aoserv-client Show documentation
Show all versions of aoserv-client Show documentation
Java client for the AOServ Platform.
/*
* aoserv-client - Java client for the AOServ platform.
* Copyright (C) 2001-2013, 2014, 2015, 2016 AO Industries, Inc.
* [email protected]
* 7262 Bull Pen Cir
* Mobile, AL 36695
*
* This file is part of aoserv-client.
*
* aoserv-client is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* aoserv-client is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with aoserv-client. If not, see AOSH is a command interpreter and scripting language
* based on the Bourne shell. It may be used to control the
* AOServ Client utilities.
*
* @author AO Industries, Inc.
*/
final public class AOSH extends ShellInterpreter {
private static final Logger logger = Logger.getLogger(AOSH.class.getName());
private static final Reader nullInput=new CharArrayReader(new char[0]);
final private AOServConnector connector;
public AOSH(AOServConnector connector, Reader in, TerminalWriter out, TerminalWriter err) {
super(in, out, err);
this.connector=connector;
}
public AOSH(AOServConnector connector, Reader in, TerminalWriter out, TerminalWriter err, String ... args) {
super(in, out, err, args);
this.connector=connector;
}
static boolean checkMinParamCount(String function, String[] args, int minCount, PrintWriter err) {
int paramCount=args.length-1;
if(paramCountmaxCount) {
err.print("aosh: ");
err.print(function);
err.println(": too many parameters");
err.flush();
return false;
}
return true;
}
private void echo(String[] args) {
for(int c=1;c1) out.print(' ');
out.print(args[c]);
}
out.println();
out.flush();
}
static String executeCommand(AOServConnector connector, String[] args) throws IOException, SQLException {
StringWriter buff = new StringWriter();
TerminalWriter out=new TerminalWriter(buff);
out.setEnabled(false);
AOSH sh=new AOSH(connector, nullInput, out, out);
sh.handleCommand(args);
out.flush();
return buff.toString();
}
@Override
protected String getName() {
return "aosh";
}
@Override
protected String getPrompt() throws SQLException, IOException {
return '['+connector.getThisBusinessAdministrator().toString()+'@'+connector.getHostname()+"]$ ";
}
/** Avoid repeated array copies. */
private static final int numTables = SchemaTable.TableID.values().length;
/**
* Processes one command and returns.
*
* @param args the command and argments to process
*
* @return true if more commands should be processed
*/
@Override
protected boolean handleCommand(String[] args) throws IOException, SQLException {
int argCount=args.length;
if(argCount>0) {
String command=args[0];
if(AOSHCommand.EXIT.equalsIgnoreCase(command)) {
if(argCount!=1) {
err.println("aosh: "+AOSHCommand.EXIT+": too many parameters");
err.flush();
} else return false;
} else {
if(AOSHCommand.CLEAR.equalsIgnoreCase(command)) clear(args);
else if(AOSHCommand.ECHO.equalsIgnoreCase(command)) echo(args);
else if(AOSHCommand.INVALIDATE.equalsIgnoreCase(command)) invalidate(args);
else if(AOSHCommand.JOBS.equalsIgnoreCase(command)) jobs(args);
else if(AOSHCommand.PING.equalsIgnoreCase(command)) ping(args);
else if(AOSHCommand.REPEAT.equalsIgnoreCase(command)) repeat(args);
else if(AOSHCommand.SLEEP.equalsIgnoreCase(command)) sleep(args);
else if(AOSHCommand.SU.equalsIgnoreCase(command)) su(args);
else if(AOSHCommand.TIME.equalsIgnoreCase(command)) time(args);
else if(AOSHCommand.WHOAMI.equalsIgnoreCase(command)) whoami(args);
else {
boolean done=false;
// Use the aosh_commands table for faster lookups
String lowerCommand=command.toLowerCase();
AOSHCommand aoshCommand=connector.getAoshCommands().get(lowerCommand);
if(aoshCommand!=null) {
AOServTable table=aoshCommand.getSchemaTable(connector).getAOServTable(connector);
done=table.handleCommand(args, in, out, err, isInteractive());
if(!done) throw new RuntimeException("AOSHCommand found, but command not processed. command='"+lowerCommand+"', table='"+table.getTableName()+'\'');
}
/*
for(int c=0;c=0) {
connector.getSimpleAOClient().invalidate(tableID, args.length>2?args[2]:null);
} else {
err.print("aosh: "+AOSHCommand.INVALIDATE+": unable to find table: ");
err.println(tableName);
err.flush();
}
}
}
public static void main(String[] args) {
TerminalWriter out=new TerminalWriter(new BufferedWriter(new OutputStreamWriter(System.out)));
TerminalWriter err=System.out==System.err ? out : new TerminalWriter(new BufferedWriter(new OutputStreamWriter(System.err)));
try {
String username=getConfigUsername(System.in, err);
String password=getConfigPassword(System.in, err);
AOServConnector connector=AOServConnector.getConnector(username, password, logger);
AOSH aosh=new AOSH(connector, new BufferedReader(new InputStreamReader(System.in)), out, err, args);
aosh.run();
if(aosh.isInteractive()) {
out.println();
out.flush();
}
} catch(IOException exc) {
err.println("aosh: unable to connect: "+exc.getMessage());
err.flush();
}
}
public static String getConfigUsername(InputStream in, TerminalWriter err) throws IOException {
String username=AOServClientConfiguration.getUsername();
if(username==null || username.length()==0) {
// Prompt for the username
err.print("Username: ");
err.flush();
username=readLine(in);
}
return username;
}
public static String getConfigPassword(InputStream in, TerminalWriter err) throws IOException {
String password=AOServClientConfiguration.getPassword();
if(password==null || password.length()==0) {
// Prompt for the password
err.print("Password: ");
err.flush();
password=readLine(in);
err.flush();
}
return password;
}
@Override
protected AOSH newShellInterpreter(Reader in, TerminalWriter out, TerminalWriter err, String[] args) {
return new AOSH(connector, in, out, err, args);
}
static AccountingCode parseAccountingCode(String S, String field) {
try {
return AccountingCode.valueOf(S);
} catch(ValidationException err) {
throw new IllegalArgumentException("invalid argument for accounting "+field+": "+S, err);
}
}
static boolean parseBoolean(String S, String field) {
if(
S.equalsIgnoreCase("true")
|| S.equalsIgnoreCase("t")
|| S.equalsIgnoreCase("yes")
|| S.equalsIgnoreCase("y")
|| S.equalsIgnoreCase("vang")
|| S.equalsIgnoreCase("da")
|| S.equalsIgnoreCase("si")
|| S.equalsIgnoreCase("oui")
|| S.equalsIgnoreCase("ja")
|| S.equalsIgnoreCase("nam")
) return true;
else if(
S.equalsIgnoreCase("false")
|| S.equalsIgnoreCase("f")
|| S.equalsIgnoreCase("no")
|| S.equalsIgnoreCase("n")
|| S.equalsIgnoreCase("khong")
|| S.equalsIgnoreCase("nyet")
|| S.equalsIgnoreCase("non")
|| S.equalsIgnoreCase("nien")
|| S.equalsIgnoreCase("la")
) return false;
else throw new IllegalArgumentException("invalid argument for boolean "+field+": "+S);
}
static Date parseDate(String S, String field) {
try {
return SQLUtility.getDate(S);
} catch(NumberFormatException err) {
throw new IllegalArgumentException("invalid argument for date "+field+": "+S, err);
}
}
static int parseInt(String S, String field) {
try {
return Integer.parseInt(S);
} catch(NumberFormatException err) {
throw new IllegalArgumentException("invalid argument for int "+field+": "+S, err);
}
}
static float parseFloat(String S, String field) {
try {
return Float.parseFloat(S);
} catch(NumberFormatException err) {
throw new IllegalArgumentException("invalid argument for float "+field+": "+S, err);
}
}
static long parseLong(String S, String field) {
try {
return Long.parseLong(S);
} catch(NumberFormatException err) {
throw new IllegalArgumentException("invalid argument for long "+field+": "+S, err);
}
}
static int parseMillis(String S, String field) {
try {
return SQLUtility.getMillis(S);
} catch(NumberFormatException err) {
throw new IllegalArgumentException("invalid argument for decimal "+field+": "+S, err);
}
}
static int parseOctalInt(String S, String field) {
try {
return Integer.parseInt(S, 8);
} catch(NumberFormatException err) {
throw new IllegalArgumentException("invalid argument for octal int "+field+": "+S, err);
}
}
static long parseOctalLong(String S, String field) {
try {
return Long.parseLong(S, 8);
} catch(NumberFormatException err) {
throw new IllegalArgumentException("invalid argument for octal long "+field+": "+S, err);
}
}
static int parsePennies(String S, String field) {
try {
return SQLUtility.getPennies(S);
} catch(NumberFormatException err) {
throw new IllegalArgumentException("invalid argument for decimal "+field+": "+S, err);
}
}
static short parseShort(String S, String field) {
try {
return Short.parseShort(S);
} catch(NumberFormatException err) {
throw new IllegalArgumentException("invalid argument for short "+field+": "+S, err);
}
}
static DomainName parseDomainName(String S, String field) {
try {
return DomainName.valueOf(S);
} catch(ValidationException err) {
throw new IllegalArgumentException("invalid argument for domain_name "+field+": "+S, err);
}
}
static Gecos parseGecos(String S, String field) {
try {
return Gecos.valueOf(S);
} catch(ValidationException err) {
throw new IllegalArgumentException("invalid argument for gecos "+field+": "+S, err);
}
}
static InetAddress parseInetAddress(String S, String field) {
try {
return InetAddress.valueOf(S);
} catch(ValidationException err) {
throw new IllegalArgumentException("invalid argument for ip_address "+field+": "+S, err);
}
}
private void ping(String[] args) throws IOException, SQLException {
if(checkParamCount(AOSHCommand.PING, args, 0, err)) {
out.print(connector.getSimpleAOClient().ping());
out.println(" ms");
out.flush();
}
}
public static String readLine(InputStream in) throws IOException {
StringBuilder SB=new StringBuilder();
readLine(in, SB);
return SB.toString();
}
public static void readLine(InputStream in, StringBuilder SB) throws IOException {
SB.setLength(0);
int ch;
while((ch=in.read())!=-1 && ch!='\n') if(ch!='\r') SB.append((char)ch);
}
private void repeat(String[] args) throws IOException, SQLException {
int argCount=args.length;
if(argCount>2) {
try {
int count=Integer.parseInt(args[1]);
if(count>=0) {
String[] newArgs=new String[argCount-2];
System.arraycopy(args, 2, newArgs, 0, argCount-2);
for(int c=0;c1) {
try {
for(int c=1;c=2) {
String[] newArgs=new String[argCount+(isInteractive()?0:-1)];
int pos=0;
if(isInteractive()) newArgs[pos++]="-i";
newArgs[pos++]="--";
System.arraycopy(args, 2, newArgs, pos, argCount-2);
new AOSH(connector.switchUsers(args[1]), in, out, err, newArgs).run();
} else {
err.println("aosh: "+AOSHCommand.SU+": not enough parameters");
err.flush();
}
}
private void time(String[] args) throws IOException, SQLException {
int argCount=args.length;
if(argCount>1) {
String[] newArgs=new String[argCount-1];
System.arraycopy(args, 1, newArgs, 0, argCount-1);
long startTime=System.currentTimeMillis();
try {
handleCommand(newArgs);
} finally {
long timeSpan=System.currentTimeMillis()-startTime;
int mins=(int)(timeSpan/60000);
int secs=(int)(timeSpan%60000);
out.println();
out.print("real ");
out.print(mins);
out.print('m');
out.print(SQLUtility.getMilliDecimal(secs));
out.println('s');
out.flush();
}
} else {
err.println("aosh: "+AOSHCommand.TIME+": not enough parameters");
err.flush();
}
}
private void whoami(String[] args) throws SQLException, IOException {
if(args.length==1) {
out.println(connector.getThisBusinessAdministrator().getUsername().getUsername());
out.flush();
} else {
err.println("aosh: "+AOSHCommand.WHOAMI+": too many parameters");
err.flush();
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy