com.hibegin.http.server.config.LocalFileStaticResourceLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of simplewebserver Show documentation
Simple, flexible, less dependent, more extended. Less memory footprint, can quickly build Web project. Can quickly run embedded, Android devices
There is a newer version: 0.3.162
Show newest version
package com.hibegin.http.server.config;

import com.hibegin.http.io.LengthByteArrayInputStream;
import com.hibegin.template.BasicTemplateRender;

import java.io.*;
import java.text.SimpleDateFormat;
import java.util.*;

public class LocalFileStaticResourceLoader implements StaticResourceLoader {

    private final boolean enableAutoIndex;
    private final String aliasPath;
    private final String location;


    private String buildHtmlStr(File file, String basePath) {
        StringBuilder sb = new StringBuilder();
        File[] fileArr = file.listFiles();
        if (fileArr != null) {
            List files = Arrays.asList(fileArr);
            if (!Objects.equals(new File(basePath).toString(), file.toString())) {
                sb.append("../\n");
            }
            if (files.isEmpty()) {
                sb.append("Empty folder");
            }
            sortFiles(files);
            for (File f : files) {
                appendFileInfo(sb, f, basePath, 48);
            }
        }
        return sb.toString();
    }

    private void appendFileInfo(StringBuilder sb, File f, String basePath, int maxLength) {
        String[] arr = fullBlankChar(f.getName(), maxLength);
        if (f.isDirectory()) {
            sb.append("").
                    append(arr[0]).append("/").append(arr[1].subSequence(0, arr[1].length() - 1)).
                    append(formatDateTime(f.lastModified()))
                    .append("                   -\n");
        } else {
            sb.append("")
                    .append(arr[0]).append("").append(arr[1]).
                    append(formatDateTime(f.lastModified()))
                    .append("                   ").append(f.length()).append("\n");
        }
    }

    public static String repeat(String str, int times) {
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < times; i++) {
            sb.append(str);
        }
        return sb.toString();
    }


    private static String[] fullBlankChar(String str, int maxLength) {
        String newStr = str;
        String[] newStrArr = new String[2];
        String blank = "";
        if (str.length() > maxLength) {
            newStr = newStr.substring(0, maxLength - 3) + "...";
        } else {
            blank = repeat(" ", maxLength - str.length());
        }
        newStrArr[0] = newStr;
        newStrArr[1] = blank + "   ";
        return newStrArr;
    }

    private static String formatDateTime(long time) {
        return new SimpleDateFormat("yyyy-MM-dd HH:mm").format(new Date(time));
    }


    private static void sortFiles(List files) {
        // 将隐藏文件放在列表前面
        files.sort((f1, f2) -> {
            if (f1.isHidden() && !f2.isHidden()) {
                return -1;
            } else if (!f1.isHidden() && f2.isHidden()) {
                return 1;
            } else {
                return f1.getName().compareTo(f2.getName());
            }
        });
    }

    private String changeFileSplitUriPath(String file) {
        return file.replace("\\", "/");
    }

    public LocalFileStaticResourceLoader(String location, String aliasPath) {
        this.enableAutoIndex = false;
        this.aliasPath = aliasPath;
        this.location = location;
    }

    public LocalFileStaticResourceLoader(boolean enableAutoIndex, String location, String aliasPath) {
        this.enableAutoIndex = enableAutoIndex;
        this.aliasPath = new File(changeFileSplitUriPath(aliasPath)).toString();
        this.location = location;
    }

    public boolean isDirectory(String path) {
        return new File(path).isDirectory();
    }

    public boolean isEnableAutoIndex() {
        return enableAutoIndex;
    }

    @Override
    public InputStream getInputStream(String path) {
        File file = new File(path);
        if (!file.exists()) {
            return null;
        }
        if (file.isDirectory()) {
            String fileFolder = file.toString().substring(aliasPath.length());
            //System.out.println("uri = " + fileFolder);
            Map map = new HashMap<>();
            String startPath = changeFileSplitUriPath(location + fileFolder + "/").replace("//", "/");
            map.put("startPath", startPath);
            map.put("fileHtmlStr", buildHtmlStr(file, startPath));
            return new LengthByteArrayInputStream(new BasicTemplateRender(map, LocalFileStaticResourceLoader.class).renderByTemplateName("/template/sf/index.html").getBytes());
        }
        try {
            return new FileInputStream(file);
        } catch (FileNotFoundException e) {
            throw new RuntimeException(e);
        }
    }
}