/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
    * with the License.  You may obtain a copy of the License at
    *
   *   http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  package org.apache.maven.scm.provider.hg;
  
  import java.io.File;
  import java.io.IOException;
  import java.util.ArrayList;
  import java.util.List;
  
  import org.apache.maven.scm.ScmFileSet;
  import org.apache.maven.scm.ScmResult;
  import org.apache.maven.scm.ScmTestCase;
  import org.apache.maven.scm.provider.hg.command.HgCommandConstants;
  import org.codehaus.plexus.util.FileUtils;
  
  /**
   * Common code used in all tests.
   *
   * @author <a href="mailto:thurner.rupert@ymono.net">thurner rupert</a>
   */
  public class HgRepoUtils extends ScmTestCase {
      /** 'hg' command line */
      public static final String HG_COMMAND_LINE = "hg";
  
      public static final String[] filesInTestBranch =
              new String[] {"pom.xml", "readme.txt", "src/main/java/Application.java", "src/test/java/Test.java"};
  
      public static final String TCK_FILE_CONSTANT = "/";
  
      public static final String BRANCH_NAME = "target" + File.separator + "test-branch";
  
      public static final File WORKING_DIR = new File(getBasedir(), BRANCH_NAME);
  
      public static final String COMMIT_MESSAGE = "Add files to test branch";
  
      public static String getScmUrl() throws Exception {
          return "scm:hg:" + WORKING_DIR.getAbsolutePath();
      }
  
      public static void initRepo() throws Exception {
          // Prepare tmp directory
          if (WORKING_DIR.exists()) {
              FileUtils.deleteDirectory(WORKING_DIR);
  
              if (WORKING_DIR.exists()) {
                  throw new IOException(WORKING_DIR.getAbsolutePath() + " wasn't deleted.");
              }
          }
  
          boolean workingDirReady = WORKING_DIR.mkdirs();
  
          if (!workingDirReady) {
              throw new IOException("Could not initiate test branch at: " + WORKING_DIR);
          }
  
          // Init repository
          String[] init_cmd = new String[] {HgCommandConstants.INIT_CMD};
          HgUtils.execute(WORKING_DIR, init_cmd);
  
          // Create and add files to repository
          List<File> files = new ArrayList<>();
          for (int i = 0; i < filesInTestBranch.length; i++) {
              File file = new File(WORKING_DIR.getAbsolutePath(), filesInTestBranch[i]);
              if (file.getParentFile() != null && !file.getParentFile().exists()) {
                  boolean success = file.getParentFile().mkdirs();
                  if (!success) {
                      throw new IOException("Could not create directories in branch for: " + file);
                  }
              }
              file.createNewFile();
  
              FileUtils.fileWrite(file.getAbsolutePath(), TCK_FILE_CONSTANT + filesInTestBranch[i]);
  
              files.add(file);
          }
  
          // Add to repository
          String[] add_cmd = new String[] {HgCommandConstants.ADD_CMD};
          ScmFileSet filesToAdd = new ScmFileSet(new File(""), files);
          add_cmd = HgUtils.expandCommandLine(add_cmd, filesToAdd);
          ScmResult result = HgUtils.execute(WORKING_DIR, add_cmd);
          if (!result.isSuccess()) {
              String message =
                     "Provider message: " + result.getProviderMessage() + "\n" + "Output: " + result.getCommandOutput();
             throw new Exception(message);
         }
 
         // Commit the initial repository
         String[] commit_cmd =
                 new String[] {HgCommandConstants.COMMIT_CMD, HgCommandConstants.MESSAGE_OPTION, COMMIT_MESSAGE};
         result = HgUtils.execute(WORKING_DIR, commit_cmd);
         if (!result.isSuccess()) {
             String message =
                     "Provider message: " + result.getProviderMessage() + "\n" + "Output: " + result.getCommandOutput();
             throw new Exception(message);
         }
     }
 }