com.thoughtworks.go.helper.SvnRemoteRepository.java Source code

Introduction

Here is the source code for com.thoughtworks.go.helper.SvnRemoteRepository.java

Source

/*
 * Copyright 2019 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.
 */
package com.thoughtworks.go.helper;

import com.thoughtworks.go.util.FileUtil;
import com.thoughtworks.go.util.ProcessWrapper;
import com.thoughtworks.go.util.ReflectionUtil;
import com.thoughtworks.go.util.command.CommandLine;
import com.thoughtworks.go.util.command.EnvironmentVariableContext;
import com.thoughtworks.go.util.command.InMemoryStreamConsumer;
import org.apache.commons.io.FileUtils;
import org.junit.rules.TemporaryFolder;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;

import static com.thoughtworks.go.util.command.ProcessOutputStreamConsumer.inMemoryConsumer;
import static java.nio.charset.StandardCharsets.UTF_8;

public class SvnRemoteRepository {
    public SvnTestRepo repo;
    public ProcessWrapper processWrapper;
    public InMemoryStreamConsumer consumer;
    public int port;

    public SvnRemoteRepository(TemporaryFolder temporaryFolder) throws IOException {
        this.repo = new SvnTestRepo(temporaryFolder);

    }

    public void start() throws Exception {
        if (processWrapper != null) {
            throw new RuntimeException("Cannot start repository twice!");
        }

        port = RandomPort.find(toString());
        CommandLine svnserve = CommandLine
                .createCommandLine("svnserve").withArgs("-d", "--foreground", "--listen-port",
                        Integer.toString(port), "-r", repo.projectRepositoryRoot().getCanonicalPath())
                .withEncoding("utf-8");
        consumer = inMemoryConsumer();
        processWrapper = svnserve.execute(consumer, new EnvironmentVariableContext(), null);

        RandomPort.waitForPort(port);

        if (!processWrapper.isRunning()) {
            throw new RuntimeException("Could not run command " + svnserve);
        }
    }

    public void addUser(String username, String password) throws Exception {
        enableAuthentication();
        File passwdFile = new File(repo.projectRepositoryRoot(), "conf/passwd");
        String passwd = String.join(FileUtil.lineSeparator(), Files.readAllLines(passwdFile.toPath()));
        if (!(passwd.contains("\n[users]\n"))) {
            passwd = passwd + "\n[users]\n";
        }
        passwd = passwd + String.format("\n%s = %s\n", username, password);
        FileUtils.writeStringToFile(passwdFile, passwd, UTF_8);
    }

    private void enableAuthentication() throws IOException {
        File confFile = new File(repo.projectRepositoryRoot(), "conf/svnserve.conf");
        String passwd = "[general]\n" + "anon-access = none\n" + "auth-access = read\n" + "auth-access = write\n"
                + "password-db = passwd\n";
        FileUtils.writeStringToFile(confFile, passwd, UTF_8);
    }

    public void stop() throws Exception {
        if (processWrapper != null) {
            Process process = (Process) ReflectionUtil.getField(processWrapper, "process");
            process.destroy();
            processWrapper.waitForExit();
            processWrapper = null;
        }
    }

    public void tearDown() throws Exception {
        stop();
        repo.tearDown();
    }

    public String getUrl() {
        return "svn://127.0.0.1:" + port;
    }
}