io.github.deweyjose.graphqlcodegen.SchemaFileManifest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of graphqlcodegen-maven-plugin Show documentation
Maven port of the Netflix DGS GraphQL Codegen gradle build plugin
The newest version!
package io.github.deweyjose.graphqlcodegen;

import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import nu.studer.java.util.OrderedProperties;
import nu.studer.java.util.OrderedProperties.OrderedPropertiesBuilder;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.math.BigInteger;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.security.MessageDigest;
import java.util.HashSet;
import java.util.Set;

@Slf4j
public class SchemaFileManifest {
    private Set files;
    private final File manifestPath;
    private final File projectPath;

    /**
     * Manifest constructor loads the properties file into memory.
     * The properties file has a property for each path with the
     * previous known checksum.
     *
     * @param files
     * @param manifestPath
     * @param projectPath
     */
    public SchemaFileManifest(Set files, File manifestPath, File projectPath) {
        this.files = files;
        this.manifestPath = manifestPath;
        this.projectPath = projectPath;
    }

    public SchemaFileManifest(File manifestPath, File projectPath) {
        this.manifestPath = manifestPath;
        this.projectPath = projectPath;
    }

    @SneakyThrows
    public static String generateChecksum(File path) {
        byte[] data = Files.readAllBytes(Paths.get(path.toURI()));
        byte[] hash = MessageDigest.getInstance("MD5").digest(data);
        String checksum = new BigInteger(1, hash).toString(16);
        return checksum;
    }

    /**
     * We only care about files ending with .graphql(s)
     *
     * @param file
     * @return boolean
     */
    public static boolean isGraphqlFile(File file) {
        return file.getName().endsWith(".graphqls") ||
            file.getName().endsWith(".graphql");
    }

    /**
     * Traverse the directory structure collecting .graphql(s) files.
     *
     * @param directory
     * @return Set
     */
    public static Set findGraphQLSFiles(File directory) {
        Set result = new HashSet<>();

        File[] contents = directory.listFiles();
        if (contents != null) {
            for (File content : contents) {
                if (content.isFile() && isGraphqlFile(content)) {
                    result.add(content);
                } else if (content.isDirectory()) {
                    Set subdirectoryGraphQLSFiles = findGraphQLSFiles(content);
                    result.addAll(subdirectoryGraphQLSFiles);
                }
            }
        }

        return result;
    }

    public void setFiles(Set files) {
        this.files = files;
    }

    /**
     * Computes the Set of files that have changed or are new
     * and need to trigger code generation.
     *
     * @return Set
     */
    public Set getChangedFiles() {
        Set changed = new HashSet<>();
        OrderedProperties manifest = loadManifest();
        for (File file : files) {
            String oldChecksum = manifest.getProperty(relativizeToProject(file));
            if (oldChecksum == null) {
                log.info("{} is new, will generate code", file.getName());
            } else if (!oldChecksum.equals(generateChecksum(file))) {
                log.info("{} has changed, will generate code", file.getName());
            } else {
                log.info("{} has not changed, will not generate code", file.getName());
                continue;
            }
            changed.add(file);
        }
        return changed;
    }

    /**
     * Clear the old manifest, compute new checksums
     * for each file and save the properties file.
     */
    @SneakyThrows
    public void syncManifest() {
        OrderedProperties manifest = new OrderedPropertiesBuilder()
                .withSuppressDateInComment(true)
                .build();
        for (File file : files) {
            manifest.setProperty(relativizeToProject(file), generateChecksum(file));
        }

        if (!manifestPath.exists()) {
            manifestPath.getParentFile().mkdirs();
        }

        try (FileOutputStream fos = new FileOutputStream(manifestPath)) {
            manifest.store(fos, "Schema Manifest");
            fos.flush();
        }
    }

    @SneakyThrows
    private OrderedProperties loadManifest() {
        OrderedProperties properties = new OrderedPropertiesBuilder()
                .withSuppressDateInComment(true)
                .build();
        if (manifestPath.exists()) {
            try (FileInputStream fis = new FileInputStream(manifestPath)) {
                properties.load(fis);
            }
        }
        return properties;
    }

    private String relativizeToProject(File file) {
        return projectPath.toPath().relativize(file.toPath()).toString();
    }
}