import java.io.File;
import java.net.URL;
import java.nio.file.Files;

import javax.jcr.Repository;
import javax.jcr.Session;

import org.infinispan.schematic.document.ParsingException;
import org.modeshape.common.collection.Problems;
import org.modeshape.jcr.ModeShapeEngine;
import org.modeshape.jcr.RepositoryConfiguration;
import org.modeshape.jcr.api.RepositoryManager;

public class ModeShapeBug {
    private static final String REPO_CONFIG_FILE = "repo-config.json";
    private static final String WORKSPACE_NAME = "configuration";

    private static URL repositoryUrl;
    private static ModeShapeEngine engine;
    private static RepositoryConfiguration config;

    public static void main(String[] args) throws Exception {
        repositoryUrl = ModeShapeBug.class.getClassLoader().getResource(REPO_CONFIG_FILE);
        engine = new ModeShapeEngine();

        engine.start();
        try {
            config = readConfig(repositoryUrl);
            engine.deploy(config);
            File tempDir = Files.createTempDirectory("modeshape-restore-bug-").toFile();

            String oneBackslash = "oneBackslash";
            String twoBackslashes = "twoBackslashes";
            String threeBackslashes = "threeBackslashes";

            setPropertyValue(oneBackslash, "one\\backslash");
            setPropertyValue(twoBackslashes, "two\\\\backslashes");
            setPropertyValue(threeBackslashes, "three\\\\\\backslashes");

            System.out.printf("Values before creating backup:\n");
            printPropertyValue(oneBackslash);
            printPropertyValue(twoBackslashes);
            printPropertyValue(threeBackslashes);

            createBackup(tempDir);
            restoreBackup(tempDir);

            System.out.printf("Values after restoring backup:\n");
            printPropertyValue(oneBackslash);
            printPropertyValue(twoBackslashes);
            printPropertyValue(threeBackslashes);
        } finally {
            engine.shutdown().get();
        }
    }

    private static String printPropertyValue(String property) throws Exception {
        Session session = startSession();
        try {
            String value = session.getRootNode().getProperty(property).getString();
            System.out.printf("The value of '%s' is '%s'\n", property, value);
            return value;
        } finally {
            session.logout();
        }
    }

    private static void setPropertyValue(String property, String value) throws Exception {
        Session session = startSession();
        try {
            session.getRootNode().setProperty(property, value);
            session.save();
        } finally {
            session.logout();
        }
    }

    private static void createBackup(File tempDir) throws Exception {
        Session session = startSession();
        try {
            RepositoryManager repositoryManager =
                    ((org.modeshape.jcr.api.Session)session).getWorkspace().getRepositoryManager();
            org.modeshape.jcr.api.Problems problems = repositoryManager.backupRepository(tempDir);
            if (problems.hasProblems()){
                throw new RuntimeException("Problems creating backup: " + problems);
            }
        } finally {
            session.logout();
        }
    }

    private static void restoreBackup(File tempDir) throws Exception {
        Session session = startSession();
        try {
            RepositoryManager repositoryManager =
                    ((org.modeshape.jcr.api.Session)session).getWorkspace().getRepositoryManager();
            org.modeshape.jcr.api.Problems problems = repositoryManager.restoreRepository(tempDir);
            if (problems.hasProblems()){
                throw new RuntimeException("Problems restoring backup: " + problems);
            }
        } finally {
            session.logout();
        }
    }

    private static RepositoryConfiguration readConfig(URL repositoryUrl) throws ParsingException {
        RepositoryConfiguration config = RepositoryConfiguration.read(repositoryUrl);
        Problems problems = config.validate();
        if (problems.hasErrors()) {
            throw new RuntimeException("Problems starting the engine: " + problems);
        }
        return config;
    }

    private static Session startSession()
            throws Exception {
        Repository repository = engine.getRepository(config.getName());
        Session session = repository.login(WORKSPACE_NAME);
        return session;
    }
}