Stirling-PDF/src/main/java/stirling/software/SPDF/utils/GeneralUtils.java

package stirling.software.SPDF.utils;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.nio.file.FileVisitResult;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import org.springframework.web.multipart.MultipartFile;

import io.github.pixee.security.HostValidator;
import io.github.pixee.security.Urls;

public class GeneralUtils {

    public static File convertMultipartFileToFile(MultipartFile multipartFile) throws IOException {
        File tempFile = Files.createTempFile("temp", null).toFile();
        try (FileOutputStream os = new FileOutputStream(tempFile)) {
            os.write(multipartFile.getBytes());
        }
        return tempFile;
    }

    public static void deleteDirectory(Path path) throws IOException {
        Files.walkFileTree(
                path,
                new SimpleFileVisitor<Path>() {
                    @Override
                    public FileVisitResult visitFile(Path file, BasicFileAttributes attrs)
                            throws IOException {
                        Files.delete(file);
                        return FileVisitResult.CONTINUE;
                    }

                    @Override
                    public FileVisitResult postVisitDirectory(Path dir, IOException exc)
                            throws IOException {
                        Files.delete(dir);
                        return FileVisitResult.CONTINUE;
                    }
                });
    }

    public static String convertToFileName(String name) {
        String safeName = name.replaceAll("[^a-zA-Z0-9]", "_");
        if (safeName.length() > 50) {
            safeName = safeName.substring(0, 50);
        }
        return safeName;
    }

    public static boolean isValidURL(String urlStr) {
        try {
            Urls.create(
                    urlStr, Urls.HTTP_PROTOCOLS, HostValidator.DENY_COMMON_INFRASTRUCTURE_TARGETS);
            return true;
        } catch (MalformedURLException e) {
            return false;
        }
    }

    public static File multipartToFile(MultipartFile multipart) throws IOException {
        Path tempFile = Files.createTempFile("overlay-", ".pdf");
        try (InputStream in = multipart.getInputStream();
                FileOutputStream out = new FileOutputStream(tempFile.toFile())) {
            byte[] buffer = new byte[1024];
            int bytesRead;
            while ((bytesRead = in.read(buffer)) != -1) {
                out.write(buffer, 0, bytesRead);
            }
        }
        return tempFile.toFile();
    }

    public static Long convertSizeToBytes(String sizeStr) {
        if (sizeStr == null) {
            return null;
        }

        sizeStr = sizeStr.trim().toUpperCase();
        try {
            if (sizeStr.endsWith("KB")) {
                return (long)
                        (Double.parseDouble(sizeStr.substring(0, sizeStr.length() - 2)) * 1024);
            } else if (sizeStr.endsWith("MB")) {
                return (long)
                        (Double.parseDouble(sizeStr.substring(0, sizeStr.length() - 2))
                                * 1024
                                * 1024);
            } else if (sizeStr.endsWith("GB")) {
                return (long)
                        (Double.parseDouble(sizeStr.substring(0, sizeStr.length() - 2))
                                * 1024
                                * 1024
                                * 1024);
            } else if (sizeStr.endsWith("B")) {
                return Long.parseLong(sizeStr.substring(0, sizeStr.length() - 1));
            } else {
                // Assume MB if no unit is specified
                return (long) (Double.parseDouble(sizeStr) * 1024 * 1024);
            }
        } catch (NumberFormatException e) {
            // The numeric part of the input string cannot be parsed, handle this case
        }

        return null;
    }

    public static List<Integer> parsePageString(String pageOrder, int totalPages) {
        return parsePageString(pageOrder, totalPages, false);
    }

    public static List<Integer> parsePageString(
            String pageOrder, int totalPages, boolean isOneBased) {
        if (pageOrder == null || pageOrder.isEmpty()) {
            return Collections.singletonList(1);
        }
        if (pageOrder.matches("\\d+")) {
            // Convert the single number string to an integer and return it in a list
            return Collections.singletonList(Integer.parseInt(pageOrder));
        }
        return parsePageList(pageOrder.split(","), totalPages, isOneBased);
    }

    public static List<Integer> parsePageList(String[] pageOrderArr, int totalPages) {
        return parsePageList(pageOrderArr, totalPages, false);
    }

    public static List<Integer> parsePageList(
            String[] pageOrderArr, int totalPages, boolean isOneBased) {
        List<Integer> newPageOrder = new ArrayList<>();

        int adjustmentFactor = isOneBased ? 1 : 0;

        // loop through the page order array
        for (String element : pageOrderArr) {
            if ("all".equalsIgnoreCase(element)) {
                for (int i = 0; i < totalPages; i++) {
                    newPageOrder.add(i + adjustmentFactor);
                }
                // As all pages are already added, no need to check further
                break;
            } else if (element.matches("\\d*n\\+?-?\\d*|\\d*\\+?n")) {
                // Handle page order as a function
                int coefficient = 0;
                int constant = 0;
                boolean coefficientExists = false;
                boolean constantExists = false;

                if (element.contains("n")) {
                    String[] parts = element.split("n");
                    if (!"".equals(parts[0]) && parts[0] != null) {
                        coefficient = Integer.parseInt(parts[0]);
                        coefficientExists = true;
                    }
                    if (parts.length > 1 && !"".equals(parts[1]) && parts[1] != null) {
                        constant = Integer.parseInt(parts[1]);
                        constantExists = true;
                    }
                } else if (element.contains("+")) {
                    constant = Integer.parseInt(element.replace("+", ""));
                    constantExists = true;
                }

                for (int i = 1; i <= totalPages; i++) {
                    int pageNum = coefficientExists ? coefficient * i : i;
                    pageNum += constantExists ? constant : 0;

                    if (pageNum <= totalPages && pageNum > 0) {
                        newPageOrder.add(pageNum - adjustmentFactor);
                    }
                }
            } else if (element.contains("-")) {
                // split the range into start and end page
                String[] range = element.split("-");
                int start = Integer.parseInt(range[0]);
                int end = Integer.parseInt(range[1]);
                // check if the end page is greater than total pages
                if (end > totalPages) {
                    end = totalPages;
                }
                // loop through the range of pages
                for (int j = start; j <= end; j++) {
                    // print the current index
                    newPageOrder.add(j - adjustmentFactor);
                }
            } else {
                // if the element is a single page
                newPageOrder.add(Integer.parseInt(element) - adjustmentFactor);
            }
        }

        return newPageOrder;
    }

    public static boolean createDir(String path) {
        Path folder = Paths.get(path);
        if (!Files.exists(folder)) {
            try {
                Files.createDirectories(folder);
            } catch (IOException e) {
                e.printStackTrace();
                return false;
            }
        }
        return true;
    }
}