Example #1
0
  @Test
  public void masking() throws IOException, InterruptedException {
    PipedOutputStream outputStream = new PipedOutputStream();
    PipedInputStream pipedInputStream = new PipedInputStream(outputStream);

    KeyOperation deletePrevChar = new KeyOperation(Key.CTRL_H, Operation.DELETE_PREV_CHAR);

    Console console = getTestConsole(pipedInputStream);
    console.setPrompt(new Prompt("", new Character('\u0000')));
    console.setConsoleCallback(
        new ConsoleCallback() {
          @Override
          public int readConsoleOutput(ConsoleOperation output) {
            assertEquals("mypasswor", output.getBuffer());
            return 0;
          }
        });
    console.start();
    outputStream.write(("mypassword").getBytes());
    outputStream.write(deletePrevChar.getFirstValue());
    outputStream.write(("\n").getBytes());

    Thread.sleep(100);

    console.stop();
  }
Example #2
0
  @Test
  public void paste() throws IOException, InterruptedException {
    final PipedOutputStream outputStream = new PipedOutputStream();
    PipedInputStream pipedInputStream = new PipedInputStream(outputStream);

    String pasteLine1 =
        "connect" + Config.getLineSeparator() + "admin" + Config.getLineSeparator() + "admin!";
    String pasteLine2 = "234" + Config.getLineSeparator() + "exit" + Config.getLineSeparator();

    final Console console = getTestConsole(pipedInputStream);
    console.setConsoleCallback(
        new AeshConsoleCallback() {
          boolean password = false;

          @Override
          public int execute(ConsoleOperation output) {
            if (output.getBuffer().equals("admin")) {
              console.setPrompt(new Prompt("", new Character('\u0000')));
              password = true;
              return 0;
            }
            if (password) {
              assertEquals("admin!234", output.getBuffer());
              password = false;
            }
            return 0;
          }
        });
    console.start();
    outputStream.write(pasteLine1.getBytes());
    outputStream.write(pasteLine2.getBytes());

    Thread.sleep(500);

    console.stop();
  }
