package com.intellij.flex.compiler;

import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

/**
 * Splits input String to tokens being aware of quoted tokens,
 * usually used for splitting command line to separate arguments that may contain space symbols.
 * Escaped symbols are not handled so there's no way to get token that itself contains quotation mark.
 */
public class CommandLineArgumentsTokenizer extends StringTokenizer {

  private static String DEFAULT_DELIMITERS = " \t\n\r\f";
  // keep source level 1.4
  private List myTokens = new ArrayList();
  private int myCurrentToken = 0;

  public CommandLineArgumentsTokenizer(String str) {
    super(str, DEFAULT_DELIMITERS, true);
    parseTokens();
  }

  public CommandLineArgumentsTokenizer(String str, String delim) {
    super(str, delim, true);
    parseTokens();
  }

  public boolean hasMoreTokens() {
    return myCurrentToken < myTokens.size();
  }

  public String nextToken() {
    return (String)myTokens.get(myCurrentToken++);
  }

  public int countTokens() {
    return myTokens.size() - myCurrentToken;
  }


  public String nextToken(String delim) {
    throw new RuntimeException("Not implemented");
  }

  private void parseTokens() {
    String token;
    while ((token = nextTokenInternal()) != null) {
      myTokens.add(token);
    }
  }

  private String nextTokenInternal() {
    String nextToken;
    do {
      if (super.hasMoreTokens()) {
        nextToken = super.nextToken();
      }
      else {
        nextToken = null;
      }
    }
    while (nextToken != null && nextToken.length() == 1 && DEFAULT_DELIMITERS.indexOf(nextToken.charAt(0)) >= 0);

    if (nextToken == null) {
      return null;
    }

    int i;
    int quotationMarks = 0;
    final StringBuffer buffer = new StringBuffer();

    do {
      while ((i = nextToken.indexOf('"')) >= 0) {
        quotationMarks++;
        buffer.append(nextToken.substring(0, i));
        nextToken = nextToken.substring(i + 1);
      }
      buffer.append(nextToken);
      if (quotationMarks % 2 == 1 && super.hasMoreTokens()) {
        nextToken = super.nextToken();
      }
      else {
        nextToken = null;
      }
    }
    while (nextToken != null);

    return buffer.toString();
  }
}