com.marklogic.client.ext.file.CacheBusterDocumentFileProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of marklogic-data-hub Show documentation

Library for Creating an Operational Data Hub on MarkLogic

There is a newer version: 6.1.1

/*
 * Copyright (c) 2021 MarkLogic Corporation
 *
 * 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.marklogic.client.ext.file;

import com.marklogic.client.ext.helper.FilenameUtil;
import com.marklogic.client.ext.helper.LoggingObject;
import com.marklogic.client.io.Format;
import org.apache.commons.io.IOUtils;
import org.springframework.core.io.Resource;
import org.springframework.util.FileCopyUtils;

import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.util.UUID;

public class CacheBusterDocumentFileProcessor extends LoggingObject implements DocumentFileProcessor {

    @Override
    public DocumentFile processDocumentFile(DocumentFile documentFile) {
        if (documentFile.getFormat() == Format.BINARY) {
            return documentFile;
        } else {
            String text = documentFile.getModifiedContent();
            if (text == null) {
                Resource resource = documentFile.getResource();
                if (resource != null) {
                    InputStream inputStream = null;
                    try {
                        inputStream = resource.getInputStream();
                        text = new String(FileCopyUtils.copyToByteArray(inputStream), StandardCharsets.UTF_8);
                    } catch (IOException ie) {
                        logger.warn("Unable to replace tokens in file: " + documentFile.getUri() + "; cause: " + ie.getMessage());
                    } finally {
                        IOUtils.closeQuietly(inputStream);
                    }
                }
            }
            if (text != null) {
                Resource resource = documentFile.getResource();
                String comment = "";
                if (FilenameUtil.isJavascriptFile(resource.getFilename())) {
                    comment = "// cache buster: " + UUID.randomUUID() + "\n";
                } else if (FilenameUtil.isXqueryFile(resource.getFilename())) {
                    comment = "(: cache buster: " + UUID.randomUUID() + " :)\n";
                }

                text = comment + text;
                documentFile.setModifiedContent(text);
            }
            return documentFile;
        }
    }
}