Example #3
0
  public static void main(String[] args) throws IOException {

    // Settings.getInstance().setAnsiConsole(false);
    SettingsBuilder builder = new SettingsBuilder();
    builder.readInputrc(false).logging(true);
    builder.logfile(
        System.getProperty("user.dir") + System.getProperty("file.separator") + "aesh_example.log");
    // Settings.getInstance().setAliasEnabled(true);
    // Settings.getInstance().setAliasFile(new
    // File(System.getProperty("user.dir")+Config.getPathSeparator()+"aesh_aliases.txt"));
    // Settings.getInstance().setPersistAlias(true);
    // Settings.getInstance().enableOperatorParser(false);
    // Settings.getInstance().setHistoryDisabled(true);
    // Settings.getInstance().setHistoryPersistent(false);
    List<TerminalCharacter> chars = new ArrayList<TerminalCharacter>();
    chars.add(new TerminalCharacter('[', new TerminalColor(Color.BLUE, Color.DEFAULT)));
    chars.add(
        new TerminalCharacter(
            't', new TerminalColor(Color.RED, Color.DEFAULT), CharacterType.ITALIC));
    chars.add(
        new TerminalCharacter(
            'e', new TerminalColor(Color.RED, Color.DEFAULT), CharacterType.INVERT));
    chars.add(
        new TerminalCharacter(
            's', new TerminalColor(Color.RED, Color.DEFAULT), CharacterType.CROSSED_OUT));
    chars.add(
        new TerminalCharacter(
            't', new TerminalColor(Color.RED, Color.DEFAULT), CharacterType.BOLD));
    chars.add(
        new TerminalCharacter(
            ']', new TerminalColor(Color.BLUE, Color.DEFAULT), CharacterType.FAINT));
    chars.add(
        new TerminalCharacter(
            '$', new TerminalColor(Color.GREEN, Color.DEFAULT), CharacterType.UNDERLINE));
    chars.add(new TerminalCharacter(' ', new TerminalColor(Color.DEFAULT, Color.DEFAULT)));

    final Prompt prompt = new Prompt(chars);
    // String prompt = ANSI.redText()+"[test@foo]"+ANSI.reset()+"$ ";

    // a simple interruptHook
    builder.interruptHook(
        new InterruptHook() {
          @Override
          public void handleInterrupt(Console console) {
            console.getShell().out().println("^C");
            console.clearBufferAndDisplayPrompt();
          }
        });

    final Console exampleConsole = new Console(builder.create());

    Completion completer =
        new Completion() {
          @Override
          public void complete(CompleteOperation co) {
            // very simple completor
            List<String> commands = new ArrayList<String>();
            if (co.getBuffer().equals("fo") || co.getBuffer().equals("foo")) {
              commands.add("foo");
              commands.add("foobaa");
              commands.add("foobar");
              commands.add("foobaxxxxxx");
              commands.add("foobbx");
              commands.add("foobcx");
              commands.add("foobdx");
            }
            if (co.getBuffer().equals("p")) {
              commands.add("profile=foo");
              co.setOffset(0);
            }
            /*
            if(co.getBuffer().equals("p")) {
                commands.add("profile=bar");
                co.setOffset(0);
            }
            */
            if (co.getBuffer().equals("profile=")) commands.add("profile=foo");
            if (co.getBuffer().equals("profile=")) commands.add("profile=bar");
            if (co.getBuffer().equals("--")) {
              commands.add("--help-");
            }
            if ("--help-me".startsWith(co.getBuffer())) {
              commands.add("--help-me");
            }
            if (co.getBuffer().equals("fooba")) {
              commands.add("foobaa");
              commands.add("foobar");
              commands.add("foobaxxxxxx");
            }
            if (co.getBuffer().equals("foobar")) {
              commands.add("foobar");
            }
            if (co.getBuffer().equals("bar")) {
              commands.add("bar/");
            }
            if (co.getBuffer().equals("h")) {
              commands.add("help.history");
              commands.add("help");
              co.setOffset(0);
            }
            if (co.getBuffer().equals("help")) {
              commands.add("help.history");
              commands.add("help");
            }
            if (co.getBuffer().equals("help.")) {
              commands.add("help.history");
            }
            if (co.getBuffer().equals("deploy")) {
              commands.add("deploy /home/blabla/foo/bar/alkdfe/en/to/tre");
            }
            if (co.getBuffer().equals("testing")) {
              commands.add("testing YAY");
            }
            if (co.getBuffer().equals("val") || co.getBuffer().equals("value ")) {
              commands.add("value 1");
              commands.add("value 2");
              commands.add("value 10");
              commands.add("value 20");
            }
            if (co.getBuffer().equals("valu")) commands.add("value 10");

            co.setCompletionCandidates(commands);
          }
        };

    exampleConsole.addCompletion(completer);

    final ConsoleCallback consoleCallback =
        new ConsoleCallback() {
          @Override
          public int readConsoleOutput(ConsoleOperation output) {
            try {
              // To change body of implemented methods use File | Settings | File Templates.
              exampleConsole.getShell().out().print("======>\"" + output.getBuffer() + "\"\n");
              if (masking) {
                exampleConsole
                    .getShell()
                    .out()
                    .print("got password: "******", stopping masking");
                masking = false;
                exampleConsole.setPrompt(prompt);
              } else if (output.getBuffer().equalsIgnoreCase("quit")
                  || output.getBuffer().equalsIgnoreCase("exit")
                  || output.getBuffer().equalsIgnoreCase("reset")) {
                exampleConsole.stop();
              } else if (output.getBuffer().equalsIgnoreCase("password")) {
                masking = true;
                exampleConsole.setPrompt(new Prompt("password: "******"blah")) {
                exampleConsole.getShell().err().print("blah. command not found.\n");
                exampleConsole.getShell().out().print("BAH" + Config.getLineSeparator());
              } else if (output.getBuffer().equals("clear")) exampleConsole.clear();
              else if (output.getBuffer().startsWith("man")) {
                // exampleConsole.attachProcess(test);
                ConsoleCommand test = new ExampleConsoleCommand(exampleConsole, output);
                exampleConsole.attachProcess(test);
              } else if (output.getBuffer().startsWith("login")) {
                exampleConsole.setConsoleCallback(passwordCallback);
                exampleConsole.setPrompt(new Prompt("Username: "******"Exception: " + ioe.getMessage());
              return -1;
            }
          }
        };
    exampleConsole.setConsoleCallback(consoleCallback);
    exampleConsole.start();
    exampleConsole.setPrompt(prompt);

    passwordCallback =
        new ConsoleCallback() {
          private boolean hasUsername = false;

          @Override
          public int readConsoleOutput(ConsoleOperation output) {
            if (hasUsername) {
              password = output.getBuffer();
              hasPassword = true;
              exampleConsole
                  .getShell()
                  .out()
                  .print(
                      "Username: "******", password: "******"Password: "******"[FOO]ยป ",
                  new TerminalColor(Color.RED, Color.DEFAULT),
                  new TerminalTextStyle(CharacterType.BOLD))));
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
  }