/*************************GO-LICENSE-START*********************************
 * Copyright 2014 ThoughtWorks, Inc.
 *
 * Licensed 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.
 *************************GO-LICENSE-END***********************************/

package com.thoughtworks.go.helper;

import com.thoughtworks.go.config.materials.svn.SvnMaterial;
import com.thoughtworks.go.config.materials.svn.SvnMaterialConfig;
import com.thoughtworks.go.domain.materials.*;
import com.thoughtworks.go.util.FileUtil;
import com.thoughtworks.go.util.TestFileUtil;
import com.thoughtworks.go.util.command.InMemoryStreamConsumer;
import com.thoughtworks.go.util.command.ProcessOutputStreamConsumer;
import org.junit.Assert;

import java.io.File;
import java.io.IOException;
import java.util.List;

import static com.thoughtworks.go.util.command.ProcessOutputStreamConsumer.inMemoryConsumer;
import static org.apache.commons.io.FileUtils.copyDirectory;

public class SvnTestRepo extends TestRepo {
    protected File tempRepo;

    private static final String REPO_TEST_DATA_FOLDER = "../common/test-resources/unit/data/svnrepo";

    public SvnTestRepo() throws IOException {
        this("testSvnRepo-" + System.currentTimeMillis());
    }

    public SvnTestRepo(String tempFolderName) throws IOException {
        tempRepo = TestFileUtil.createUniqueTempFolder(tempFolderName);
        tmpFolders.add(tempRepo);
        try {
            copyDirectory(new File(REPO_TEST_DATA_FOLDER), tempRepo);
        } catch (IOException e) {
            Assert.fail("Could not copy test repo [" + REPO_TEST_DATA_FOLDER + "] into [" + tempRepo + "] beacuse of " + e.getMessage());
        }
        new File(tempRepo, "/project1/db/transactions").mkdir();
    }

    public String urlFor(String project) {
        return repositoryUrl(project);
    }

    public String projectRepositoryUrl() {
        return repositoryUrl("project1/trunk");
    }

    public File projectRepositoryRoot() {
        return repositoryUrlAsFile("project1");
    }

    public File projectRepositoryUrlAsFile() {
        return repositoryUrlAsFile("project1/trunk");
    }

    public String end2endRepositoryUrl() {
        return repositoryUrl("end2end");
    }

    public String repositoryUrl(String project) {
        return FileUtil.toFileURI(new File(tempRepo, project));
    }

    public File repositoryUrlAsFile(String project) {
        return new File(tempRepo, project);
    }

    public SvnMaterial material() {
        return new SvnMaterial(projectRepositoryUrl(), null, null, false);
    }

    public SvnMaterialConfig materialConfig() {
        return new SvnMaterialConfig(projectRepositoryUrl(), null, null, false);
    }

    public SvnMaterial createMaterial(String repo, String folder) {
        SvnMaterial material = MaterialsMother.svnMaterial(urlFor(repo), folder);
        material.setAutoUpdate(true);
        return material;
    }

    private Revision getLatestRevision(SvnMaterial svnMaterial) {
        final File workingCopy = TestFileUtil.createTempFolder("bar");
        tmpFolders.add(workingCopy);
        return latestRevision(svnMaterial, workingCopy, new TestSubprocessExecutionContext());
    }

    public Modification checkInOneFile(String path) throws Exception {
        return checkInOneFile(path, "adding file [" + path + "]").get(0);
    }

    public List<Modification> checkInOneFile(String filename, String message) throws Exception {
        SvnMaterial svnMaterial = material();

        return checkInOneFile(svnMaterial, filename, message);

    }

    protected List<Modification> checkInOneFile(SvnMaterial svnMaterial, String filename, String message) throws IOException {
        final File workingCopy = TestFileUtil.createTempFolder("foo" + System.currentTimeMillis());
        tmpFolders.add(workingCopy);

        InMemoryStreamConsumer consumer = inMemoryConsumer();


        Revision latestRevision = getLatestRevision(svnMaterial);
        svnMaterial.updateTo(consumer, workingCopy, new RevisionContext(latestRevision), new TestSubprocessExecutionContext());


        File newFileToAdd = new File(workingCopy, filename);
        File directoryToAddTo = newFileToAdd.getParentFile();
        boolean addedToExistingDir = directoryToAddTo.exists();
        directoryToAddTo.mkdirs();

        FileUtil.writeContentToFile("", newFileToAdd);

        svnMaterial.add(consumer, addedToExistingDir ? newFileToAdd : directoryToAddTo);
        svnMaterial.commit(consumer, workingCopy, message);
        return svnMaterial.latestModification(workingCopy, new TestSubprocessExecutionContext());
    }

    @Override
    public List<Modification> latestModification() {
        final File workingCopy = TestFileUtil.createTempFolder("foo" + System.currentTimeMillis());
        return material().latestModification(workingCopy, new TestSubprocessExecutionContext());
    }


    public void checkInOneFile(String fileName, SvnMaterial svnMaterial) throws Exception {
        final File baseDir = TestFileUtil.createTempFolder("foo");
        tmpFolders.add(baseDir);

        ProcessOutputStreamConsumer consumer = inMemoryConsumer();

        Revision revision = latestRevision(svnMaterial, baseDir, new TestSubprocessExecutionContext());
        svnMaterial.updateTo(consumer, baseDir, new RevisionContext(revision), new TestSubprocessExecutionContext());

        File workingDir = new File(baseDir, svnMaterial.getFolder());
        File newFileToAdd = new File(workingDir, fileName);
        newFileToAdd.getParentFile().mkdirs();

        FileUtil.writeContentToFile("", newFileToAdd);
        svnMaterial.add(consumer, newFileToAdd);
        svnMaterial.commit(consumer, workingDir, "adding file [" + svnMaterial.getFolder() + "/" + fileName + "]");
    }

    private Revision latestRevision(SvnMaterial material, File workingDir, TestSubprocessExecutionContext execCtx) {
        List<Modification> modifications = material.latestModification(workingDir, execCtx);
        return new Modifications(modifications).latestRevision(material);
    }

}