package com.aragost.javahg.internals;
   
   import java.io.File;
   import java.io.IOException;
   import java.util.List;
   
   import com.google.common.collect.Lists;
   
   /**
   * Helper class to implement Add and Removed Command.
   */
  public class AddRemoveCommandHelper {
  
      private AbstractCommand command;
  
      private String prefix;
  
      public AddRemoveCommandHelper(AbstractCommand command, String prefix) {
          super();
          this.command = command;
          this.prefix = prefix;
      }
  
      public List<String> execute(String... files) {
          List<String> result = Lists.newArrayList();
          for (String line : command.launchIterator(files)) {
              if (line.startsWith(prefix)) {
                  String file = new String(line.substring(prefix.length()));
                  result.add(file);
              } else {
                  command.cleanUp();
                  throw new UnexpectedCommandOutputException(this.command, line);
              }
          }
          return result;
      }
  
      public List<File> execute(File... files) {
          return stringList2fileList(execute(fileArray2StringArray(files)));
      }
  
      public List<File> execute() {
          return execute(new File[0]);
      }
  
      /**
       * Helper method to convert Files to Strings
       * 
       * @param files
       * @return
       */
      private final String[] fileArray2StringArray(File[] files) {
          return Utils.fileArray2StringArray(files);
      }
  
      /**
       * Helper method to convert Strings to Files
       * 
       * @param strings
       * @return
       * @throws IOException
       */
      private final List<File> stringList2fileList(List<String> strings) {
          String[] strArray = strings.toArray(new String[strings.size()]);
          File[] fileArray = Utils.stringArray2FileArray(this.command.getRepository().getDirectory(), strArray);
          return Lists.newArrayList(fileArray);
      }
  
  }