ruleList = StringUtils.split(accessUaBlocklist, StringUtils.LF);\n return ruleMatcher.matchAny(ruleList, currentAccessUA);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/io/EnsureContentLengthInputStreamResource.java",
"content": "/*\n * Copyright 2002-2018 the original author or authors.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * https://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage im.zhaojun.zfile.core.io;\n\nimport org.springframework.core.io.ByteArrayResource;\nimport org.springframework.core.io.InputStreamResource;\n\nimport java.io.InputStream;\n\n/**\n * \n * 自定义 EnsureContentLengthInputStreamResource 可以保证必须实现 InputStream 的 contentLength 方法返回实际的长度.\n * 此类相较于 {@link org.springframework.core.io.InputStreamResource} 仅实现了 contentLength 方法.\n *
\n * {@link org.springframework.core.io.Resource} implementation for a given {@link InputStream}.\n * Should only be used if no other specific {@code Resource} implementation\n * is applicable. In particular, prefer {@link ByteArrayResource} or any of the\n * file-based {@code Resource} implementations where possible.\n *\n *
In contrast to other {@code Resource} implementations, this is a descriptor\n * for an already opened resource - therefore returning {@code true} from\n * {@link #isOpen()}. Do not use an {@code InputStreamResource} if you need to\n * keep the resource descriptor somewhere, or if you need to read from a stream\n * multiple times.\n *\n * @author Juergen Hoeller\n * @author Sam Brannen\n * @since 28.12.2003\n * @see ByteArrayResource\n * @see org.springframework.core.io.ClassPathResource\n * @see org.springframework.core.io.FileSystemResource\n * @see org.springframework.core.io.UrlResource\n */\npublic class EnsureContentLengthInputStreamResource extends InputStreamResource {\n\n\tprivate final long contentLength;\n\n\t/**\n\t * Create a new InputStreamResource.\n\t * @param inputStream the InputStream to use\n\t */\n\tpublic EnsureContentLengthInputStreamResource(InputStream inputStream, long contentLength) {\n super(inputStream);\n this.contentLength = contentLength;\n\t}\n\n\t@Override\n\tpublic long contentLength() {\n\t\treturn contentLength;\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/model/request/PageQueryRequest.java",
"content": "package im.zhaojun.zfile.core.model.request;\n\nimport com.baomidou.mybatisplus.core.metadata.OrderItem;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport java.util.Objects;\n\n/**\n * 通用分页请求对象，可继承该类增加业务字段.\n *\n * @author zhaojun\n */\n@Data\npublic class PageQueryRequest {\n\t\n\t@Schema(title=\"分页页数\")\n\tprivate Integer page = 1;\n\t\n\t@Schema(title=\"每页条数\")\n\tprivate Integer limit = 10;\n\t\n\t@Schema(title=\"排序字段\")\n\tprivate String orderBy = \"create_date\";\n\t\n\t@Schema(title=\"排序顺序\")\n\tprivate String orderDirection = \"desc\";\n\n public OrderItem getOrderItem() {\n boolean asc = Objects.equals(orderDirection, \"asc\");\n return asc ? OrderItem.asc(orderBy) : OrderItem.desc(orderBy);\n }\n\t\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/AjaxJson.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport im.zhaojun.zfile.core.exception.ErrorCode;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\nimport lombok.ToString;\n\nimport java.io.Serializable;\n\n/**\n * ajax 请求返回 JSON 格式数据的封装\n *\n * @author zhaojun\n */\n@Data\n@ToString\npublic class AjaxJson implements Serializable {\n\n\tprivate static final long serialVersionUID = 1L; // 序列化版本号\n\n\tpublic static final String CODE_SUCCESS = \"0\"; // 成功状态码\n\n\t@Schema(title = \"业务状态码，0 为正常，其他值均为异常，异常情况下见响应消息\", example = \"0\")\n\tprivate final String code;\n\n\t@Schema(title = \"响应消息\", example = \"ok\")\n\tprivate String msg;\n\n\t@Schema(title = \"响应数据\")\n\tprivate T data;\n\n\t@Schema(title = \"数据总条数，分页情况有效\")\n\tprivate final Long dataCount;\n\t\n\t@Schema(title = \"跟踪 ID\")\n\tprivate String traceId;\n\n\tpublic AjaxJson(String code, String msg) {\n\t\tif (code == null) {\n\t\t\tcode = ErrorCode.SYSTEM_ERROR.getCode();\n\t\t}\n\t\tthis.code = code;\n\t\tthis.msg = msg;\n\t\tthis.dataCount = null;\n\t}\n\n\tpublic AjaxJson(String code, String msg, T data) {\n\t\tthis.code = code;\n\t\tthis.msg = msg;\n\t\tthis.data = data;\n\t\tthis.dataCount = null;\n\t}\n\n\tpublic AjaxJson(String code, String msg, T data, Long dataCount) {\n\t\tthis.code = code;\n\t\tthis.msg = msg;\n\t\tthis.data = data;\n\t\tthis.dataCount = dataCount;\n\t}\n\n\t// 返回成功\n\tpublic static AjaxJson getSuccess() {\n\t\treturn new AjaxJson<>(CODE_SUCCESS, \"ok\");\n\t}\n\n\tpublic static AjaxJson getSuccess(String msg) {\n\t\treturn new AjaxJson<>(CODE_SUCCESS, msg);\n\t}\n\n\tpublic static AjaxJson getSuccess(String msg, T data) {\n\t\treturn new AjaxJson<>(CODE_SUCCESS, msg, data);\n\t}\n\n\tpublic static AjaxJson getSuccessData(T data) {\n\t\treturn new AjaxJson<>(CODE_SUCCESS, \"ok\", data);\n\t}\n\n\t// 返回分页和数据的\n\tpublic static AjaxJson getPageData(Long dataCount, T data) {\n\t\treturn new AjaxJson<>(CODE_SUCCESS, \"ok\", data, dataCount);\n\t}\n\n\t// 返回错误\n\tpublic static AjaxJson getError(String msg) {\n\t\treturn new AjaxJson<>(ErrorCode.SYSTEM_ERROR.getCode(), msg);\n\t}\n\n\t// 返回未登录\n\tpublic static AjaxJson getUnauthorizedResult() {\n\t\treturn new AjaxJson<>(ErrorCode.BIZ_UNAUTHORIZED.getCode(), \"未登录，请登录后再次访问\");\n\t}\n\n\t// 返回没权限的\n\tpublic static AjaxJson getForbiddenResult() {\n\t\treturn new AjaxJson<>(ErrorCode.NO_FORBIDDEN.getCode(), \"未授权，请登录正确权限账号再试\");\n\t}\n\n\t// 返回未找到的\n\tpublic static AjaxJson getNotFoundResult() {\n\t\treturn new AjaxJson<>(ErrorCode.BIZ_NOT_FOUND.getCode(), ErrorCode.BIZ_NOT_FOUND.getMessage());\n\t}\n\n\tpublic static AjaxJson getError(String code, String msg) {\n\t\treturn new AjaxJson<>(code, msg);\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/ArrayUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\n/**\n * 数组工具类\n *\n * @author zhaojun\n */\npublic class ArrayUtils {\n\n /**\n * 数组是否为空\n *\n * @param \n * 数组元素类型\n *\n * @param array\n * 数组\n *\n * @return 是否为空\n */\n public static boolean isEmpty(T[] array) {\n return array == null || array.length == 0;\n }\n\n /**\n * 数组是否不为空\n *\n * @param \n * 数组元素类型\n *\n * @param array\n * 数组\n *\n * @return 是否不为空\n */\n public static boolean isNotEmpty(T[] array) {\n return !isEmpty(array);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/CharPool.java",
"content": "package im.zhaojun.zfile.core.util;\n\npublic interface CharPool {\n\n /**\n * CHAR 常量：斜杠 {@code '/'} ASCII 47\n */\n char SLASH_CHAR = '/';\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/CharSequenceUtil.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.core.text.StrSplitter;\nimport jakarta.annotation.Nullable;\nimport org.apache.commons.lang3.StringUtils;\n\nimport java.util.Collection;\nimport java.util.List;\n\n/**\n * 字符串工具类\n *\n * @author zhaojun\n */\npublic class CharSequenceUtil implements CharPool {\n\n /**\n * 找不到索引时的返回值\n */\n public static final int INDEX_NOT_FOUND = -1;\n\n /**\n * 字符串常量：{@code \"null\"}
\n * 注意：{@code \"null\" != null}\n */\n public static final String NULL = \"null\";\n\n /**\n * 字符串常量：空字符串 {@code \"\"}\n */\n public static final String EMPTY = \"\";\n\n /**\n * 字符串常量：空格符 {@code \" \"}\n */\n public static final String SPACE = \" \";\n\n\n /**\n * 获取 CharSequence 的长度, 如果为 null, 返回 0\n *\n * @param ch\n * 要获取长度的 CharSequence, 可能为 null\n *\n * @return CharSequence 的长度\n */\n public static int length(final @Nullable CharSequence ch) {\n return ch == null ? 0 : ch.length();\n }\n\n\n /**\n * {@link CharSequence} 转为字符串\n *\n * @param cs\n * {@link CharSequence}\n *\n * @return 字符串\n */\n public static String str(final @Nullable CharSequence cs) {\n return null == cs ? null : cs.toString();\n }\n\n\n /**\n * 判断 CharSequence 是否为空\n *\n * @param cs\n * {@link CharSequence}\n *\n * @return 是否为空\n */\n public static boolean isEmpty(final @Nullable CharSequence cs) {\n return cs == null || cs.isEmpty();\n }\n\n\n /**\n * CharSequence 是否不为空\n *\n * @param cs\n * {@link CharSequence}\n *\n * @return 是否不为空\n */\n public static boolean isNotEmpty(final @Nullable CharSequence cs) {\n return !isEmpty(cs);\n }\n\n\n /**\n * 指定字符串数组中的元素，是否全部为空字符串。
\n * 如果指定的字符串数组的长度为 0，或者所有元素都是空字符串，则返回 true。
\n *
\n *\n * 例：
\n * \n * - {@code CharSequenceUtil.isAllEmpty() // true}
\n * - {@code CharSequenceUtil.isAllEmpty(\"\", null) // true}
\n * - {@code CharSequenceUtil.isAllEmpty(\"123\", \"\") // false}
\n * - {@code CharSequenceUtil.isAllEmpty(\"123\", \"abc\") // false}
\n * - {@code CharSequenceUtil.isAllEmpty(\" \", \"\\t\", \"\\n\") // false}
\n *
\n *\n * @param strs\n * 字符串列表\n *\n * @return 所有字符串是否都为空\n */\n public static boolean isAllEmpty(final @Nullable CharSequence... strs) {\n if (strs == null) {\n return true;\n }\n\n for (CharSequence str : strs) {\n if (isNotEmpty(str)) {\n return false;\n }\n }\n return true;\n }\n\n /**\n * 是否包含空字符串。
\n * 如果指定的字符串数组的长度为 0，或者其中的任意一个元素是空字符串，则返回 true。
\n *
\n *\n * 例：
\n * \n * - {@code CharSequenceUtil.hasEmpty() // true}
\n * - {@code CharSequenceUtil.hasEmpty(\"\", null) // true}
\n * - {@code CharSequenceUtil.hasEmpty(\"123\", \"\") // true}
\n * - {@code CharSequenceUtil.hasEmpty(\"123\", \"abc\") // false}
\n * - {@code CharSequenceUtil.hasEmpty(\" \", \"\\t\", \"\\n\") // false}
\n *
\n *\n * @param strs\n * 字符串列表\n *\n * @return 是否包含空字符串\n */\n public static boolean hasEmpty(final @Nullable CharSequence... strs) {\n if (ArrayUtils.isEmpty(strs)) {\n return true;\n }\n\n for (CharSequence str : strs) {\n if (isEmpty(str)) {\n return true;\n }\n }\n return false;\n }\n\n\n /**\n * 指定字符串数组中的元素，是否都不为空字符串。
\n * 如果指定的字符串数组的长度不为 0，或者所有元素都不是空字符串，则返回 true。
\n *
\n *\n * 例：
\n * \n * - {@code CharSequenceUtil.isAllNotEmpty() // false}
\n * - {@code CharSequenceUtil.isAllNotEmpty(\"\", null) // false}
\n * - {@code CharSequenceUtil.isAllNotEmpty(\"123\", \"\") // false}
\n * - {@code CharSequenceUtil.isAllNotEmpty(\"123\", \"abc\") // true}
\n * - {@code CharSequenceUtil.isAllNotEmpty(\" \", \"\\t\", \"\\n\") // true}
\n *
\n *\n * @param args\n * 字符串数组\n *\n * @return 所有字符串是否都不为为空白\n */\n public static boolean isAllNotEmpty(final @Nullable CharSequence... args) {\n return !hasEmpty(args);\n }\n\n\n /**\n * 字符串是否为空白\n *\n * @param ch\n * 要判断的字符串, 可能为 null\n *\n * @return 是否为空白\n */\n public static boolean isBlank(final @Nullable CharSequence ch) {\n final int strLen = ch == null ? 0 : ch.length();\n if (strLen == 0) {\n return true;\n }\n for (int i = 0; i < strLen; i++) {\n if (!Character.isWhitespace(ch.charAt(i))) {\n return false;\n }\n }\n return true;\n }\n\n\n /**\n * 字符串是否不为空白\n *\n * @param cs\n * 字符串\n *\n * @return 是否不为空白\n */\n public static boolean isNotBlank(final @Nullable CharSequence cs) {\n return !isBlank(cs);\n }\n\n\n /**\n * 比较两个 CharSequence 是否相等, 区分大小写, 如果两个都为 null, 返回 true\n *\n * @param cs1\n * CharSequence 1, 可能为 null\n *\n * @param cs2\n * CharSequence 2, 可能为 null\n *\n * @return 是否相等\n */\n public static boolean equals(final @Nullable CharSequence cs1, final @Nullable CharSequence cs2) {\n if (cs1 == cs2) {\n return true;\n }\n if (cs1 == null || cs2 == null) {\n return false;\n }\n if (cs1.length() != cs2.length()) {\n return false;\n }\n if (cs1 instanceof String && cs2 instanceof String) {\n return cs1.equals(cs2);\n }\n // 逐个比较\n final int length = cs1.length();\n for (int i = 0; i < length; i++) {\n if (cs1.charAt(i) != cs2.charAt(i)) {\n return false;\n }\n }\n return true;\n }\n\n\n /**\n * 比较两个 CharSequence 是否相等, 可以选择是否忽略大小写, 如果两个都为 null, 返回 true\n *\n * @param cs1\n * 字符串 1\n *\n * @param cs2\n * 字符串 2\n *\n * @param ignoreCase\n * 是否忽略大小写\n *\n * @return 是否相等\n */\n public static boolean equals(final @Nullable CharSequence cs1,final @Nullable CharSequence cs2, boolean ignoreCase) {\n return ignoreCase ? equalsIgnoreCase(cs1, cs2) : equals(cs1, cs2);\n }\n\n\n /**\n * 字符串是否相等, 忽略大小写\n *\n * @param cs1\n * 字符串 1\n *\n * @param cs2\n * 字符串 2\n *\n * @return 忽略大小写后是否相等\n */\n public static boolean equalsIgnoreCase(final @Nullable CharSequence cs1, final @Nullable CharSequence cs2) {\n if (cs1 == cs2) {\n return true;\n }\n if (cs1 == null || cs2 == null) {\n return false;\n }\n if (cs1.length() != cs2.length()) {\n return false;\n }\n\n return cs1.toString().equalsIgnoreCase(cs2.toString());\n }\n\n\n /**\n * 切分字符串，如果分隔符不存在则返回原字符串\n *\n * @param str\n * 被切分的字符串\n *\n * @param separator\n * 分隔符\n *\n * @return 字符串\n */\n public static List split(final CharSequence str, final CharSequence separator) {\n return split(str, separator, false, false);\n }\n\n\n /**\n * 切分字符串\n *\n * @param str\n * 被切分的字符串\n *\n * @param separator\n * 分隔符字符\n *\n * @param isTrim\n * 是否去除切分字符串后每个元素两边的空格\n *\n * @param ignoreEmpty\n * 是否忽略空串\n *\n * @return 切分后的集合\n */\n public static List split(CharSequence str, CharSequence separator, boolean isTrim, boolean ignoreEmpty) {\n return split(str, separator, 0, isTrim, ignoreEmpty);\n }\n\n\n /**\n * 切分字符串\n *\n * @param str\n * 被切分的字符串\n *\n * @param separator\n * 分隔符字符\n *\n * @param limit\n * 限制分片数，-1 不限制\n *\n * @param isTrim\n * 是否去除切分字符串后每个元素两边的空格\n *\n * @param ignoreEmpty\n * 是否忽略空串\n *\n * @return 切分后的集合\n */\n public static List split(CharSequence str, CharSequence separator, int limit, boolean isTrim, boolean ignoreEmpty) {\n final String separatorStr = (null == separator) ? null : separator.toString();\n return StrSplitter.split(str, separatorStr, limit, isTrim, ignoreEmpty);\n }\n\n\n /**\n * 指定字符串是否在字符串中出现过\n *\n * @param str\n * 字符串\n *\n * @param searchStr\n * 被查找的字符串\n *\n * @return 是否包含\n */\n public static boolean contains(final @Nullable CharSequence str, final @Nullable CharSequence searchStr) {\n if (null == str || null == searchStr) {\n return false;\n }\n return str.toString().contains(searchStr);\n }\n\n\n /**\n * 查找指定字符串是否包含指定字符串列表中的任意一个字符串\n *\n * @param str\n * 指定字符串\n *\n * @param testStrs\n * 需要检查的字符串数组\n *\n * @return 是否包含任意一个字符串\n */\n public static boolean containsAny(final @Nullable CharSequence str, final @Nullable CharSequence... testStrs) {\n if (isEmpty(str) || ArrayUtils.isEmpty(testStrs)) {\n return false;\n }\n for (CharSequence checkStr : testStrs) {\n if (null != checkStr && str.toString().contains(checkStr)) {\n return true;\n }\n }\n return false;\n }\n\n\n /**\n * 查找指定字符串是否包含指定字符串列表中的任意一个字符串
\n * 忽略大小写\n *\n * @param str\n * 指定字符串\n *\n * @param testStrs\n * 需要检查的字符串数组\n *\n * @return 是否包含任意一个字符串\n */\n public static boolean containsAnyIgnoreCase(final @Nullable CharSequence str, final @Nullable CharSequence... testStrs) {\n return StringUtils.containsAnyIgnoreCase(str, testStrs);\n }\n\n\n /**\n * 以 conjunction 为分隔符将多个对象转换为字符串\n *\n * @param conjunction\n * 分隔符\n *\n * @param objs\n * 数组\n *\n * @return 连接后的字符串\n */\n public static String join(CharSequence conjunction, Object... objs) {\n StringBuilder sb = new StringBuilder();\n for (int i = 0; i < objs.length; i++) {\n Object item = objs[i];\n sb.append(item);\n if (i < objs.length - 1) {\n sb.append(conjunction);\n }\n }\n return sb.toString();\n }\n\n\n /**\n * 以 conjunction 为分隔符将 Collection 对象转换为字符串\n *\n * @param conjunction\n * 分隔符\n *\n * @param collection\n * 集合\n *\n * @return 连接后的字符串\n */\n public static String join(CharSequence conjunction, Collection collection) {\n StringBuilder sb = new StringBuilder();\n for (Object item : collection) {\n sb.append(item).append(conjunction);\n }\n if (!sb.isEmpty()) {\n sb.delete(sb.length() - conjunction.length(), sb.length());\n }\n return sb.toString();\n }\n\n\n /**\n * 是否以指定字符串开头\n *\n * @param str\n * 被监测字符串\n *\n * @param prefix\n * 开头字符串\n *\n * @return 是否以指定字符串开头\n */\n public static boolean startWith(CharSequence str, CharSequence prefix) {\n return startWith(str, prefix, false);\n }\n\n\n /**\n * 是否以指定字符串开头，忽略大小写\n *\n * @param str\n * 被监测字符串\n *\n * @param prefix\n * 开头字符串\n *\n * @return 是否以指定字符串开头\n */\n public static boolean startWithIgnoreCase(CharSequence str, CharSequence prefix) {\n return startWith(str, prefix, true);\n }\n\n\n /**\n * 是否以指定字符串开头
\n * 如果给定的字符串和开头字符串都为null则返回true，否则任意一个值为null返回false\n *\n * @param str\n * 被监测字符串\n *\n * @param prefix\n * 开头字符串\n *\n * @param ignoreCase\n * 是否忽略大小写\n *\n * @return 是否以指定字符串开头\n */\n public static boolean startWith(CharSequence str, CharSequence prefix, boolean ignoreCase) {\n return startWith(str, prefix, ignoreCase, false);\n }\n\n\n /**\n * 是否以指定字符串开头
\n * 如果给定的字符串和开头字符串都为 null 则返回 true，否则任意一个值为 null 返回 false
\n * \n * CharSequenceUtil.startWith(\"123\", \"123\", false, true); -- false\n * CharSequenceUtil.startWith(\"ABCDEF\", \"abc\", true, true); -- true\n * CharSequenceUtil.startWith(\"abc\", \"abc\", true, true); -- false\n *
\n *\n * @param str\n * 被监测字符串\n *\n * @param prefix\n * 开头字符串\n *\n * @param ignoreCase\n * 是否忽略大小写\n *\n * @param ignoreEquals\n * 是否忽略字符串相等的情况\n *\n * @return 是否以指定字符串开头\n */\n public static boolean startWith(final @Nullable CharSequence str, final @Nullable CharSequence prefix, boolean ignoreCase, boolean ignoreEquals) {\n if (null == str || null == prefix) {\n if (ignoreEquals) {\n return false;\n }\n return null == str && null == prefix;\n }\n\n boolean isStartWith = str.toString()\n .regionMatches(ignoreCase, 0, prefix.toString(), 0, prefix.length());\n\n if (isStartWith) {\n return (!ignoreEquals) || (!equals(str, prefix, ignoreCase));\n }\n return false;\n }\n\n\n /**\n * 是否以指定字符串结尾\n *\n * @param str\n * 被监测字符串\n *\n * @param suffix\n * 结尾字符串\n *\n * @return 是否以指定字符串结尾\n */\n public static boolean endWith(final @Nullable CharSequence str, final @Nullable CharSequence suffix) {\n return endWith(str, suffix, false);\n }\n\n\n /**\n * 是否以指定字符串结尾
\n * 如果给定的字符串和开头字符串都为null则返回true，否则任意一个值为null返回false\n *\n * @param str\n * 被监测字符串\n *\n * @param suffix\n * 结尾字符串\n *\n * @param ignoreCase\n * 是否忽略大小写\n *\n * @return 是否以指定字符串结尾\n */\n public static boolean endWith(final @Nullable CharSequence str, final @Nullable CharSequence suffix, boolean ignoreCase) {\n return endWith(str, suffix, ignoreCase, false);\n }\n\n\n /**\n * 是否以指定字符串结尾
\n * 如果给定的字符串和开头字符串都为null则返回true，否则任意一个值为null返回false\n *\n * @param str\n * 被监测字符串\n *\n * @param suffix\n * 结尾字符串\n *\n * @param ignoreCase\n * 是否忽略大小写\n *\n * @param ignoreEquals\n * 是否忽略字符串相等的情况\n *\n * @return 是否以指定字符串结尾\n */\n public static boolean endWith(final @Nullable CharSequence str, final @Nullable CharSequence suffix, boolean ignoreCase, boolean ignoreEquals) {\n if (null == str || null == suffix) {\n if (ignoreEquals) {\n return false;\n }\n return null == str && null == suffix;\n }\n\n final int strOffset = str.length() - suffix.length();\n boolean isEndWith = str.toString()\n .regionMatches(ignoreCase, strOffset, suffix.toString(), 0, suffix.length());\n\n if (isEndWith) {\n return (!ignoreEquals) || (!equals(str, suffix, ignoreCase));\n }\n return false;\n }\n\n\n /**\n * 去掉指定前缀\n *\n * @param str\n * 字符串\n *\n * @param prefix\n * 前缀\n *\n * @return 切掉后的字符串，若前缀不是 preffix， 返回原字符串\n */\n public static String removePrefix(final @Nullable CharSequence str, final @Nullable CharSequence prefix) {\n if (isEmpty(str) || isEmpty(prefix)) {\n return str(str);\n }\n String str2 = str.toString();\n String prefix2 = prefix.toString();\n if (str2.startsWith(prefix2)) {\n return str.subSequence(prefix.length(), str.length()).toString();\n }\n return str2; // 若前缀不是 prefix，返回原字符串\n }\n\n\n /**\n * 返回第一个非 {@code null} 元素\n *\n * @param strs\n * 多个元素\n *\n * @param \n * 元素类型\n *\n * @return 第一个非空元素，如果给定的数组为空或者都为空，返回{@code null}\n */\n @SuppressWarnings(\"unchecked\")\n public static T firstNonNull(T... strs) {\n if (ArrayUtils.isNotEmpty(strs)) {\n for (T str : strs) {\n if (isNotEmpty(str)) {\n return str;\n }\n }\n }\n return null;\n }\n\n\n /**\n * 截取分隔字符串之前的字符串，不包括分隔字符串
\n * 如果给定的字符串为空串（null或\"\"）或者分隔字符串为null，返回原字符串
\n * 如果分隔字符串为空串\"\"，则返回空串，如果分隔字符串未找到，返回原字符串，举例如下：\n *\n * \n * CharSequenceUtil.subBefore(null, *, false) = null\n * CharSequenceUtil.subBefore(\"\", *, false) = \"\"\n * CharSequenceUtil.subBefore(\"abc\", \"a\", false) = \"\"\n * CharSequenceUtil.subBefore(\"abcba\", \"b\", false) = \"a\"\n * CharSequenceUtil.subBefore(\"abc\", \"c\", false) = \"ab\"\n * CharSequenceUtil.subBefore(\"abc\", \"d\", false) = \"abc\"\n * CharSequenceUtil.subBefore(\"abc\", \"\", false) = \"\"\n * CharSequenceUtil.subBefore(\"abc\", null, false) = \"abc\"\n *
\n *\n * @param string\n * 被查找的字符串\n *\n * @param separator\n * 分隔字符串（不包括）\n *\n * @param isLastSeparator\n * 是否查找最后一个分隔字符串（多次出现分隔字符串时选取最后一个），true为选取最后一个\n *\n * @return 切割后的字符串\n */\n public static String subBefore(final @Nullable CharSequence string, final @Nullable CharSequence separator, boolean isLastSeparator) {\n if (isEmpty(string) || separator == null) {\n return null == string ? null : string.toString();\n }\n\n final String str = string.toString();\n final String sep = separator.toString();\n if (sep.isEmpty()) {\n return EMPTY;\n }\n final int pos = isLastSeparator ? str.lastIndexOf(sep) : str.indexOf(sep);\n if (INDEX_NOT_FOUND == pos) {\n return str;\n }\n if (0 == pos) {\n return EMPTY;\n }\n return str.substring(0, pos);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/ClassUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport java.lang.reflect.Field;\nimport java.lang.reflect.ParameterizedType;\nimport java.lang.reflect.Type;\n\n/**\n * Class & 反射相关工具类\n *\n * @author zhaojun\n */\npublic class ClassUtils {\n\n\tpublic static Class forName(String className) {\n try {\n return Class.forName(className);\n } catch (ClassNotFoundException e) {\n throw new RuntimeException(e);\n }\n }\n\n\t/**\n\t * 获取指定类的泛型类型, 只获取第一个泛型类型\n\t *\n\t * @param clazz\n\t * 泛型类\n\t *\n\t * @return 泛型类型\n\t */\n\tpublic static Class getClassFirstGenericsParam(Class clazz) {\n\t\tType genericSuperclass = clazz.getGenericSuperclass();\n\t\tType actualTypeArgument = ((ParameterizedType) genericSuperclass).getActualTypeArguments()[0];\n\t\treturn (Class) actualTypeArgument;\n\t}\n\n\tpublic static Class getGenericType(Field field) {\n\t\tParameterizedType listType = (ParameterizedType) field.getGenericType();\n\t\treturn (Class) listType.getActualTypeArguments()[0];\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/CollectionUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.core.lang.func.Func1;\n\nimport javax.annotation.Nullable;\nimport java.util.*;\n\npublic class CollectionUtils {\n\n\n /**\n * 判断集合是否为空\n *\n * @param collection\n * 集合\n *\n * @return 是否为空\n */\n public static boolean isEmpty(@Nullable Collection collection) {\n return (collection == null || collection.isEmpty());\n }\n\n\n /**\n * 判断集合是否不为空\n *\n * @param collection\n * 集合\n *\n * @return 是否不为空\n */\n public static boolean isNotEmpty(@Nullable Collection collection) {\n return !isEmpty(collection);\n }\n\n\n /**\n * 从集合中获取第一个元素, 如果集合为空则返回 {@code null}\n *\n * @param list\n * 集合，可能为 {@code null}\n *\n * @return 第一个元素，如果集合为空则返回 {@code null}\n */\n @Nullable\n public static T getFirst(@Nullable List list) {\n if (isEmpty(list)) {\n return null;\n }\n return list.get(0);\n }\n\n\n /**\n * 从集合中获取最后一个元素, 如果集合为空则返回 {@code null}\n *\n * @param list\n * 集合，可能为 {@code null}\n *\n * @return 最后一个元素，如果集合为空则返回 {@code null}\n */\n @Nullable\n public static T getLast(@Nullable List list) {\n if (isEmpty(list)) {\n return null;\n }\n return list.get(list.size() - 1);\n }\n\n\n /**\n * 加入全部\n *\n * @param \n * 集合元素类型\n *\n * @param collection\n * 被加入的集合 {@link Collection}\n *\n * @param values\n * 要加入的内容数组\n *\n * @return 原集合\n */\n public static Collection addAll(Collection collection, T[] values) {\n if (null != collection && null != values) {\n Collections.addAll(collection, values);\n }\n return collection;\n }\n\n\n /**\n * Iterable 转换为 Map, 根据指定的 keyFunc 函数生成 Key. Value 为 Iterable 中的元素.
\n * 可以指定将结果放入的 Map, 如不指定则会新建一个 HashMap 返回.\n *\n * @param \n * Map Key 类型\n *\n * @param \n * Map Value 类型\n *\n * @param values\n * 被转换的 Iterable\n *\n * @param map\n * 转换后的 Value 存放的 Map, 如果为 {@code null} 则新建一个 HashMap\n *\n * @param keyFunc\n * 生成 Map 的 Key 的函数\n *\n * @return 转换后的 Map\n */\n public static Map toMap(final @Nullable Iterable values, final @Nullable Map map, final @Nullable Func1 keyFunc) {\n if (values == null || keyFunc == null) {\n return Collections.emptyMap();\n }\n\n final Map result = map == null ? new HashMap<>() : map;\n\n for (V value : values) {\n try {\n result.put(keyFunc.call(value), value);\n } catch (Exception e) {\n throw new RuntimeException(e);\n }\n }\n return result;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/DnsUtil.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport com.alibaba.dcm.DnsCacheManipulator;\nimport com.alibaba.fastjson2.JSONArray;\nimport org.springframework.lang.Nullable;\n\npublic class DnsUtil {\n\n /**\n * 通过 HTTP DNS 获取域名对应的 IP 地址\n *\n * @param domain\n * 域名\n *\n * @return IP 地址数组\n */\n public static @Nullable String[] getDomainIpByHttpDns(String domain) {\n String jsonArrayStr = cn.hutool.http.HttpUtil.get(\"http://223.5.5.5/resolve?name=\" + domain + \"&short=1\", 3000);\n JSONArray jsonArray = JSONArray.parseArray(jsonArrayStr);\n if (!jsonArray.isEmpty()) {\n String[] result = new String[jsonArray.size()];\n for (int i = 0; i < jsonArray.size(); i++) {\n result[i] = jsonArray.getString(i);\n }\n return result;\n } else {\n return null;\n }\n }\n\n\n /**\n * 通过 HTTP DNS 获取域名对应的 IP 地址, 并设置 DNS 缓存.\n *\n * @param domain\n * 域名\n *\n * @param cacheTime\n * 缓存时间, 单位: 毫秒\n *\n * @return IP 地址数组\n */\n public static String[] getDomainIpByHttpDnsAndCache(String domain, int cacheTime) {\n String[] domainIpByHttpDns = getDomainIpByHttpDns(domain);\n if (domainIpByHttpDns != null) {\n // 设置 DNS 缓存\n DnsCacheManipulator.setDnsCache(cacheTime, domain, domainIpByHttpDns);\n }\n return domainIpByHttpDns;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/EnumConvertUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.core.convert.Convert;\nimport cn.hutool.core.util.ClassUtil;\nimport cn.hutool.core.util.ObjectUtil;\nimport cn.hutool.core.util.ReflectUtil;\nimport com.baomidou.mybatisplus.annotation.EnumValue;\nimport com.fasterxml.jackson.annotation.JsonValue;\n\nimport java.lang.reflect.Field;\n\n/**\n * 枚举转换工具类\n *\n * @author zhaojun\n */\npublic class EnumConvertUtils {\n\n\n\t/**\n\t * 根据枚举 class 和值获取对应的枚举对象\n\t *\n\t * @param clazz\n\t * 枚举类 Class\n\t *\n\t * @param value\n\t * 枚举值\n\t *\n\t * @return 枚举对象\n\t */\n\tpublic static Enum convertStrToEnum(Class clazz, Object value) {\n\t\tif (!ClassUtil.isEnum(clazz)) {\n\t\t\treturn null;\n\t\t}\n\n\t\tField[] fields = ReflectUtil.getFields(clazz);\n\t\tfor (Field field : fields) {\n\t\t\tboolean jsonValuePresent = field.isAnnotationPresent(JsonValue.class);\n\t\t\tboolean enumValuePresent = field.isAnnotationPresent(EnumValue.class);\n\n\t\t\tif (jsonValuePresent || enumValuePresent) {\n\t\t\t\tObject[] enumConstants = clazz.getEnumConstants();\n\n\t\t\t\tfor (Object enumObj : enumConstants) {\n\t\t\t\t\tif (ObjectUtil.equal(value, ReflectUtil.getFieldValue(enumObj, field))) {\n\t\t\t\t\t\treturn (Enum) enumObj;\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\treturn null;\n\t}\n\n\n\t/**\n\t * 转换枚举对象为字符串, 如果枚举对象没有定义 JsonValue 注解, 则使用 EnumValue 注解的值\n\t *\n\t * @param enumObj\n\t * 枚举对象\n\t *\n\t * @return 字符串\n\t */\n\tpublic static String convertEnumToStr(Object enumObj) {\n\t\tClass clazz = enumObj.getClass();\n\t\tif (!ClassUtil.isEnum(clazz)) {\n\t\t\treturn null;\n\t\t}\n\n\t\tField[] fields = ReflectUtil.getFields(clazz);\n\t\tfor (Field field : fields) {\n\t\t\tboolean jsonValuePresent = field.isAnnotationPresent(JsonValue.class);\n\t\t\tboolean enumValuePresent = field.isAnnotationPresent(EnumValue.class);\n\n\t\t\tif (jsonValuePresent || enumValuePresent) {\n\t\t\t\treturn Convert.toStr(ReflectUtil.getFieldValue(enumObj, field));\n\t\t\t}\n\t\t}\n\n\t\treturn null;\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/FileComparator.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.core.comparator.CompareUtil;\nimport im.zhaojun.zfile.module.storage.model.result.FileItemResult;\nimport im.zhaojun.zfile.module.storage.model.enums.FileTypeEnum;\n\nimport java.util.Comparator;\n\n/**\n * 文件比较器\n * \n * - 文件夹始终比文件排序高
\n * - 默认按照名称排序
\n * - 默认排序为升序
\n * - 按名称排序不区分大小写
\n *
\n * @author zhaojun\n */\npublic class FileComparator implements Comparator {\n\n private String sortBy;\n\n private String order;\n\n public FileComparator(String sortBy, String order) {\n this.sortBy = sortBy;\n this.order = order;\n }\n\n\n /**\n * 比较两个文件的大小\n *\n * @param o1\n * 第一个文件\n *\n * @param o2\n * 第二个文件\n *\n * @return 比较结果\n */\n @Override\n public int compare(FileItemResult o1, FileItemResult o2) {\n if (sortBy == null) {\n sortBy = \"name\";\n }\n\n if (order == null) {\n order = \"asc\";\n }\n FileTypeEnum o1Type = o1.getType();\n FileTypeEnum o2Type = o2.getType();\n NaturalOrderComparator naturalOrderComparator = new NaturalOrderComparator();\n if (o1Type.equals(o2Type)) {\n int result = switch (sortBy) {\n case \"time\" -> CompareUtil.compare(o1.getTime(), o2.getTime());\n case \"size\" -> CompareUtil.compare(o1.getSize(), o2.getSize());\n default -> naturalOrderComparator.compare(o1.getName(), o2.getName());\n };\n return \"asc\".equals(order) ? result : -result;\n }\n\n if (o1Type.equals(FileTypeEnum.FOLDER)) {\n return -1;\n } else {\n return 1;\n }\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/FileResponseUtil.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.core.io.FileUtil;\nimport im.zhaojun.zfile.core.exception.ErrorCode;\nimport im.zhaojun.zfile.core.exception.status.NotFoundAccessException;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.core.io.InputStreamResource;\nimport org.springframework.core.io.Resource;\nimport org.springframework.http.ContentDisposition;\nimport org.springframework.http.HttpHeaders;\nimport org.springframework.http.MediaType;\nimport org.springframework.http.ResponseEntity;\n\nimport java.io.File;\nimport java.nio.charset.StandardCharsets;\n\n/**\n * 将文件输出对象\n *\n * @author zhaojun\n */\n@Slf4j\npublic class FileResponseUtil {\n\n\n /**\n * 文件下载，单线程，不支持断点续传\n *\n * @param file\n * 文件对象\n *\n * @param fileName\n * 要保存为的文件名\n *\n * @return 文件下载对象\n */\n public static ResponseEntity exportSingleThread(File file, String fileName) {\n if (!file.exists()) {\n throw new NotFoundAccessException(ErrorCode.BIZ_FILE_NOT_EXIST);\n }\n\n MediaType mediaType = MediaType.APPLICATION_OCTET_STREAM;\n\n HttpHeaders headers = new HttpHeaders();\n\n if (StringUtils.isEmpty(fileName)) {\n fileName = file.getName();\n }\n\n ContentDisposition contentDisposition = ContentDisposition\n .builder(\"inline\")\n .filename(fileName, StandardCharsets.UTF_8)\n .build();\n headers.setContentDisposition(contentDisposition);\n\n return ResponseEntity\n .ok()\n .headers(headers)\n .contentLength(file.length())\n .contentType(mediaType)\n .body(new InputStreamResource(FileUtil.getInputStream(file)));\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/FileSizeConverter.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport java.util.regex.Matcher;\nimport java.util.regex.Pattern;\n\npublic class FileSizeConverter {\n\n private static final long KB_FACTOR = 1024L;\n private static final long MB_FACTOR = 1024L * KB_FACTOR;\n private static final long GB_FACTOR = 1024L * MB_FACTOR;\n private static final long TB_FACTOR = 1024L * GB_FACTOR;\n private static final long PB_FACTOR = 1024L * TB_FACTOR;\n\n private static final Pattern FILE_SIZE_PATTERN = Pattern.compile(\"([\\\\d.]+)\\\\s*([a-zA-Z]+)\");\n\n public static long convertFileSizeToBytes(String sizeStr) {\n if (sizeStr == null || sizeStr.trim().isEmpty()) {\n throw new IllegalArgumentException(\"输入字符串不能为空\");\n }\n\n Matcher matcher = FILE_SIZE_PATTERN.matcher(sizeStr.trim());\n\n if (!matcher.matches()) {\n throw new IllegalArgumentException(\"无效的文件大小格式: \" + sizeStr);\n }\n\n String valueStr = matcher.group(1);\n String unitStr = matcher.group(2).toUpperCase();\n\n double value;\n try {\n value = Double.parseDouble(valueStr);\n } catch (NumberFormatException e) {\n throw new IllegalArgumentException(\"无效的数字格式: \" + valueStr, e);\n }\n\n if (value < 0) {\n throw new IllegalArgumentException(\"文件大小不能为负数: \" + valueStr);\n }\n\n long multiplier = switch (unitStr) {\n case \"B\" ->\n 1L;\n case \"KB\", \"KIB\" ->\n KB_FACTOR;\n case \"MB\", \"MIB\" ->\n MB_FACTOR;\n case \"GB\", \"GIB\" ->\n GB_FACTOR;\n case \"TB\", \"TIB\" ->\n TB_FACTOR;\n case \"PB\", \"PIB\" ->\n PB_FACTOR;\n default -> throw new IllegalArgumentException(\"不支持的单位: \" + unitStr + \" (支持 B, KB, MB, GB, TB, PB)\");\n };\n\n double bytesDouble = value * multiplier;\n if (bytesDouble > Long.MAX_VALUE) {\n throw new ArithmeticException(\"转换后的字节数超过了 Long 类型的最大值: \" + bytesDouble);\n }\n\n return Math.round(bytesDouble);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/FileUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport org.apache.commons.io.FilenameUtils;\n\n/**\n * 文件相关工具类\n *\n * @author zhaojun\n */\npublic class FileUtils {\n\n public static String getName(final String fileName) {\n if (fileName == null) {\n return null;\n }\n\n int i = fileName.lastIndexOf(CharSequenceUtil.SLASH_CHAR);\n if (i >= 0 && i <= fileName.length() - 1) {\n return fileName.substring(i + 1);\n }\n\n return fileName;\n }\n\n public static String getParentPath(final String fileName) {\n String fullPathNoEndSeparator = FilenameUtils.getFullPathNoEndSeparator(StringUtils.trimEndSlashes(fileName));\n if (fullPathNoEndSeparator == null || fullPathNoEndSeparator.isEmpty()) {\n return StringUtils.SLASH;\n }\n return fullPathNoEndSeparator;\n }\n\n public static String getExtension(final String fileName) throws IllegalArgumentException {\n if (fileName == null) {\n return null;\n }\n\n int i = fileName.lastIndexOf('.');\n if (i > 0 && i < fileName.length() - 1) {\n return fileName.substring(i + 1);\n }\n\n return \"\";\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/HttpUtil.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport im.zhaojun.zfile.core.constant.ZFileConstant;\nimport im.zhaojun.zfile.core.exception.ErrorCode;\nimport im.zhaojun.zfile.core.exception.biz.GetPreviewTextContentBizException;\nimport im.zhaojun.zfile.core.exception.core.BizException;\nimport lombok.extern.slf4j.Slf4j;\n\nimport java.io.IOException;\nimport java.net.URL;\nimport java.net.URLConnection;\n\n/**\n * 网络相关工具\n *\n * @author zhaojun\n */\n@Slf4j\npublic class HttpUtil {\n\n /**\n * 获取 URL 对应的文件内容\n *\n * @param url\n * 文件 URL\n *\n * @return 文件内容\n */\n public static String getTextContent(String url) {\n long maxFileSize = 1024 * ZFileConstant.TEXT_MAX_FILE_SIZE_KB;\n\n if (getRemoteFileSize(url) > maxFileSize) {\n throw new BizException(ErrorCode.BIZ_PREVIEW_FILE_SIZE_EXCEED);\n }\n\n String result;\n try {\n result = cn.hutool.http.HttpUtil.get(url);\n } catch (Exception e) {\n throw new GetPreviewTextContentBizException(url, e);\n }\n\n return result == null ? \"\" : result;\n }\n\n\n /**\n * 获取远程文件大小\n *\n * @param url\n * 文件 URL\n *\n * @return 文件大小\n */\n public static Long getRemoteFileSize(String url) {\n long size = 0;\n URL urlObject;\n try {\n urlObject = new URL(url);\n URLConnection conn = urlObject.openConnection();\n size = conn.getContentLength();\n } catch (IOException e) {\n e.printStackTrace();\n }\n\n return size;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/NaturalOrderComparator.java",
"content": "package im.zhaojun.zfile.core.util;\n/*\n NaturalOrderComparator.java -- Perform 'natural order' comparisons of strings in Java.\n Copyright (C) 2003 by Pierre-Luc Paour \n\n Based on the C version by Martin Pool, of which this is more or less a straight conversion.\n Copyright (C) 2000 by Martin Pool \n\n This software is provided 'as-is', without any express or implied\n warranty. In no event will the authors be held liable for any damages\n arising from the use of this software.\n\n Permission is granted to anyone to use this software for any purpose,\n including commercial applications, and to alter it and redistribute it\n freely, subject to the following restrictions:\n\n 1. The origin of this software must not be misrepresented; you must not\n claim that you wrote the original software. If you use this software\n in a product, an acknowledgment in the product documentation would be\n appreciated but is not required.\n 2. Altered source versions must be plainly marked as such, and must not be\n misrepresented as being the original software.\n 3. This notice may not be removed or altered from any source distribution.\n */\n\nimport java.util.Comparator;\n\n/**\n * 类 windows 文件排序算法\n *\n * @author zhaojun\n */\npublic class NaturalOrderComparator implements Comparator {\n\n private static final char ZERO_CHAR = '0';\n\n private int compareRight(String a, String b) {\n int bias = 0, ia = 0, ib = 0;\n\n // The longest run of digits wins. That aside, the greatest\n // value wins, but we can't know that it will until we've scanned\n // both numbers to know that they have the same magnitude, so we\n // remember it in BIAS.\n for (; ; ia++, ib++) {\n char ca = charAt(a, ia);\n char cb = charAt(b, ib);\n\n if (!isDigit(ca) && !isDigit(cb)) {\n return bias;\n }\n if (!isDigit(ca)) {\n return -1;\n }\n if (!isDigit(cb)) {\n return +1;\n }\n if (ca == 0 && cb == 0) {\n return bias;\n }\n\n if (bias == 0) {\n if (ca < cb) {\n bias = -1;\n } else if (ca > cb) {\n bias = +1;\n }\n }\n }\n }\n\n @Override\n public int compare(String a, String b) {\n int ia = 0, ib = 0;\n int nza, nzb;\n char ca, cb;\n\n while (true) {\n // Only count the number of zeroes leading the last number compared\n nza = nzb = 0;\n\n ca = charAt(a, ia);\n cb = charAt(b, ib);\n\n // skip over leading spaces or zeros\n while (Character.isSpaceChar(ca) || ca == ZERO_CHAR) {\n if (ca == ZERO_CHAR) {\n nza++;\n } else {\n // Only count consecutive zeroes\n nza = 0;\n }\n\n ca = charAt(a, ++ia);\n }\n\n while (Character.isSpaceChar(cb) || cb == '0') {\n if (cb == '0') {\n nzb++;\n } else {\n // Only count consecutive zeroes\n nzb = 0;\n }\n\n cb = charAt(b, ++ib);\n }\n\n // Process run of digits\n if (Character.isDigit(ca) && Character.isDigit(cb)) {\n int bias = compareRight(a.substring(ia), b.substring(ib));\n if (bias != 0) {\n return bias;\n }\n }\n\n if (ca == 0 && cb == 0) {\n // The strings compare the same. Perhaps the caller\n // will want to call strcmp to break the tie.\n return compareEqual(a, b, nza, nzb);\n }\n if (ca < cb) {\n return -1;\n }\n if (ca > cb) {\n return +1;\n }\n\n ++ia;\n ++ib;\n }\n }\n\n private static boolean isDigit(char c) {\n return Character.isDigit(c) || c == '.' || c == ',';\n }\n\n private static char charAt(String s, int i) {\n return i >= s.length() ? 0 : s.charAt(i);\n }\n\n private static int compareEqual(String a, String b, int nza, int nzb) {\n if (nza - nzb != 0) {\n return nza - nzb;\n }\n\n if (a.length() == b.length()) {\n return a.compareTo(b);\n }\n\n return a.length() - b.length();\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/NumberUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\n/**\n * 数字工具类\n *\n * @author zhaojun\n */\npublic class NumberUtils {\n\n public static boolean isNullOrZero(Integer number) {\n return number == null || number == 0;\n }\n\n public static boolean isNotNullOrZero(Integer number) {\n return number != null && number != 0;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/OnlyOfficeKeyCacheUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.cache.Cache;\nimport cn.hutool.cache.CacheUtil;\nimport cn.hutool.cache.impl.CacheObj;\nimport im.zhaojun.zfile.module.onlyoffice.model.OnlyOfficeFile;\nimport lombok.extern.slf4j.Slf4j;\nimport org.apache.commons.lang3.BooleanUtils;\nimport org.apache.commons.lang3.RandomStringUtils;\n\nimport java.util.ArrayList;\nimport java.util.Iterator;\nimport java.util.List;\nimport java.util.concurrent.TimeUnit;\nimport java.util.concurrent.locks.ReentrantLock;\n\n/**\n * OnlyOffice 文件信息与 Key 缓存工具类\n *\n * @author zhaojun\n */\n@Slf4j\npublic class OnlyOfficeKeyCacheUtils {\n\n /**\n * 存储 OnlyOffice 文件信息与 Key 的映射关系. 最多存储 10000 个 Key, 防止内存溢出.\n */\n private static final Cache ONLY_OFFICE_FILE_KEY_MAP = CacheUtil.newLRUCache(10000);\n\n /**\n * 存储 OnlyOffice Key 与文件信息的映射关系. 最多存储 10000 个 Key, 防止内存溢出.\n */\n private static final Cache ONLY_OFFICE_KEY_FILE_MAP = CacheUtil.newLRUCache(10000);\n\n /**\n * 存储文件锁, 防止并发操作文件缓存时出现问题.\n */\n private static final Cache locks = CacheUtil.newLRUCache(300);\n\n /**\n * 获取该文件缓存的 key, 如果不存在则生成一个新的 key 并缓存.\n *\n * @param onlyOfficeFile\n * OnlyOffice 文件信息\n *\n * @return 该文件唯一标识\n */\n public static String getKeyOrPutNew(OnlyOfficeFile onlyOfficeFile, long timeout) {\n ReentrantLock lock = getLock(onlyOfficeFile);\n try {\n boolean getLock = lock.tryLock(timeout, TimeUnit.MILLISECONDS);\n if (BooleanUtils.isFalse(getLock)) {\n log.warn(\"{} 尝试获取锁超时, 强制忽略锁直接操作文件.\", onlyOfficeFile);\n }\n try {\n if (ONLY_OFFICE_FILE_KEY_MAP.containsKey(onlyOfficeFile)) {\n return ONLY_OFFICE_FILE_KEY_MAP.get(onlyOfficeFile);\n } else {\n String key = RandomStringUtils.randomAlphabetic(10);\n ONLY_OFFICE_FILE_KEY_MAP.put(onlyOfficeFile, key);\n ONLY_OFFICE_KEY_FILE_MAP.put(key, onlyOfficeFile);\n return key;\n }\n } finally {\n lock.unlock();\n }\n } catch (InterruptedException e) {\n Thread.currentThread().interrupt();\n throw new IllegalStateException(\"Thread was interrupted\", e);\n }\n\n }\n\n /**\n * 清理缓存中的 Key 与文件信息的映射关系.(文件发生了变化, 需要重新生成 OnlyOffice 预览链接时调用)\n *\n * @param key\n * 文件唯一标识\n */\n public static OnlyOfficeFile removeByKey(String key) {\n OnlyOfficeFile onlyOfficeFile = ONLY_OFFICE_KEY_FILE_MAP.get(key);\n if (onlyOfficeFile == null) {\n return null;\n }\n ONLY_OFFICE_FILE_KEY_MAP.remove(onlyOfficeFile);\n ONLY_OFFICE_KEY_FILE_MAP.remove(key);\n return onlyOfficeFile;\n }\n\n /**\n * 清理缓存中的文件信息与 Key 的映射关系.(文件发生了变化, 需要重新生成 OnlyOffice 预览链接时调用)\n *\n * @param onlyOfficeFile\n * OnlyOffice 文件信息\n */\n public static OnlyOfficeFile removeByFile(OnlyOfficeFile onlyOfficeFile) {\n String key = ONLY_OFFICE_FILE_KEY_MAP.get(onlyOfficeFile);\n if (key == null) {\n return null;\n }\n ONLY_OFFICE_FILE_KEY_MAP.remove(onlyOfficeFile);\n ONLY_OFFICE_KEY_FILE_MAP.remove(key);\n return onlyOfficeFile;\n }\n\n\n /**\n * 清理缓存中的某个文件夹下所有文件信息与 Key 的映射关系.(文件发生了变化, 需要重新生成 OnlyOffice 预览链接时调用)\n *\n * @param onlyOfficeFile\n * OnlyOffice 文件信息\n */\n public static List removeByFolder(OnlyOfficeFile onlyOfficeFile) {\n List caches = new ArrayList<>();\n Iterator> cacheObjIterator = ONLY_OFFICE_FILE_KEY_MAP.cacheObjIterator();\n while (cacheObjIterator.hasNext()) {\n CacheObj cacheObj = cacheObjIterator.next();\n OnlyOfficeFile cacheOnlyOfficeFile = cacheObj.getKey();\n if (cacheOnlyOfficeFile.getStorageKey().equals(onlyOfficeFile.getStorageKey())\n && StringUtils.startWith(cacheOnlyOfficeFile.getPathAndName(), onlyOfficeFile.getPathAndName())) {\n ONLY_OFFICE_FILE_KEY_MAP.remove(cacheObj.getKey());\n ONLY_OFFICE_KEY_FILE_MAP.remove(cacheObj.getValue());\n caches.add(cacheOnlyOfficeFile);\n }\n }\n return caches;\n }\n\n /**\n * 获取文件锁, 防止并发操作文件缓存时出现问题.\n *\n * @param key\n * 文件唯一标识\n *\n * @return 锁对象\n */\n public static ReentrantLock getLock(OnlyOfficeFile key) {\n return locks.get(key, true, ReentrantLock::new);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/PatternMatcherUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport java.nio.file.FileSystems;\nimport java.nio.file.PathMatcher;\nimport java.nio.file.Paths;\nimport java.util.HashMap;\nimport java.util.Map;\n\n/**\n * 规则表达式工具类\n *\n * @author zhaojun\n */\npublic class PatternMatcherUtils {\n\t\n\tprivate static final Map PATH_MATCHER_MAP = new HashMap<>();\n\t\n\t/**\n\t * 兼容模式的 glob 表达式匹配.\n\t * 默认的 glob 表达式是不支持以下情况的:
\n\t * \n\t * - pattern: /a/**
\n\t * - test1: /a
\n\t * - test2: /a/
\n\t * \n\t * test1 和 test 2 均无法匹配这种情况, 此方法兼容了这种情况, 即对 test 内容后拼接 \"/xx\"(其实任意字符都可以), 使其可以匹配上 pattern.\n\t *
注意：但此方法对包含文件名的情况无效, 仅支持 test 为 路径的情况.\n\t *\n\t * @param \tpattern\n\t *\t\t\tglob 规则表达式\n\t *\n\t * @param \ttest\n\t *\t\t\t匹配内容\n\t *\n\t * @return \t是否匹配.\n\t */\n\tpublic static boolean testCompatibilityGlobPattern(String pattern, String test) {\n\t\t// 如果规则表达式最开始没有 /, 则兼容在最前方加上 /.\n\t\tif (!StringUtils.startWith(pattern, StringUtils.SLASH)) {\n\t\t\tpattern = StringUtils.SLASH + pattern;\n\t\t}\n\t\t\n\t\t// 兼容性处理.\n\t\ttest = StringUtils.concat(test, StringUtils.SLASH);\n\t\tif (StringUtils.endWith(pattern, \"/**\") || StringUtils.endWith(pattern, \"/*\")) {\n\t\t\ttest += \"xxx\";\n\t\t}\n\t\treturn testGlobPattern(pattern, test);\n\t}\n\t\n\t\n\t/**\n\t * 测试密码规则表达式和文件路径是否匹配\n\t *\n\t * @param \tpattern\n\t *\t\t\tglob 规则表达式\n\t *\n\t * @param \ttest\n\t *\t\t\t测试字符串\n\t */\n\tprivate static boolean testGlobPattern(String pattern, String test) {\n\t\t// 从缓存取出 PathMatcher, 防止重复初始化\n\t\tPathMatcher pathMatcher = PATH_MATCHER_MAP.getOrDefault(pattern, FileSystems.getDefault().getPathMatcher(\"glob:\" + pattern));\n\t\tPATH_MATCHER_MAP.put(pattern, pathMatcher);\n\t\t\n\t\treturn pathMatcher.matches(Paths.get(test)) || StringUtils.equals(pattern, test);\n\t}\n\t\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/PlaceholderUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.extra.spring.SpringUtil;\nimport lombok.extern.slf4j.Slf4j;\n\nimport java.util.*;\n \n/**\n * 配置文件或模板中的占位符替换工具类\n *\n * @author zhaojun\n */\n@Slf4j\npublic class PlaceholderUtils {\n\n /**\n * Prefix for system property placeholders: \"${\"\n */\n public static final String PLACEHOLDER_PREFIX = \"${\";\n\n /**\n * Suffix for system property placeholders: \"}\"\n */\n public static final String PLACEHOLDER_SUFFIX = \"}\";\n\n\n /**\n * 解析占位符, 将指定的占位符替换为指定的值. 变量值从 Spring 环境中获取, 如没取到, 则默认为空.\n *
\n * 必须在 Spring 环境下使用, 否则会抛出异常.\n *\n *\n * @param formatStr\n * 模板字符串\n *\n * @return 替换后的字符串\n */\n public static String resolvePlaceholdersBySpringProperties(String formatStr) {\n String placeholderName = getFirstPlaceholderName(formatStr);\n if (StringUtils.isEmpty(placeholderName)) {\n return formatStr;\n }\n\n String propertyValue = SpringUtil.getProperty(placeholderName);\n Map map = new HashMap<>();\n map.put(placeholderName, propertyValue);\n return resolvePlaceholders(formatStr, map);\n }\n\n\n /**\n * 解析占位符, 将指定的占位符替换为指定的值.\n *\n * @param formatStr\n * 模板字符串\n *\n * @param parameter\n * 参数列表\n *\n * @return 替换后的字符串\n */\n public static String resolvePlaceholders(String formatStr, Map parameter) {\n if (parameter == null || parameter.isEmpty()) {\n return formatStr;\n }\n StringBuilder sb = new StringBuilder(formatStr);\n int startIndex = sb.indexOf(PLACEHOLDER_PREFIX);\n while (startIndex != -1) {\n int endIndex = sb.indexOf(PLACEHOLDER_SUFFIX, startIndex + PLACEHOLDER_PREFIX.length());\n if (endIndex != -1) {\n String placeholder = sb.substring(startIndex + PLACEHOLDER_PREFIX.length(), endIndex);\n int nextIndex = endIndex + PLACEHOLDER_SUFFIX.length();\n try {\n String propVal = parameter.get(placeholder);\n if (propVal != null) {\n sb.replace(startIndex, endIndex + PLACEHOLDER_SUFFIX.length(), propVal);\n nextIndex = startIndex + propVal.length();\n } else {\n log.warn(\"Could not resolve placeholder '{}' in [{}] \", placeholder, formatStr);\n }\n } catch (Exception ex) {\n log.error(\"Could not resolve placeholder '{}' in [{}]: \", placeholder, formatStr, ex);\n }\n startIndex = sb.indexOf(PLACEHOLDER_PREFIX, nextIndex);\n } else {\n startIndex = -1;\n }\n }\n return sb.toString();\n }\n\n\n /**\n * 获取模板字符串第一个占位符的名称, 如 \"我的名字是: ${name}, 我的年龄是: ${age}\", 返回 \"name\".\n *\n * @param formatStr\n * 模板字符串\n *\n * @return 占位符名称\n */\n public static String getFirstPlaceholderName(String formatStr) {\n List list = getPlaceholderNames(formatStr);\n if (CollectionUtils.isNotEmpty(list)) {\n return list.getFirst();\n }\n return null;\n }\n\n\n /**\n * 获取模板字符串第一个占位符的名称, 如 \"我的名字是: ${name}, 我的年龄是: ${age}\", 返回 [\"name\", \"age].\n *\n * @param formatStr\n * 模板字符串\n *\n * @return 占位符名称\n */\n public static List getPlaceholderNames(String formatStr) {\n if (StringUtils.isEmpty(formatStr)) {\n return Collections.emptyList();\n }\n\n List placeholderNameList = new ArrayList<>();\n\n StringBuilder sb = new StringBuilder(formatStr);\n int startIndex = sb.indexOf(PLACEHOLDER_PREFIX);\n while (startIndex != -1) {\n int endIndex = sb.indexOf(PLACEHOLDER_SUFFIX, startIndex + PLACEHOLDER_PREFIX.length());\n if (endIndex != -1) {\n String placeholder = sb.substring(startIndex + PLACEHOLDER_PREFIX.length(), endIndex);\n int nextIndex = endIndex + PLACEHOLDER_SUFFIX.length();\n startIndex = sb.indexOf(PLACEHOLDER_PREFIX, nextIndex);\n placeholderNameList.add(placeholder);\n } else {\n startIndex = -1;\n }\n }\n return placeholderNameList;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/ProxyDownloadUrlUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.core.convert.Convert;\nimport cn.hutool.core.date.DateUtil;\nimport cn.hutool.core.util.HexUtil;\nimport cn.hutool.crypto.symmetric.SymmetricAlgorithm;\nimport cn.hutool.crypto.symmetric.SymmetricCrypto;\nimport cn.hutool.extra.spring.SpringUtil;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport lombok.extern.slf4j.Slf4j;\n\nimport java.util.HashMap;\nimport java.util.List;\nimport java.util.Map;\n\n/**\n * 代理下载链接工具类\n *\n * @author zhaojun\n */\n@Slf4j\npublic class ProxyDownloadUrlUtils {\n\n\tprivate static SystemConfigService systemConfigService;\n\n\tprivate static final String PROXY_DOWNLOAD_LINK_DELIMITER = \":\";\n\n\tprivate static final Map AES_CACHE = new HashMap<>();\n\n\t/**\n\t * 服务器代理下载 URL 有效期 (秒).\n\t */\n\tpublic static final Integer PROXY_DOWNLOAD_LINK_EFFECTIVE_SECOND = 1800;\n\n\t/**\n\t * 生成签名：根据系统设置中 AES 密钥对生成签名.\n\t *\n\t * @param \tstorageId\n\t * \t\t\t存储源 ID\n\t *\n\t * @param \tpathAndName\n\t * \t\t\t文件路径及文件名称\n\t *\n\t * @param \teffectiveSecond\n\t * \t\t\t有效时间, 单位: 秒\n\t *\n\t * @return\t签名\n\t */\n\tpublic static String generatorSignature(Integer storageId, String pathAndName, Integer effectiveSecond) {\n\t\tif (systemConfigService == null) {\n\t\t\tsystemConfigService = SpringUtil.getBean(SystemConfigService.class);\n\t\t}\n\n\t\t// 如果有效时间为空, 则设置 30 分钟过期\n\t\tif (effectiveSecond == null || effectiveSecond < 1) {\n\t\t\teffectiveSecond = PROXY_DOWNLOAD_LINK_EFFECTIVE_SECOND;\n\t\t}\n\n\t\t// 过期时间的秒数\n\t\tlong second = DateUtil.offsetSecond(DateUtil.date(), effectiveSecond).getTime();\n\t\tString content = storageId + PROXY_DOWNLOAD_LINK_DELIMITER + pathAndName + PROXY_DOWNLOAD_LINK_DELIMITER + second;\n\n\t\tString aesHexKey = systemConfigService.getAesHexKeyOrGenerate();\n\t\tSymmetricCrypto aes = AES_CACHE.computeIfAbsent(aesHexKey, k -> new SymmetricCrypto(SymmetricAlgorithm.AES, HexUtil.decodeHex(k)));\n\n\t\t//加密\n\t\treturn aes.encryptHex(content);\n\t}\n\n\n\tpublic static boolean validSignatureExpired(Integer expectedStorageId, String expectedPathAndName, String signature) {\n\t\tif (systemConfigService == null) {\n\t\t\tsystemConfigService = SpringUtil.getBean(SystemConfigService.class);\n\t\t}\n\n\t\tString aesHexKey = systemConfigService.getAesHexKeyOrGenerate();\n\t\tSymmetricCrypto aes = AES_CACHE.computeIfAbsent(aesHexKey, k -> new SymmetricCrypto(SymmetricAlgorithm.AES, HexUtil.decodeHex(k)));\n\n\t\tlong currentTimeMillis = System.currentTimeMillis();\n\t\t\n\t\tString storageId = null;\n\t\tString pathAndName = null;\n\t\tString expiredSecond = null;\n\t\t\n\t\ttry {\n\t\t\t//解密\n\t\t\tString decryptStr = aes.decryptStr(signature);\n\t\t\tList split = StringUtils.split(decryptStr, PROXY_DOWNLOAD_LINK_DELIMITER);\n\t\t\tstorageId = split.get(0);\n\t\t\tpathAndName = split.get(1);\n\t\t\texpiredSecond = split.get(2);\n\t\t\t\n\t\t\t// 校验存储源 ID 和文件路径及是否过期.\n\t\t\tif (StringUtils.equals(storageId, Convert.toStr(expectedStorageId))\n\t\t\t\t&& StringUtils.equals(StringUtils.concat(pathAndName), StringUtils.concat(expectedPathAndName))\n\t\t\t\t&& currentTimeMillis < Convert.toLong(expiredSecond)) {\n\t\t\t\treturn true;\n\t\t\t}\n\t\t\t\n\t\t\tlog.warn(\"校验链接已过期或不匹配, signature: {}, storageId={}, pathAndName={}, expiredSecond={}, now:={}\", signature, storageId, pathAndName, expiredSecond, currentTimeMillis);\n\t\t} catch (Exception e) {\n\t\t\tlog.error(\"校验签名链接异常, signature: {}, storageId={}, pathAndName={}, expiredSecond={}, now:={}\", signature, storageId, pathAndName, expiredSecond, currentTimeMillis);\n\t\t\treturn false;\n\t\t}\n\n\t\treturn false;\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/RequestHolder.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.extra.servlet.JakartaServletUtil;\nimport im.zhaojun.zfile.core.constant.ZFileHttpHeaderConstant;\nimport im.zhaojun.zfile.core.exception.ErrorCode;\nimport im.zhaojun.zfile.core.exception.core.BizException;\nimport im.zhaojun.zfile.core.exception.core.SystemException;\nimport jakarta.servlet.http.HttpServletRequest;\nimport jakarta.servlet.http.HttpServletResponse;\nimport lombok.extern.slf4j.Slf4j;\nimport org.apache.commons.io.IOUtils;\nimport org.springframework.http.ContentDisposition;\nimport org.springframework.http.HttpHeaders;\nimport org.springframework.http.HttpRange;\nimport org.springframework.http.MediaType;\nimport org.springframework.http.MediaTypeFactory;\nimport org.springframework.util.StreamUtils;\nimport org.springframework.web.context.request.RequestContextHolder;\nimport org.springframework.web.context.request.ServletRequestAttributes;\n\nimport java.io.IOException;\nimport java.io.InputStream;\nimport java.io.OutputStream;\nimport java.nio.charset.StandardCharsets;\nimport java.util.Collections;\nimport java.util.List;\nimport java.util.Objects;\n\n/**\n * 获取 Request 工具类\n *\n * @author zhaojun\n */\n@Slf4j\npublic class RequestHolder {\n\n /**\n * 获取 HttpServletRequest\n *\n * @return HttpServletRequest\n */\n public static HttpServletRequest getRequest() {\n return ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();\n }\n\n\n /**\n * 获取 HttpServletResponse\n *\n * @return HttpServletResponse\n */\n public static HttpServletResponse getResponse() {\n return ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getResponse();\n }\n\n\n /**\n * 向 response 写入文件流.\n *\n * @param inputStream\n * 文件输入流\n *\n * @param fileName\n * 文件名称\n *\n * @param fileSize\n * 文件大小\n *\n * @param isPartialContentFromInputStream\n * 表示输入流是否为部分内容。\n * 当该变量为 true 时，表示输入流已经根据 range 规则从存储源获取部分内容。\n * 在这种情况下，不需要跳过 range start 部分，可以直接从输入流的全部内容复制到输出流。\n *\n * @param forceDownload\n * 是否强制下载\n */\n public static void writeFile(InputStream inputStream, String fileName, Long fileSize, boolean isPartialContentFromInputStream, boolean forceDownload) {\n if (inputStream == null) {\n throw new BizException(ErrorCode.BIZ_FILE_NOT_EXIST);\n }\n OutputStream outputStream = null;\n try (InputStream innerInputStream = inputStream) {\n HttpServletResponse response = RequestHolder.getResponse();\n\n ContentDisposition contentDisposition = ContentDisposition\n .builder(forceDownload ? \"attachment\" : \"inline\")\n .filename(fileName, StandardCharsets.UTF_8)\n .build();\n response.setHeader(HttpHeaders.CONTENT_DISPOSITION, contentDisposition.toString());\n if (forceDownload) {\n response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);\n } else {\n response.setContentType(MediaTypeFactory.getMediaType(fileName).orElse(MediaType.APPLICATION_OCTET_STREAM).toString());\n }\n\n outputStream = response.getOutputStream();\n\n if (fileSize != null && fileSize > 0) {\n String range = RequestHolder.getRequest().getHeader(HttpHeaders.RANGE);\n List httpRanges = HttpRange.parseRanges(range);\n if (httpRanges.isEmpty()) {\n httpRanges = Collections.singletonList(HttpRange.createByteRange(0, fileSize - 1));\n } else {\n response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);\n }\n HttpRange httpRange = CollectionUtils.getFirst(httpRanges);\n long startPos = httpRange.getRangeStart(fileSize);\n long endPos = httpRange.getRangeEnd(fileSize);\n if (response.getStatus() == HttpServletResponse.SC_PARTIAL_CONTENT) {\n response.setHeader(HttpHeaders.CONTENT_RANGE, \"bytes \" + startPos + \"-\" + endPos + StringUtils.SLASH + fileSize);\n }\n\n response.setHeader(HttpHeaders.ACCEPT_RANGES, \"bytes\");\n response.setContentLengthLong(endPos - startPos + 1);\n if (isPartialContentFromInputStream) {\n StreamUtils.copy(innerInputStream, outputStream);\n } else {\n StreamUtils.copyRange(innerInputStream, outputStream, startPos, endPos);\n }\n return;\n }\n\n StreamUtils.copy(innerInputStream, outputStream);\n } catch (IOException e) {\n boolean isBrokenPipe = e.getMessage().contains(\"Broken pipe\");\n boolean isConnectionResetByPeer = e.getMessage().contains(\"Connection reset by peer\");\n if (isBrokenPipe || isConnectionResetByPeer) {\n if (log.isDebugEnabled()) {\n log.debug(\"skip IOException: {}\", e.getMessage());\n }\n } else {\n throw new SystemException(e);\n }\n } finally {\n IOUtils.closeQuietly(inputStream);\n IOUtils.closeQuietly(outputStream);\n }\n }\n\n public static boolean isAxiosRequest() {\n HttpServletRequest request = RequestHolder.getRequest();\n String axiosRequest = JakartaServletUtil.getHeaderIgnoreCase(request, ZFileHttpHeaderConstant.AXIOS_REQUEST);\n return StringUtils.isNotEmpty(axiosRequest);\n }\n\n\n /**\n * 获取请求头中的 Axios-From 字段\n *\n * @return Axios-From 字段\n */\n public static String getAxiosFrom() {\n if (RequestContextHolder.getRequestAttributes() == null) {\n return null;\n }\n HttpServletRequest request = RequestHolder.getRequest();\n return JakartaServletUtil.getHeaderIgnoreCase(request, ZFileHttpHeaderConstant.AXIOS_FROM);\n }\n\n /**\n * 获取后端服务地址，如果经过了反向代理，需反向代理正确配置\n */\n public static String getRequestServerAddress() {\n if (RequestContextHolder.getRequestAttributes() == null) {\n return null;\n }\n HttpServletRequest request = RequestHolder.getRequest();\n\n String forwardedHost = JakartaServletUtil.getHeaderIgnoreCase(request, \"X-Forwarded-Host\");\n String forwardedPort = JakartaServletUtil.getHeaderIgnoreCase(request, \"X-Forwarded-Port\");\n String forwardedProto = JakartaServletUtil.getHeaderIgnoreCase(request, \"X-Forwarded-Proto\");\n\n String scheme = StringUtils.isBlank(forwardedProto) ? request.getScheme() : forwardedProto;\n \n // 优先使用 X-Forwarded-Host，其次使用 Host 头，最后使用 request.getServerName()\n String serverName;\n String hostHeader = StringUtils.isNotBlank(forwardedHost) ? forwardedHost : request.getHeader(\"Host\");\n if (StringUtils.isNotBlank(hostHeader)) {\n // Host 头可能包含端口信息，如 \"example.com:8080\"\n String[] hostParts = hostHeader.split(\":\");\n serverName = hostParts[0];\n // 如果 Host 头包含端口且没有显式设置 X-Forwarded-Port，则使用 Host 头中的端口\n if (hostParts.length > 1 && StringUtils.isBlank(forwardedPort)) {\n forwardedPort = hostParts[1];\n }\n } else {\n serverName = request.getServerName();\n }\n \n // 端口处理逻辑\n String port;\n if (StringUtils.isNotBlank(forwardedPort)) {\n port = forwardedPort;\n } else if (StringUtils.isNotBlank(forwardedProto)) {\n // 如果设置了转发协议但没有设置端口，使用协议默认端口\n port = \"https\".equalsIgnoreCase(forwardedProto) ? \"443\" : \"80\";\n } else {\n port = String.valueOf(request.getServerPort());\n }\n\n // 移除默认端口\n if (\"443\".equals(port) && \"https\".equalsIgnoreCase(scheme)) {\n port = \"\";\n }\n if (\"80\".equals(port) && \"http\".equalsIgnoreCase(scheme)) {\n port = \"\";\n }\n\n if (StringUtils.isBlank(port)) {\n return scheme + \"://\" + serverName;\n } else {\n return scheme + \"://\" + serverName + \":\" + port;\n }\n }\n\n\n /**\n * 获取当前请求的 Origin 请求头\n *\n * @return Origin 请求头值\n */\n public static String getOriginAddress() {\n if (RequestContextHolder.getRequestAttributes() == null) {\n return null;\n }\n HttpServletRequest request = RequestHolder.getRequest();\n return JakartaServletUtil.getHeaderIgnoreCase(request, HttpHeaders.ORIGIN);\n }\n\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/RequestUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport jakarta.servlet.http.HttpServletRequest;\nimport org.springframework.http.HttpHeaders;\nimport org.springframework.http.HttpRange;\nimport org.springframework.util.CollectionUtils;\n\npublic class RequestUtils {\n\n public static HttpRange getRequestRange(HttpServletRequest request) {\n String rangeHeader = request.getHeader(HttpHeaders.RANGE);\n if (rangeHeader == null) {\n return null;\n }\n return CollectionUtils.firstElement(HttpRange.parseRanges(rangeHeader));\n }\n\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/SizeToStrUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.core.util.NumberUtil;\n\n/**\n * 文件大小或带宽大小转可读单位\n *\n * @author zhaojun\n */\npublic class SizeToStrUtils {\n\n /**\n * 将文件大小转换为可读单位\n *\n * @param bytes\n * 字节数\n *\n * @return 文件大小可读单位\n */\n public static String bytesToSize(long bytes) {\n if (bytes == 0) {\n return \"0\";\n }\n\n double k = 1024;\n String[] sizes = new String[]{\"B\", \"KB\", \"MB\", \"GB\", \"TB\", \"PB\", \"EB\", \"ZB\", \"YB\"};\n double i = Math.floor(Math.log(bytes) / Math.log(k));\n return NumberUtil.round(bytes / Math.pow(k, i), 3) + \" \" + sizes[(int) i];\n }\n\n\n /**\n * 将带宽大小转换为可读单位\n *\n * @param bps\n * 字节数\n *\n * @return 带宽大小可读单位\n */\n public static String bpsToSize(long bps) {\n if (bps == 0) {\n return \"0\";\n }\n\n double k = 1000;\n String[] sizes = new String[]{\"B\", \"KB\", \"MB\", \"GB\", \"TB\", \"PB\", \"EB\", \"ZB\", \"YB\"};\n double i = Math.floor(Math.log(bps) / Math.log(k));\n return NumberUtil.round(bps / Math.pow(k, i), 3) + \" \" + sizes[(int) i];\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/SpringMvcUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport org.springframework.util.AntPathMatcher;\nimport org.springframework.web.servlet.HandlerMapping;\n\nimport jakarta.servlet.http.HttpServletRequest;\n\n/**\n * @author zhaojun\n */\npublic class SpringMvcUtils {\n\n public static String getExtractPathWithinPattern() {\n HttpServletRequest httpServletRequest = RequestHolder.getRequest();\n String path = (String) httpServletRequest.getAttribute(HandlerMapping.PATH_WITHIN_HANDLER_MAPPING_ATTRIBUTE);\n String bestMatchPattern = (String) httpServletRequest.getAttribute(HandlerMapping.BEST_MATCHING_PATTERN_ATTRIBUTE);\n AntPathMatcher apm = new AntPathMatcher();\n return apm.extractPathWithinPattern(bestMatchPattern, path);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/StrPool.java",
"content": "package im.zhaojun.zfile.core.util;\n\npublic interface StrPool {\n\n /**\n * 字符串常量：制表符 {@code \"\\t\"}\n */\n String TAB = \"\t\";\n\n /**\n * 字符串常量：点 {@code \".\"}\n */\n String DOT = \".\";\n\n /**\n * 字符串常量：双点 {@code \"..\"}
\n * 用途：作为指向上级文件夹的路径，如：{@code \"../path\"}\n */\n String DOUBLE_DOT = \"..\";\n\n /**\n * 字符串常量：斜杠 {@code \"/\"}\n */\n String SLASH = \"/\";\n\n /**\n * 字符串常量：反斜杠 {@code \"\\\\\"}\n */\n String BACKSLASH = \"\\\\\";\n\n /**\n * 字符串常量：回车符 {@code \"\\r\"}
\n * 解释：该字符常用于表示 Linux 系统和 MacOS 系统下的文本换行\n */\n String CR = \"\\r\";\n\n /**\n * 字符串常量：换行符 {@code \"\\n\"}\n */\n String LF = \"\\n\";\n\n /**\n * 字符串常量：Windows 换行 {@code \"\\r\\n\"}
\n * 解释：该字符串常用于表示 Windows 系统下的文本换行\n */\n String CRLF = \"\\r\\n\";\n\n /**\n * 字符串常量：下划线 {@code \"_\"}\n */\n String UNDERLINE = \"_\";\n\n /**\n * 字符串常量：减号（连接符） {@code \"-\"}\n */\n String DASHED = \"-\";\n\n /**\n * 字符串常量：逗号 {@code \",\"}\n */\n String COMMA = \",\";\n\n /**\n * 字符串常量：花括号（左） \"{\"\n */\n String DELIM_START = \"{\";\n\n /**\n * 字符串常量：花括号（右） \"}\"\n */\n String DELIM_END = \"}\";\n\n /**\n * 字符串常量：中括号（左） {@code \"[\"}\n */\n String BRACKET_START = \"[\";\n\n /**\n * 字符串常量：中括号（右） {@code \"]\"}\n */\n String BRACKET_END = \"]\";\n\n /**\n * 字符串常量：冒号 {@code \":\"}\n */\n String COLON = \":\";\n\n /**\n * 字符串常量：艾特 {@code \"@\"}\n */\n String AT = \"@\";\n\n /**\n * 字符串常量：空 JSON {@code \"{}\"}\n */\n String EMPTY_JSON = \"{}\";\n\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/StringUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.cache.CacheUtil;\nimport cn.hutool.cache.impl.LRUCache;\nimport cn.hutool.core.net.URLEncodeUtil;\nimport cn.hutool.core.util.StrUtil;\nimport cn.hutool.core.util.URLUtil;\n\nimport java.net.MalformedURLException;\nimport java.net.URL;\nimport java.nio.charset.StandardCharsets;\n\n/**\n * 字符串相关工具类\n *\n * @author zhaojun\n */\npublic class StringUtils extends CharSequenceUtil implements StrPool {\n\n public static final String HTTP = \"http\";\n\n public static final String PROTOCOL_MARKER = \"://\";\n\n private static final LRUCache CACHE = CacheUtil.newLRUCache(1000);\n\n /**\n * 移除 URL 中的前后的所有 '/'\n *\n * @param path\n * 路径\n *\n * @return 如 path = '/folder1/file1/', 返回 'folder1/file1'\n * 如 path = '///folder1/file1//', 返回 'folder1/file1'\n */\n public static String trimSlashes(String path) {\n path = trimStartSlashes(path);\n path = trimEndSlashes(path);\n return path;\n }\n\n\n /**\n * 移除 URL 中的前面的所有 '/'\n *\n * @param path\n * 路径\n *\n * @return 如 path = '/folder1/file1', 返回 'folder1/file1'\n * 如 path = '//folder1/file1', 返回 'folder1/file1'\n *\n */\n public static String trimStartSlashes(String path) {\n if (isEmpty(path)) {\n return path;\n }\n\n while (path.startsWith(SLASH)) {\n path = path.substring(1);\n }\n\n return path;\n }\n\n\n /**\n * 移除 URL 中结尾的所有 '/'\n *\n * @param path\n * 路径\n *\n * @return 如 path = '/folder1/file1/', 返回 '/folder1/file1'\n * 如 path = '/folder1/file1///', 返回 '/folder1/file1'\n */\n public static String trimEndSlashes(String path) {\n if (isEmpty(path)) {\n return path;\n }\n\n while (path.endsWith(SLASH)) {\n path = path.substring(0, path.length() - 1);\n }\n\n return path;\n }\n\n\n /**\n * 去除路径中所有重复的 '/'，如果最开始的协议头前有 / 也一并去除。\n *\n * @param path\n * 路径\n *\n * @return 如 path = '/folder1//file1/', 返回 '/folder1/file1/'\n * 如 path = '/folder1////file1///', 返回 '/folder1/file1/'\n */\n public static String removeDuplicateSlashes(String path) {\n if (isEmpty(path)) {\n return path;\n }\n\n return CACHE.get(path, false, () -> {\n StringBuilder sb = new StringBuilder(path.length());\n int protocolIndex = path.indexOf(PROTOCOL_MARKER);\n\n int pathStartIndex = 0;\n\n // 1. 处理协议部分\n if (protocolIndex > -1) {\n // 找到协议名称的实际开始位置\n int schemeStartIndex = 0;\n while (schemeStartIndex < protocolIndex && path.charAt(schemeStartIndex) == '/') {\n schemeStartIndex++;\n }\n\n sb.append(path, schemeStartIndex, protocolIndex);\n sb.append(PROTOCOL_MARKER);\n\n pathStartIndex = protocolIndex + PROTOCOL_MARKER.length();\n }\n\n if (pathStartIndex < path.length()) {\n char lastChar;\n char firstPathChar = path.charAt(pathStartIndex);\n sb.append(firstPathChar);\n lastChar = firstPathChar;\n\n for (int i = pathStartIndex + 1; i < path.length(); i++) {\n char current = path.charAt(i);\n if (current != SLASH_CHAR || lastChar != SLASH_CHAR) {\n sb.append(current);\n lastChar = current;\n }\n }\n }\n\n return sb.toString();\n });\n }\n\n\n /**\n * 去除路径中所有重复的 '/', 并且去除开头的 '/'\n *\n * @param path\n * 路径\n *\n * @return 如 path = '/folder1//file1/', 返回 'folder1/file1/'\n * 如 path = '///folder1////file1///', 返回 'folder1/file1/'\n */\n public static String removeDuplicateSlashesAndTrimStart(String path) {\n path = removeDuplicateSlashes(path);\n path = trimStartSlashes(path);\n return path;\n }\n\n\n /**\n * 去除路径中所有重复的 '/', 并且去除结尾的 '/'\n *\n * @param path\n * 路径\n *\n * @return 如 path = '/folder1//file1/', 返回 '/folder1/file1'\n * 如 path = '///folder1////file1///', 返回 '/folder1/file1'\n */\n public static String removeDuplicateSlashesAndTrimEnd(String path) {\n path = removeDuplicateSlashes(path);\n path = trimEndSlashes(path);\n return path;\n }\n\n\n /**\n * 拼接 URL，并去除重复的分隔符 '/'，并去除开头的 '/', 但不会影响 http:// 和 https:// 这种头部.\n *\n * @param strs\n * 拼接的字符数组\n *\n * @return 拼接结果\n */\n public static String concatTrimStartSlashes(String... strs) {\n return trimStartSlashes(concat(strs));\n }\n\n\n /**\n * 拼接 URL，并去除重复的分隔符 '/'，并去除结尾的 '/', 但不会影响 http:// 和 https:// 这种头部.\n *\n * @param strs\n * 拼接的字符数组\n *\n * @return 拼接结果\n */\n public static String concatTrimEndSlashes(String... strs) {\n return trimEndSlashes(concat(strs));\n }\n\n\n /**\n * 拼接 URL，并去除重复的分隔符 '/'，并去除开头和结尾的 '/', 但不会影响 http:// 和 https:// 这种头部.\n *\n * @param strs\n * 拼接的字符数组\n *\n * @return 拼接结果\n */\n public static String concatTrimSlashes(String... strs) {\n return trimSlashes(concat(strs));\n }\n\n\n /**\n * 拼接 URL，并去除重复的分隔符 '/'，但不会影响 http:// 和 https:// 这种头部.\n *\n * @param strs\n * 拼接的字符数组\n *\n * @return 拼接结果\n */\n public static String concat(String... strs) {\n StringBuilder sb = new StringBuilder(SLASH);\n for (int i = 0; i < strs.length; i++) {\n String str = strs[i];\n if (isEmpty(str)) {\n continue;\n }\n sb.append(str);\n if (i != strs.length - 1) {\n sb.append(SLASH_CHAR);\n }\n }\n return removeDuplicateSlashes(sb.toString());\n }\n\n\n /**\n * 拼接 URL，并去除重复的分隔符 '/'，但不会影响 http:// 和 https:// 这种头部.\n *\n * @param encodeAllIgnoreSlashes\n * 是否 encode 编码 (忽略 /)\n *\n * @param strs\n * 拼接的字符数组\n *\n * @return 拼接结果\n */\n public static String concat(boolean encodeAllIgnoreSlashes, String... strs) {\n String res = concat(strs);\n if (encodeAllIgnoreSlashes) {\n return encodeAllIgnoreSlashes(res);\n } else {\n return res;\n }\n }\n\n\n /**\n * 替换 URL 中的 Host 部分，如替换 http://a.com/1.txt 为 https://abc.com/1.txt\n *\n * @param originUrl\n * 原 URL\n *\n * @param replaceHost\n * 替换的 HOST\n *\n * @return 替换后的 URL\n */\n public static String replaceHost(String originUrl, String replaceHost) {\n try {\n String path = new URL(originUrl).getFile();\n return concat(replaceHost, path);\n } catch (MalformedURLException e) {\n e.printStackTrace();\n }\n return null;\n }\n\n\n /**\n * 编码 URL，默认使用 UTF-8 编码\n * URL 的 Fragment URLEncoder\n * 默认的编码器针对Fragment，定义如下：\n *\n * \n * fragment = *( pchar / \"/\" / \"?\" )\n * pchar = unreserved / pct-encoded / sub-delims / \":\" / \"@\"\n * unreserved = ALPHA / DIGIT / \"-\" / \".\" / \"_\" / \"~\"\n * sub-delims = \"!\" / \"$\" / \"&\" / \"'\" / \"(\" / \")\" / \"*\" / \"+\" / \",\" / \";\" / \"=\"\n *
\n *\n * 具体见：https://datatracker.ietf.org/doc/html/rfc3986#section-3.5\n *\n * @param url\n * 被编码内容\n *\n * @return 编码后的字符\n */\n public static String encode(String url) {\n return URLEncodeUtil.encodeFragment(url);\n }\n\n\n /**\n * 编码全部字符\n *\n * @param str\n * 被编码内容\n *\n * @return 编码后的字符\n */\n public static String encodeAllIgnoreSlashes(String str) {\n if (isEmpty(str)) {\n return str;\n }\n\n StringBuilder sb = new StringBuilder();\n\n int prevIndex = -1;\n for (int i = 0; i < str.length(); i++) {\n char c = str.charAt(i);\n if (c == StringUtils.SLASH_CHAR) {\n if (prevIndex < i) {\n String substring = str.substring(prevIndex + 1, i);\n sb.append(URLEncodeUtil.encodeAll(substring));\n prevIndex = i;\n }\n sb.append(c);\n }\n\n if (i == str.length() - 1 && prevIndex < i) {\n String substring = str.substring(prevIndex + 1, i + 1);\n sb.append(URLEncodeUtil.encodeAll(substring));\n }\n }\n\n return sb.toString();\n }\n\n\n /**\n * 解码 URL, 默认使用 UTF8 编码. 不会将 + 转为空格.\n *\n * @param url\n * 被解码内容\n *\n * @return 解码后的内容\n */\n public static String decode(String url) {\n return URLUtil.decode(url, StandardCharsets.UTF_8, false);\n }\n\n\n /**\n * 移除字符串中所有换行符并去除前后空格\n *\n * @param str\n * URL\n *\n * @return 移除协议后的 URL\n */\n public static String removeAllLineBreaksAndTrim(String str) {\n String removeResult = StrUtil.removeAllLineBreaks(str);\n return trim(removeResult);\n }\n\n\n /**\n * 移除字符串前后空格\n *\n * @param str\n * 字符串\n *\n * @return 移除前后空格后的字符串\n */\n public static String trim(final String str) {\n return str == null ? null : str.trim();\n }\n\n\n /**\n * 如果给定字符串不是以suffix结尾的，在尾部补充 suffix\n *\n * @param str 字符串\n * @param suffix 后缀\n * @return 补充后的字符串\n */\n public static String addSuffixIfNot(CharSequence str, CharSequence suffix) {\n if (isEmpty(str) || isEmpty(suffix)) {\n return str.toString();\n }\n\n if (str.toString().endsWith(suffix.toString())) {\n return str.toString();\n }\n\n return str.toString() + suffix;\n }\n\n /**\n * 是否包含特定字符，忽略大小写，如果给定两个参数都为{@code null}，返回true\n *\n * @param str 被检测字符串\n * @param testStr 被测试是否包含的字符串\n * @return 是否包含\n */\n public static boolean containsIgnoreCase(CharSequence str, CharSequence testStr) {\n if (null == str) {\n // 如果被监测字符串和\n return null == testStr;\n }\n return StrUtil.indexOfIgnoreCase(str, testStr) > -1;\n }\n\n\n /**\n * 指定范围内查找指定字符\n *\n * @param str\n * 字符串\n *\n * @param searchChar\n * 被查找的字符\n *\n * @return 位置\n */\n public static int indexOf(String str, char searchChar) {\n if (isEmpty(str)) {\n return INDEX_NOT_FOUND;\n }\n return str.indexOf(searchChar);\n }\n\n\n /**\n * 字符串驼峰转下划线格式\n *\n * @param param\n * 驼峰格式字符串\n *\n * @return 下划线格式字符串\n */\n public static String camelToUnderline(String param) {\n if (isEmpty(param)) {\n return EMPTY;\n }\n\n int len = param.length();\n StringBuilder sb = new StringBuilder(len);\n for (int i = 0; i < len; i++) {\n char c = param.charAt(i);\n if (Character.isUpperCase(c) && i > 0) {\n sb.append(UNDERLINE);\n }\n sb.append(Character.toLowerCase(c));\n }\n return sb.toString();\n }\n\n\n /**\n * 强制给 URL 设置协议\n *\n * @param url\n * URL 地址，可以是带协议的，也可以是不带协议的，写会忽略大小写\n *\n * @param schema\n * 协议，如 http, https, http://, https://\n *\n * @return 设置协议后的 URL\n */\n public static String setSchema(String url, String schema) {\n if (StringUtils.isEmpty(url) || StringUtils.isEmpty(schema)) {\n return url;\n }\n\n if (!schema.endsWith(\"://\")) {\n schema += \"://\";\n }\n\n String lowerUrl = url.toLowerCase();\n if (lowerUrl.startsWith(\"http://\")) {\n url = url.substring(7);\n } else if (lowerUrl.startsWith(\"https://\")) {\n url = url.substring(8);\n }\n\n return schema + url;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/UrlUtils.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.hutool.core.util.StrUtil;\n\n/**\n * url 相关工具类\n *\n * @author zhaojun\n */\npublic class UrlUtils {\n\n\t/**\n\t * 判断 URL 是否包含协议部分\n\t *\n\t * @param \turl\n\t * \t\t\tURL 地址\n\t *\n\t * @return\t是否包含协议部分\n\t */\n\tpublic static boolean hasScheme(String url) {\n\t\treturn url.startsWith(\"http://\") || url.startsWith(\"https://\");\n\t}\n\n\t/**\n\t * 为 URL 拼接参数\n\t *\n\t * @param \turl\n\t * \t\t\t原始 URL\n\t *\n\t * @param \tname\n\t * \t\t\t参数名称\n\t *\n\t * @param \tvalue\n\t * \t\t\t参数值\n\t *\n\t * @return\t拼接后的 URL\n\t */\n\tpublic static String concatQueryParam(String url, String name, String value) {\n\t\tif (StringUtils.contains(url, \"?\")) {\n\t\t\treturn url + \"&\" + name + \"=\" + value;\n\t\t} else {\n\t\t\treturn url + \"?\" + name + \"=\" + value;\n\t\t}\n\t}\n\n\t/**\n\t * 获取 URL 中的协议部分\n\t *\n\t * @param \turl\n\t * \t\t\tURL 地址\n\t *\n\t * @return\t协议部分\n\t */\n\tpublic static String getSchema(String url) {\n\t\tif (StringUtils.startWithIgnoreCase(url, \"http://\")) {\n\t\t\treturn \"http\";\n\t\t} else if (StringUtils.startWithIgnoreCase(url, \"https://\")) {\n\t\t\treturn \"https\";\n\t\t} else {\n\t\t\treturn \"http\";\n\t\t}\n\t}\n\n\t/**\n\t * 移除 URL 中的协议部分\n\t *\n\t * @param \turl\n\t * \t\t\tURL 地址\n\t *\n\t * @return\t移除协议部分后的 URL\n\t */\n\tpublic static String removeScheme(String url) {\n\t\tif (StringUtils.startWithIgnoreCase(url, \"http://\")) {\n\t\t\turl = url.substring(7);\n\t\t} else if (StringUtils.startWithIgnoreCase(url, \"https://\")) {\n\t\t\turl = url.substring(8);\n\t\t}\n\n\t\treturn url;\n\t}\n\n\t/**\n\t * 获取 URL 中的域名部分\n\t *\n\t * @param \turl\n\t * \t\t\tURL 地址\n\t *\n\t * @return\t域名部分\n\t */\n\tpublic static String getDomain(String url) {\n\t\tif (!StringUtils.isEmpty(url)) {\n\t\t\t//替换指定前缀\n\t\t\tString newStr = url.replace(\"http://\", \"\");\n\t\t\tnewStr = newStr.replace(\"https://\", \"\");\n\n\t\t\tint index = StrUtil.indexOf(newStr, '/');\n\t\t\tif (index > 0) {\n\t\t\t\tnewStr = newStr.substring(0, index);\n\t\t\t}\n\n\t\t\tString[] split = newStr.split(\":\");\n\t\t\tif (split.length > 1) {\n\t\t\t\treturn split[0];\n\t\t\t} else {\n\t\t\t\treturn newStr;\n\t\t\t}\n\t\t} else {\n\t\t\treturn null;\n\t\t}\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/ZFileAuthUtil.java",
"content": "package im.zhaojun.zfile.core.util;\n\nimport cn.dev33.satoken.stp.StpUtil;\nimport cn.hutool.extra.spring.SpringUtil;\nimport im.zhaojun.zfile.module.share.context.ShareAccessContext;\nimport im.zhaojun.zfile.module.user.model.constant.UserConstant;\nimport im.zhaojun.zfile.module.user.model.entity.User;\nimport im.zhaojun.zfile.module.user.service.UserService;\n\n/**\n * 登录认证工具类\n *\n * @author zhaojun\n */\npublic class ZFileAuthUtil {\n\t\n\tprivate static UserService userService;\n\n\tpublic static User getCurrentUser() {\n\t\tif (userService == null) {\n\t\t\tuserService = SpringUtil.getBean(UserService.class);\n\t\t}\n\n // 检查是否为分享访问，如果是则返回分享者用户 ID\n if (ShareAccessContext.isShareAccess()) {\n Integer shareUserId = ShareAccessContext.getShareUserId();\n return userService.getById(shareUserId);\n }\n\n return userService.getById(StpUtil.getLoginId(UserConstant.ANONYMOUS_ID));\n\t}\n\n\tpublic static Integer getCurrentUserId() {\n\t\tif (userService == null) {\n\t\t\tuserService = SpringUtil.getBean(UserService.class);\n\t\t}\n\n // 检查是否为分享访问，如果是则返回分享者用户 ID\n if (ShareAccessContext.isShareAccess()) {\n return ShareAccessContext.getShareUserId();\n }\n\n try {\n\t\t\treturn StpUtil.getLoginId(UserConstant.ANONYMOUS_ID);\n\t\t} catch (Exception e) {\n\t\t\treturn UserConstant.ANONYMOUS_ID;\n\t\t}\n\t}\n\t\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/matcher/AbstractRuleMatcher.java",
"content": "package im.zhaojun.zfile.core.util.matcher;\n\nimport java.util.Collection;\n\n/**\n * 抽象规则匹配器, 实现了部分方法, 用于简化规则匹配器的实现.\n *\n * @author zhaojun\n */\npublic abstract class AbstractRuleMatcher implements IRuleMatcher {\n\n @Override\n public boolean contains(String ruleExpression, String testStr) {\n return match(ruleExpression, testStr);\n }\n\n @Override\n public boolean matchAny(Collection ruleExpressionList, String testStr) {\n if (ruleExpressionList == null || ruleExpressionList.isEmpty()) {\n return false;\n }\n for (String ruleExpression : ruleExpressionList) {\n if (match(ruleExpression, testStr)) {\n return true;\n }\n }\n return false;\n }\n\n @Override\n public String matchAnyReturnFirst(Collection ruleExpressionList, String testStr) {\n if (ruleExpressionList == null || ruleExpressionList.isEmpty()) {\n return null;\n }\n for (String ruleExpression : ruleExpressionList) {\n if (match(ruleExpression, testStr)) {\n return ruleExpression;\n }\n }\n return null;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/matcher/IRuleMatcher.java",
"content": "package im.zhaojun.zfile.core.util.matcher;\n\nimport java.util.Collection;\n\n/**\n * 规则匹配器接口\n *\n * @author zhaojun\n */\npublic interface IRuleMatcher {\n\n /**\n * 匹配规则\n *\n * @param ruleExpression\n * 规则表达式\n *\n * @param testStr\n * 测试字符串\n *\n * @return 是否匹配\n */\n boolean match(String ruleExpression, String testStr);\n\n /**\n * 部分匹配规则\n *\n * @param ruleExpression\n * 规则表达式\n *\n * @param testStr\n * 测试字符串\n *\n * @return 是否部分匹配\n */\n boolean contains(String ruleExpression, String testStr);\n\n\n /**\n * 匹配规则, 可以匹配多个规则表达式, 只要有一个匹配成功, 则返回 true\n *\n * @param ruleExpressionList\n * 规则表达式列表\n *\n * @param testStr\n * 测试字符串\n *\n * @return 是否匹配\n */\n boolean matchAny(Collection ruleExpressionList, String testStr);\n\n\n /**\n * 匹配规则, 可以匹配多个规则表达式, 只要有一个匹配成功, 则返回第一个匹配成功的表达式。\n *\n * @param ruleExpressionList\n * 规则表达式列表\n *\n * @param testStr\n * 测试字符串\n *\n * @return 匹配成功的第一个表达式\n */\n String matchAnyReturnFirst(Collection ruleExpressionList, String testStr);\n\n\n /**\n * 获取规则类型\n *\n * @return 规则类型\n */\n String getRuleType();\n\n\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/matcher/RuleMatcherFactory.java",
"content": "package im.zhaojun.zfile.core.util.matcher;\n\nimport im.zhaojun.zfile.core.util.matcher.impl.AntPathRuleMatcher;\nimport im.zhaojun.zfile.core.util.matcher.impl.IpRuleMatcher;\nimport im.zhaojun.zfile.core.util.matcher.impl.RegexRuleMatcher;\nimport im.zhaojun.zfile.core.util.matcher.impl.SpringSimpleRuleMatcher;\n\nimport java.util.HashMap;\nimport java.util.Map;\n\n/**\n * 规则匹配器工厂, 用于获取规则匹配器实例.\n *\n * @author zhaojun\n */\npublic class RuleMatcherFactory {\n\n private static final Map RULE_MATCHER_MAP = new HashMap<>();\n\n static {\n IpRuleMatcher ipRuleMatcher = new IpRuleMatcher();\n RULE_MATCHER_MAP.put(ipRuleMatcher.getRuleType(), ipRuleMatcher);\n\n RegexRuleMatcher regexRuleMatcher = new RegexRuleMatcher();\n RULE_MATCHER_MAP.put(regexRuleMatcher.getRuleType(), regexRuleMatcher);\n\n AntPathRuleMatcher antPathRuleMatcher = new AntPathRuleMatcher();\n RULE_MATCHER_MAP.put(antPathRuleMatcher.getRuleType(), antPathRuleMatcher);\n\n SpringSimpleRuleMatcher springSimpleRuleMatcher = new SpringSimpleRuleMatcher();\n RULE_MATCHER_MAP.put(springSimpleRuleMatcher.getRuleType(), springSimpleRuleMatcher);\n }\n\n public static IRuleMatcher getRuleMatcher(String ruleType) {\n if (ruleType == null || ruleType.isEmpty()) {\n return null;\n }\n\n return RULE_MATCHER_MAP.get(ruleType);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/matcher/impl/AntPathRuleMatcher.java",
"content": "package im.zhaojun.zfile.core.util.matcher.impl;\n\nimport im.zhaojun.zfile.core.constant.RuleTypeConstant;\nimport im.zhaojun.zfile.core.util.matcher.AbstractRuleMatcher;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.util.AntPathMatcher;\n\n/**\n * Ant 路径匹配器, 用于匹配路径规则.\n *\n * @author zhaojun\n */\n@Slf4j\npublic class AntPathRuleMatcher extends AbstractRuleMatcher {\n\n private final AntPathMatcher pathMatcher = new AntPathMatcher();\n\n @Override\n public boolean match(String ruleExpression, String testStr) {\n boolean match = pathMatcher.match(ruleExpression, testStr);\n if (log.isDebugEnabled()) {\n log.debug(\"Ant 表达式匹配结果: {}, 规则表达式: {}, 测试值: {}\", match, ruleExpression, testStr);\n }\n return match;\n }\n\n @Override\n public String getRuleType() {\n return RuleTypeConstant.ANT_PATH;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/matcher/impl/IpRuleMatcher.java",
"content": "package im.zhaojun.zfile.core.util.matcher.impl;\n\nimport im.zhaojun.zfile.core.constant.RuleTypeConstant;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.core.util.matcher.AbstractRuleMatcher;\nimport lombok.extern.slf4j.Slf4j;\n\nimport java.net.Inet6Address;\nimport java.net.InetAddress;\nimport java.net.UnknownHostException;\nimport java.util.regex.Pattern;\n\n/**\n * IP 匹配器, 用于 IP 规则，支持匹配完整 IP 或 IP 段，同时支持 IPV4 和 IPV6.
\n *\n *\n * \n * - IPV4 示例：
\n * \n * - 127.0.0.1
\n * - 192.168.0.0/24
\n * \n *
\n *\n * \n *\n *
\n * - IPV6 示例：
\n * \n * - 0:0:0:0:0:0:0:1
\n * - 0:0:0:0:0:0:0:0/64
\n * \n *
\n * \n *\n * @author zhaojun\n */\n@Slf4j\npublic class IpRuleMatcher extends AbstractRuleMatcher {\n\n @Override\n public boolean match(String ruleExpression, String testStr) {\n try {\n InetAddress inetAddress = InetAddress.getByName(testStr);\n IpRule rule = createRule(ruleExpression);\n boolean match = rule != null && rule.matches(inetAddress);\n if (log.isDebugEnabled()) {\n log.debug(\"IP 匹配结果: {}, 规则表达式: {}, 测试值: {}, 校验规则: {}\", match, ruleExpression, testStr, rule);\n }\n return match;\n } catch (UnknownHostException e) {\n log.error(\"IP 地址解析失败, ruleExpression: {}, testStr: {}\", ruleExpression, testStr);\n }\n return false;\n }\n\n @Override\n public String getRuleType() {\n return RuleTypeConstant.IP;\n }\n\n private IpRule createRule(String ruleExpression) {\n if (isValidIpv4(ruleExpression)) {\n return new Ipv4Rule(ruleExpression);\n } else if (isValidIpv6(ruleExpression)) {\n return new Ipv6Rule(ruleExpression);\n } else if (isValidIpv4Range(ruleExpression)) {\n return new Ipv4RangeRule(ruleExpression);\n } else if (isValidIpv6Range(ruleExpression)) {\n return new Ipv6RangeRule(ruleExpression);\n } else {\n return null;\n }\n }\n\n private boolean isValidIpv4(String ipAddress) {\n String ipv4Pattern = \"^([01]?\\\\d\\\\d?|2[0-4]\\\\d|25[0-5])\\\\.\"\n + \"([01]?\\\\d\\\\d?|2[0-4]\\\\d|25[0-5])\\\\.\"\n + \"([01]?\\\\d\\\\d?|2[0-4]\\\\d|25[0-5])\\\\.\"\n + \"([01]?\\\\d\\\\d?|2[0-4]\\\\d|25[0-5])$\";\n return Pattern.matches(ipv4Pattern, ipAddress);\n }\n\n private boolean isValidIpv6(String ipAddress) {\n try {\n InetAddress.getByName(ipAddress);\n return ipAddress.contains(\":\");\n } catch (UnknownHostException e) {\n return false;\n }\n }\n\n private boolean isValidIpv4Range(String ipRange) {\n String ipv4RangePattern = \"^\\\\d{1,3}\\\\.\\\\d{1,3}\\\\.\\\\d{1,3}\\\\.\\\\d{1,3}/\\\\d{1,2}$\";\n return Pattern.matches(ipv4RangePattern, ipRange);\n }\n\n private boolean isValidIpv6Range(String ipRange) {\n String ipv6RangePattern = \"^([0-9A-Fa-f]{0,4}:){1,7}([0-9A-Fa-f]{0,4})?/[0-9]{1,3}$\";\n return Pattern.matches(ipv6RangePattern, ipRange);\n }\n \n private interface IpRule {\n\n boolean matches(InetAddress ipAddress);\n\n String getExpression();\n\n }\n\n private static class Ipv4Rule implements IpRule {\n \n private final String expression;\n\n Ipv4Rule(String expression) {\n this.expression = expression;\n }\n\n @Override\n public boolean matches(InetAddress ipAddress) {\n if (ipAddress instanceof Inet6Address) {\n return false;\n }\n return ipAddress.getHostAddress().equals(expression);\n }\n\n @Override\n public String getExpression() {\n return expression;\n }\n }\n\n private static class Ipv6Rule implements IpRule {\n \n private final String expression;\n\n Ipv6Rule(String expression) {\n this.expression = expression;\n }\n\n @Override\n public boolean matches(InetAddress ipAddress) {\n if (ipAddress instanceof Inet6Address) {\n return ipAddress.getHostAddress().equals(expression);\n }\n return false;\n }\n\n @Override\n public String getExpression() {\n return expression;\n }\n }\n\n private static class Ipv4RangeRule implements IpRule {\n \n private final String expression;\n private final int prefixLength;\n\n Ipv4RangeRule(String expression) {\n this.expression = expression.substring(0, expression.indexOf('/'));\n this.prefixLength = Integer.parseInt(expression.substring(expression.indexOf('/') + 1));\n }\n\n @Override\n public boolean matches(InetAddress ipAddress) {\n if (ipAddress instanceof Inet6Address) {\n return false;\n }\n String[] rangeParts = expression.split(\"\\\\.\");\n byte[] rangeAddrBytes = new byte[4];\n for (int i = 0; i < rangeParts.length; i++) {\n rangeAddrBytes[i] = (byte) Integer.parseInt(rangeParts[i]);\n }\n\n byte[] ipValue = ipAddress.getAddress();\n if (ipValue.length != 4) {\n return false;\n }\n\n for (int i = 0; i < prefixLength / 8; i++) {\n if (rangeAddrBytes[i] != ipValue[i]) {\n return false;\n }\n }\n\n int remainingBits = prefixLength % 8;\n if (remainingBits != 0) {\n int rangeByte = rangeAddrBytes[prefixLength / 8];\n int ipByte = ipValue[prefixLength / 8];\n\n int shift = 8 - remainingBits;\n int mask = 0xFF >> shift;\n\n return (rangeByte >> shift & mask) == (ipByte >> shift & mask);\n }\n\n return true;\n }\n\n @Override\n public String getExpression() {\n return expression + StringUtils.SLASH + prefixLength;\n }\n }\n\n private static class Ipv6RangeRule implements IpRule {\n \n private final String expression;\n private final int prefixLength;\n\n Ipv6RangeRule(String expression) {\n this.expression = expression.substring(0, expression.indexOf('/'));\n this.prefixLength = Integer.parseInt(expression.substring(expression.indexOf('/') + 1));\n }\n\n @Override\n public boolean matches(InetAddress ipAddress) {\n if (ipAddress instanceof Inet6Address) {\n byte[] ipValue = ipAddress.getAddress();\n byte[] rangeValue = Inet6AddressConverter.convert(expression);\n\n if (ipValue.length != 16) {\n return false;\n }\n\n for (int i = 0; i < prefixLength / 8; i++) {\n if (rangeValue[i] != ipValue[i]) {\n return false;\n }\n }\n\n int remainingBits = prefixLength % 8;\n if (remainingBits != 0) {\n int rangeByte = rangeValue[prefixLength / 8];\n int ipByte = ipValue[prefixLength / 8];\n\n int shift = 8 - remainingBits;\n int mask = 0xFF >> shift;\n\n return (rangeByte >> shift & mask) == (ipByte >> shift & mask);\n }\n\n return true;\n }\n return false;\n }\n\n @Override\n public String getExpression() {\n return expression + StringUtils.SLASH + prefixLength;\n }\n }\n\n // Utility class to convert IPv6 address to byte array\n private static class Inet6AddressConverter {\n public static byte[] convert(String ipv6Address) {\n byte[] ipAddress = new byte[16];\n String[] blocks = ipv6Address.split(\":\");\n\n for (int i = 0; i < blocks.length; i++) {\n String block = blocks[i];\n if (!block.isEmpty()) {\n ipAddress[i * 2] = (byte) Integer.parseInt(block.substring(0, 2), 16);\n ipAddress[i * 2 + 1] = (byte) Integer.parseInt(block.substring(2, 4), 16);\n }\n }\n\n return ipAddress;\n }\n }\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/matcher/impl/RegexRuleMatcher.java",
"content": "package im.zhaojun.zfile.core.util.matcher.impl;\n\nimport cn.hutool.core.util.ReUtil;\nimport im.zhaojun.zfile.core.constant.RuleTypeConstant;\nimport im.zhaojun.zfile.core.util.matcher.AbstractRuleMatcher;\nimport lombok.extern.slf4j.Slf4j;\n\n/**\n * 正则匹配器\n *\n * @author zhaojun\n */\n@Slf4j\npublic class RegexRuleMatcher extends AbstractRuleMatcher {\n\n @Override\n public boolean match(String ruleExpression, String testStr) {\n boolean match = ReUtil.isMatch(ruleExpression, testStr);\n if (log.isDebugEnabled()) {\n log.debug(\"正则匹配结果: {}, 规则表达式: {}, 测试值: {}\", match, ruleExpression, testStr);\n }\n return match;\n }\n\n @Override\n public boolean contains(String ruleExpression, String testStr) {\n boolean match = ReUtil.contains(ruleExpression, testStr);\n if (log.isDebugEnabled()) {\n log.debug(\"正则部分匹配结果: {}, 规则表达式: {}, 测试值: {}\", match, ruleExpression, testStr);\n }\n return match;\n }\n\n @Override\n public String getRuleType() {\n return RuleTypeConstant.REGEX;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/util/matcher/impl/SpringSimpleRuleMatcher.java",
"content": "package im.zhaojun.zfile.core.util.matcher.impl;\n\nimport im.zhaojun.zfile.core.constant.RuleTypeConstant;\nimport im.zhaojun.zfile.core.util.matcher.AbstractRuleMatcher;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.util.PatternMatchUtils;\n\n/**\n * 使用 {@link org.springframework.util.PatternMatchUtils} 来匹配规则\n *\n * @author zhaojun\n */\n@Slf4j\npublic class SpringSimpleRuleMatcher extends AbstractRuleMatcher {\n\n @Override\n public boolean match(String ruleExpression, String testStr) {\n boolean match = PatternMatchUtils.simpleMatch(ruleExpression, testStr);\n if (log.isDebugEnabled()) {\n log.debug(\"Spring Simple 规则匹配结果: {}, 规则表达式: {}, 测试值: {}\", match, ruleExpression, testStr);\n }\n return match;\n }\n\n @Override\n public String getRuleType() {\n return RuleTypeConstant.SPRING_SIMPLE;\n }\n\n @Override\n public boolean contains(String ruleExpression, String testStr) {\n return match(\"*\" + ruleExpression + \"*\", testStr);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/validation/StringListValue.java",
"content": "package im.zhaojun.zfile.core.validation;\n\nimport jakarta.validation.Constraint;\nimport jakarta.validation.Payload;\nimport java.lang.annotation.Documented;\nimport java.lang.annotation.ElementType;\nimport java.lang.annotation.Retention;\nimport java.lang.annotation.Target;\n\nimport static java.lang.annotation.RetentionPolicy.RUNTIME;\n\n/**\n * 字符串列表值校验注解\n *\n * @author zhaojun\n */\n@Documented\n@Constraint(validatedBy = { StringListValueConstraintValidator.class })\n@Target({ ElementType.METHOD, ElementType.FIELD, ElementType.ANNOTATION_TYPE })\n@Retention(RUNTIME)\npublic @interface StringListValue {\n\n String message() default \"\";\n\n Class[] groups() default { };\n\n Class[] payload() default { };\n\n String[] vals() default { };\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/core/validation/StringListValueConstraintValidator.java",
"content": "package im.zhaojun.zfile.core.validation;\n\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport jakarta.validation.ConstraintValidator;\nimport jakarta.validation.ConstraintValidatorContext;\n\nimport java.util.Arrays;\nimport java.util.HashSet;\nimport java.util.Set;\n\n/**\n * 字符串列表值校验器\n *\n * @author zhaojun\n */\npublic class StringListValueConstraintValidator implements ConstraintValidator {\n\n\tprivate final Set set = new HashSet<>();\n\n\t/**\n\t * 初始化方法\n\t *\n\t * @param constraintAnnotation\n\t * 校验注解对象\n\t */\n\t@Override\n\tpublic void initialize(StringListValue constraintAnnotation) {\n\t\tString[] vals = constraintAnnotation.vals();\n\t\tset.addAll(Arrays.asList(vals));\n\n\t}\n\n\n\t/**\n\t * 判断是否校验成功\n\t *\n\t * @param value\n\t * 需要校验的值\n\t *\n\t * @param context\n\t * 校验上下文\n\t *\n\t * @return 是否校验成功\n\t */\n\t@Override\n\tpublic boolean isValid(String value, ConstraintValidatorContext context) {\n\t\tif (StringUtils.isEmpty(value)) {\n\t\t\treturn true;\n\t\t}\n\t\treturn set.contains(value);\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/admin/controller/IpHelperController.java",
"content": "package im.zhaojun.zfile.module.admin.controller;\n\nimport cn.hutool.extra.servlet.JakartaServletUtil;\nimport com.alibaba.fastjson2.JSON;\nimport com.alibaba.fastjson2.JSONWriter;\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.core.util.RequestHolder;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport jakarta.servlet.http.HttpServletRequest;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.web.bind.annotation.GetMapping;\nimport org.springframework.web.bind.annotation.RequestMapping;\nimport org.springframework.web.bind.annotation.RestController;\n\nimport java.util.List;\nimport java.util.Map;\n\n/**\n * @author zhaojun\n */\n@Tag(name = \"IP 地址辅助 Controller\")\n@Slf4j\n@RequestMapping(\"/admin\")\n@RestController\npublic class IpHelperController {\n\n @Resource\n private HttpServletRequest httpServletRequest;\n\n @GetMapping(\"clientIp\")\n @Operation(summary = \"获取客户端IP\", description =\"获取当前请求的客户端IP地址\")\n public AjaxJson clientIp() {\n String clientIp = JakartaServletUtil.getClientIP(httpServletRequest);\n return AjaxJson.getSuccessData(clientIp);\n }\n\n @GetMapping(\"serverAddress\")\n @Operation(summary = \"获取服务器地址\", description = \"获取当前请求的服务器地址(如果是反向代理过，可能获取到的是反向代理服务器的地址)\")\n public AjaxJson serverAddress() {\n return AjaxJson.getSuccessData(RequestHolder.getRequestServerAddress());\n }\n\n @GetMapping(\"headers\")\n @Operation(summary = \"获取 Headers\", description = \"获取服务器接收到的请求头信息，可用于排查反向代理配置问题\")\n public AjaxJson headers() {\n Map> headersMap = JakartaServletUtil.getHeadersMap(httpServletRequest);\n Map singleValueHeaderMap = headersMap.entrySet().stream()\n .collect(java.util.stream.Collectors.toMap(\n Map.Entry::getKey,\n entry -> String.join(\",\", entry.getValue())\n ));\n return AjaxJson.getSuccessData(JSON.toJSONString(singleValueHeaderMap, JSONWriter.Feature.PrettyFormat));\n }\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/admin/controller/RuleMatcherTestController.java",
"content": "package im.zhaojun.zfile.module.admin.controller;\n\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.core.util.matcher.IRuleMatcher;\nimport im.zhaojun.zfile.core.util.matcher.RuleMatcherFactory;\nimport im.zhaojun.zfile.module.admin.model.request.TestRuleMatcherRequest;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.validation.Valid;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.web.bind.annotation.PostMapping;\nimport org.springframework.web.bind.annotation.RequestBody;\nimport org.springframework.web.bind.annotation.RequestMapping;\nimport org.springframework.web.bind.annotation.RestController;\n\nimport java.util.List;\n\n/**\n * @author zhaojun\n */\n@Tag(name = \"规则匹配辅助 Controller\")\n@Slf4j\n@RequestMapping(\"/admin\")\n@RestController\npublic class RuleMatcherTestController {\n\n /**\n * 根据传入的规则和测试值, 测试规则是否匹配, 规则支持多个, 用换行符分割. 如果匹配, 则返回匹配的规则表达式行.\n * @param testRuleMatcherRequest\n * 测试规则匹配请求\n *\n * @return 匹配成功的第一个表达式\n */\n @PostMapping(\"/rule-test\")\n public AjaxJson testRule(@RequestBody @Valid TestRuleMatcherRequest testRuleMatcherRequest) {\n if (testRuleMatcherRequest == null) {\n return AjaxJson.getSuccessData(null);\n }\n String rules = testRuleMatcherRequest.getRules();\n String testValue = testRuleMatcherRequest.getTestValue();\n if (StringUtils.isBlank(testValue) || StringUtils.isBlank(rules)) {\n return AjaxJson.getSuccessData(null);\n }\n\n List ruleList = StringUtils.split(rules, StringUtils.LF);\n IRuleMatcher ipRuleMatcher = RuleMatcherFactory.getRuleMatcher(testRuleMatcherRequest.getRuleType());\n return AjaxJson.getSuccessData(ipRuleMatcher.matchAnyReturnFirst(ruleList, testValue));\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/admin/model/request/TestRuleMatcherRequest.java",
"content": "package im.zhaojun.zfile.module.admin.model.request;\n\nimport lombok.Data;\n\nimport jakarta.validation.constraints.NotBlank;\n\n@Data\npublic class TestRuleMatcherRequest {\n\n @NotBlank(message = \"规则类型不能为空\")\n private String ruleType;\n\n @NotBlank(message = \"规则不能为空\")\n private String rules;\n\n @NotBlank(message = \"测试值不能为空\")\n private String testValue;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/annotation/JSONStringParse.java",
"content": "package im.zhaojun.zfile.module.config.annotation;\n\nimport java.lang.annotation.ElementType;\nimport java.lang.annotation.Retention;\nimport java.lang.annotation.RetentionPolicy;\nimport java.lang.annotation.Target;\n\n/**\n * 标注是否按 JSON 字符串解析\n *\n * @author zhaojun\n */\n@Target(ElementType.FIELD)\n@Retention(RetentionPolicy.RUNTIME)\npublic @interface JSONStringParse {\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/constant/SystemConfigConstant.java",
"content": "package im.zhaojun.zfile.module.config.constant;\n\n/**\n * 系统设置字段常量.\n *\n * @author zhaojun\n */\npublic class SystemConfigConstant {\n\n public static final String USERNAME = \"username\";\n\n public static final String PASSWORD = \"password\";\n\n public static final String LOGIN_VERIFY_MODE = \"loginVerifyMode\";\n\n // 这里名称和值不一样是历史遗留问题，最开始设计时弄混了名称，实际使用的是 aes\n public static final String AES_HEX_KEY = \"rsaHexKey\";\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/controller/SettingController.java",
"content": "package im.zhaojun.zfile.module.config.controller;\n\nimport cn.hutool.core.util.StrUtil;\nimport com.alibaba.fastjson2.JSON;\nimport com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\nimport com.github.xiaoymin.knife4j.annotations.ApiSort;\nimport im.zhaojun.zfile.core.annotation.DemoDisable;\nimport im.zhaojun.zfile.core.config.ZFileProperties;\nimport im.zhaojun.zfile.core.exception.core.BizException;\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.module.config.model.dto.SystemConfigDTO;\nimport im.zhaojun.zfile.module.config.model.request.*;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport jakarta.validation.Valid;\nimport org.apache.commons.lang3.BooleanUtils;\nimport org.springframework.beans.BeanUtils;\nimport org.springframework.web.bind.annotation.*;\n\n/**\n * 站点设定值接口\n *\n * @author zhaojun\n */\n@Tag(name = \"站点设置模块\")\n@ApiSort(2)\n@RestController\n@RequestMapping(\"/admin\")\npublic class SettingController {\n\n @Resource\n private SystemConfigService systemConfigService;\n\n @Resource\n private ZFileProperties zFileProperties;\n\n @ApiOperationSupport(order = 1)\n @Operation(summary = \"获取站点信息\", description = \"获取站点相关信息，如站点名称，风格样式，是否显示公告，是否显示文档区，自定义 CSS，JS 等参数\")\n @GetMapping(\"/config\")\n public AjaxJson getConfig() {\n SystemConfigDTO systemConfigDTO = systemConfigService.getSystemConfig();\n if (zFileProperties != null && zFileProperties.isDemoSite()) {\n SystemConfigDTO copy = JSON.parseObject(JSON.toJSONString(systemConfigDTO), SystemConfigDTO.class);\n copy.setAuthCode(null);\n copy.setRsaHexKey(null);\n return AjaxJson.getSuccessData(copy);\n }\n return AjaxJson.getSuccessData(systemConfigDTO);\n }\n\n @ApiOperationSupport(order = 3)\n @Operation(summary = \"修改站点设置\")\n @PutMapping(\"/config/site\")\n @DemoDisable\n public AjaxJson updateSiteSetting(@Valid @RequestBody UpdateSiteSettingRequest settingRequest) {\n if (StrUtil.length(settingRequest.getAuthCode()) > 36 && StrUtil.length(settingRequest.getAuthCode()) < 100) {\n throw new BizException(\"授权码长度异常，请检查是否额外复制了空格或特殊字符！\");\n }\n SystemConfigDTO systemConfigDTO = new SystemConfigDTO();\n BeanUtils.copyProperties(settingRequest, systemConfigDTO);\n systemConfigService.updateSystemConfig(systemConfigDTO);\n return AjaxJson.getSuccess();\n }\n\n\n @ApiOperationSupport(order = 4)\n @Operation(summary = \"修改显示设置\")\n @PutMapping(\"/config/view\")\n @DemoDisable\n public AjaxJson updateViewSetting(@Valid @RequestBody UpdateViewSettingRequest settingRequest) {\n SystemConfigDTO systemConfigDTO = new SystemConfigDTO();\n BeanUtils.copyProperties(settingRequest, systemConfigDTO);\n systemConfigService.updateSystemConfig(systemConfigDTO);\n return AjaxJson.getSuccess();\n }\n\n\n @ApiOperationSupport(order = 5)\n @Operation(summary = \"修改登陆安全设置\")\n @PutMapping(\"/config/security\")\n @DemoDisable\n public AjaxJson updateSecuritySetting(@Valid @RequestBody UpdateSecuritySettingRequest settingRequest) {\n SystemConfigDTO systemConfigDTO = new SystemConfigDTO();\n BeanUtils.copyProperties(settingRequest, systemConfigDTO);\n if (BooleanUtils.isNotTrue(settingRequest.getAdminTwoFactorVerify())) {\n systemConfigDTO.setLoginVerifySecret(\"\");\n }\n systemConfigService.updateSystemConfig(systemConfigDTO);\n return AjaxJson.getSuccess();\n }\n\n\n @ApiOperationSupport(order = 6)\n @Operation(summary = \"修改直链设置\")\n @PutMapping(\"/config/link\")\n @DemoDisable\n public AjaxJson updateLinkSetting(@Valid @RequestBody UpdateLinkSettingRequest settingRequest) {\n SystemConfigDTO systemConfigDTO = new SystemConfigDTO();\n BeanUtils.copyProperties(settingRequest, systemConfigDTO);\n systemConfigService.updateSystemConfig(systemConfigDTO);\n return AjaxJson.getSuccess();\n }\n\n @ApiOperationSupport(order = 7)\n @Operation(summary = \"修改访问控制设置\")\n @PutMapping(\"/config/access\")\n @DemoDisable\n public AjaxJson updateSecuritySetting(@Valid @RequestBody UpdateAccessSettingRequest updateAccessSettingRequest) {\n SystemConfigDTO systemConfigDTO = new SystemConfigDTO();\n BeanUtils.copyProperties(updateAccessSettingRequest, systemConfigDTO);\n systemConfigService.updateSystemConfig(systemConfigDTO);\n return AjaxJson.getSuccess();\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/controller/SiteController.java",
"content": "package im.zhaojun.zfile.module.config.controller;\n\nimport com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\nimport com.github.xiaoymin.knife4j.annotations.ApiSort;\nimport im.zhaojun.zfile.core.config.ZFileProperties;\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.core.util.ZFileAuthUtil;\nimport im.zhaojun.zfile.module.config.model.dto.SystemConfigDTO;\nimport im.zhaojun.zfile.module.config.model.result.FrontSiteConfigResult;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.storage.annotation.ProCheck;\nimport im.zhaojun.zfile.module.storage.context.StorageSourceContext;\nimport im.zhaojun.zfile.module.storage.model.request.base.FileListConfigRequest;\nimport im.zhaojun.zfile.module.storage.model.result.StorageSourceConfigResult;\nimport im.zhaojun.zfile.module.storage.service.StorageSourceService;\nimport im.zhaojun.zfile.module.storage.service.base.AbstractBaseFileService;\nimport im.zhaojun.zfile.module.user.model.constant.UserConstant;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport jakarta.validation.Valid;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.beans.BeanUtils;\nimport org.springframework.web.bind.annotation.*;\n\nimport java.util.Objects;\n\n/**\n * 面向前台的站点基础模块接口\n *\n * @author zhaojun\n */\n@Tag(name = \"站点基础模块\")\n@ApiSort(1)\n@Slf4j\n@RequestMapping(\"/api/site\")\n@RestController\npublic class SiteController {\n\n\t@Resource\n\tprivate ZFileProperties zFileProperties;\n\n\t@Resource\n\tprivate StorageSourceService storageSourceService;\n\n\t@Resource\n\tprivate SystemConfigService systemConfigService;\n\n\t@ApiOperationSupport(order = 1)\n\t@Operation(summary = \"获取站点全局设置\", description = \"获取站点全局设置, 包括是否页面布局、列表尺寸、公告、配置信息\")\n\t@GetMapping(\"/config/global\")\n\t@ProCheck\n\tpublic AjaxJson globalConfig() {\n\t\tSystemConfigDTO systemConfig = systemConfigService.getSystemConfig();\n\n\t\tFrontSiteConfigResult frontSiteConfigResult = new FrontSiteConfigResult();\n\t\tBeanUtils.copyProperties(systemConfig, frontSiteConfigResult);\n\n\t\tfrontSiteConfigResult.setDebugMode(zFileProperties.isDebug());\n\t\tboolean guestUser = Objects.equals(ZFileAuthUtil.getCurrentUserId(), UserConstant.ANONYMOUS_ID);\n\t\tboolean guestIndexNotBlank = StringUtils.isNotBlank(systemConfig.getGuestIndexHtml());\n\t\tfrontSiteConfigResult.setGuest(guestUser && guestIndexNotBlank);\n\t\treturn AjaxJson.getSuccessData(frontSiteConfigResult);\n\t}\n\n\n\t@ApiOperationSupport(order = 2)\n\t@Operation(summary = \"获取存储源设置\", description = \"获取某个存储源的设置信息, 包括是否启用, 名称, 存储源类型, 存储源配置信息\")\n\t@PostMapping(\"/config/storage\")\n\tpublic AjaxJson storageList(@Valid @RequestBody FileListConfigRequest fileListConfigRequest) {\n\t\tStorageSourceConfigResult storageSourceConfigResult = storageSourceService.getStorageConfigSource(fileListConfigRequest);\n\t\treturn AjaxJson.getSuccessData(storageSourceConfigResult);\n\t}\n\n\n\t@ApiOperationSupport(order = 3)\n\t@Operation(summary = \"获取用户存储源路径\", description = \"获取用户存储源路径\")\n\t@GetMapping(\"/config/userRootPath/{storageKey}\")\n\tpublic AjaxJson getUserRootPath(@PathVariable(\"storageKey\") String storageKey) {\n\t\tAbstractBaseFileService baseFileService = StorageSourceContext.getByStorageKey(storageKey);\n\t\tif (baseFileService == null || baseFileService.getCurrentUserBasePath() == null) {\n\t\t\treturn AjaxJson.getSuccessData(\"\");\n\t\t}\n\t\treturn AjaxJson.getSuccessData(baseFileService.getCurrentUserBasePath());\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/event/DirectLinkPrefixModifyHandler.java",
"content": "package im.zhaojun.zfile.module.config.event;\n\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.module.config.model.entity.SystemConfig;\nimport im.zhaojun.zfile.module.link.controller.DirectLinkController;\nimport im.zhaojun.zfile.module.link.service.DynamicDirectLinkPrefixService;\nimport jakarta.annotation.Resource;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.stereotype.Component;\nimport org.springframework.web.servlet.mvc.method.RequestMappingInfo;\n\n/**\n * 接收系统设置修改事件, 修改直链前缀时, 动态更新直链前缀.\n *\n * @author zhaojun\n */\n@Slf4j\n@Component\npublic class DirectLinkPrefixModifyHandler implements ISystemConfigModifyHandler {\n\n @Resource\n private DynamicDirectLinkPrefixService dynamicDirectLinkPrefixService;\n\n @Override\n public void modify(SystemConfig originalSystemConfig, SystemConfig newSystemConfig) {\n String oldValue = originalSystemConfig.getValue();\n String newValue = newSystemConfig.getValue();\n if (StringUtils.equals(oldValue, newValue)) {\n log.info(\"检测到修改了直链前缀, 但是新值和旧值相同, 不做处理.\");\n return;\n }\n\n RequestMappingInfo requestMappingInfo = RequestMappingInfo.paths(newValue + DirectLinkController.DIRECT_LINK_SUFFIX_PATH).build();\n dynamicDirectLinkPrefixService.updateRegisterMappingHandler(SystemConfig.DIRECT_LINK_PREFIX_NAME, requestMappingInfo);\n log.info(\"检测到修改了直链前缀, [{}] -> [{}], 已自动更新直链前缀.\", oldValue, newValue);\n }\n\n @Override\n public boolean matches(String name) {\n return SystemConfig.DIRECT_LINK_PREFIX_NAME.equals(name);\n }\n\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/event/ISystemConfigModifyHandler.java",
"content": "package im.zhaojun.zfile.module.config.event;\n\nimport im.zhaojun.zfile.module.config.model.entity.SystemConfig;\n\n/**\n * 系统设置修改事件\n *\n * @author zhaojun\n */\npublic interface ISystemConfigModifyHandler {\n\n /**\n * 修改系统设置时会触发此事件\n *\n */\n void modify(SystemConfig originalSystemConfig, SystemConfig newSystemConfig);\n\n /**\n * 判断是否匹配当前处理器\n *\n * @param name\n * 配置项名称\n *\n * @return 是否匹配\n */\n boolean matches(String name);\n\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/event/SecureLoginEntryModifyHandler.java",
"content": "package im.zhaojun.zfile.module.config.event;\n\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.module.config.model.entity.SystemConfig;\nimport im.zhaojun.zfile.module.user.service.DynamicLoginEntryService;\nimport im.zhaojun.zfile.module.user.util.LoginEntryPathUtils;\nimport jakarta.annotation.Resource;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.stereotype.Component;\nimport org.springframework.web.servlet.mvc.method.RequestMappingInfo;\n\n/**\n * 监听安全登录入口配置变更，动态更新登录接口映射。\n *\n * @author zhaojun\n */\n@Slf4j\n@Component\npublic class SecureLoginEntryModifyHandler implements ISystemConfigModifyHandler {\n\n @Resource\n private DynamicLoginEntryService dynamicLoginEntryService;\n\n @Override\n public void modify(SystemConfig originalSystemConfig, SystemConfig newSystemConfig) {\n String oldPath = LoginEntryPathUtils.resolveLoginPath(originalSystemConfig.getValue());\n String newPath = LoginEntryPathUtils.resolveLoginPath(newSystemConfig.getValue());\n\n if (StringUtils.equals(oldPath, newPath)) {\n log.info(\"检测到修改安全登录入口，但实际登录路径未变化，跳过处理。\");\n return;\n }\n\n RequestMappingInfo requestMappingInfo = dynamicLoginEntryService.buildLoginRequestMappingInfo(newSystemConfig.getValue());\n dynamicLoginEntryService.updateRegisterMappingHandler(SystemConfig.SECURE_LOGIN_ENTRY_NAME, requestMappingInfo);\n log.info(\"安全登录入口已更新，{} -> {}\", oldPath, newPath);\n }\n\n @Override\n public boolean matches(String name) {\n return SystemConfig.SECURE_LOGIN_ENTRY_NAME.equals(name);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/event/SystemConfigModifyHandlerChain.java",
"content": "package im.zhaojun.zfile.module.config.event;\n\nimport im.zhaojun.zfile.module.config.model.entity.SystemConfig;\nimport org.springframework.stereotype.Component;\n\nimport jakarta.annotation.Resource;\nimport java.util.List;\n\n/**\n * @author zhaojun\n */\n@Component\npublic class SystemConfigModifyHandlerChain {\n\n @Resource\n private List handlers;\n\n public void execute(SystemConfig originalSystemConfig, SystemConfig newSystemConfig) {\n handlers.stream()\n .filter(handler -> handler.matches(originalSystemConfig.getName()))\n .forEach(handler -> handler.modify(originalSystemConfig, newSystemConfig));\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/mapper/SystemConfigMapper.java",
"content": "package im.zhaojun.zfile.module.config.mapper;\n\nimport com.baomidou.mybatisplus.core.mapper.BaseMapper;\nimport im.zhaojun.zfile.module.config.model.entity.SystemConfig;\nimport org.apache.ibatis.annotations.Mapper;\nimport org.apache.ibatis.annotations.Param;\n\nimport java.util.List;\n\n\n/**\n * 系统配置 Mapper 接口\n *\n * @author zhaojun\n */\n@Mapper\npublic interface SystemConfigMapper extends BaseMapper {\n\n /**\n * 获取所有系统设置\n *\n * @return 系统设置列表\n */\n List findAll();\n\n\n /**\n * 根据系统设置名称获取设置信息\n *\n * @param name\n * 系统设置名称\n *\n * @return 系统设置信息\n */\n SystemConfig findByName(@Param(\"name\")String name);\n\n\n /**\n * 批量保存系统设置\n *\n * @param list\n * 系统设置列表\n *\n * @return 保存记录数\n */\n int saveAll(@Param(\"list\")List list);\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/dto/LinkExpireDTO.java",
"content": "package im.zhaojun.zfile.module.config.model.dto;\n\nimport lombok.Data;\n\nimport java.io.Serializable;\n\n@Data\npublic class LinkExpireDTO implements Serializable {\n\n private static final long serialVersionUID = 1L;\n\n private Integer value;\n\n private String unit;\n\n private Long seconds;\n\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/dto/SystemConfigDTO.java",
"content": "package im.zhaojun.zfile.module.config.model.dto;\n\nimport com.fasterxml.jackson.annotation.JsonIgnore;\nimport im.zhaojun.zfile.module.config.annotation.JSONStringParse;\nimport im.zhaojun.zfile.module.config.model.enums.FileClickModeEnum;\nimport im.zhaojun.zfile.module.link.model.enums.RefererTypeEnum;\nimport im.zhaojun.zfile.module.user.model.enums.LoginLogModeEnum;\nimport im.zhaojun.zfile.module.user.model.enums.LoginVerifyModeEnum;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport java.io.Serializable;\nimport java.util.ArrayList;\nimport java.util.List;\n\n/**\n * 系统设置传输类\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"系统设置类\")\npublic class SystemConfigDTO implements Serializable {\n\n private static final long serialVersionUID = 1L;\n\n @Schema(title = \"站点名称\", example = \"ZFile Site Name\")\n private String siteName;\n\n @Schema(title = \"用户名\", example = \"admin\")\n @Deprecated\n private String username;\n\n @Schema(title = \"头像地址\", example = \"https://cube.elemecdn.com/3/7c/3ea6beec64369c2642b92c6726f1epng.png\")\n private String avatar;\n\n @Schema(title = \"备案号\", example = \"冀ICP备12345678号-1\")\n private String icp;\n\n @JsonIgnore\n @Deprecated\n private String password;\n\n @Schema(title = \"自定义 JS\")\n private String customJs;\n\n @Schema(title = \"自定义 CSS\")\n private String customCss;\n\n @Schema(title = \"列表尺寸\", description =\"large:大,default:中,small:小\", example = \"default\")\n private String tableSize;\n\n @Schema(title = \"是否显示文档区\", example = \"true\")\n private Boolean showDocument;\n\n @Schema(title = \"网站公告\", example = \"ZFile 网站公告\")\n private String announcement;\n\n @Schema(title = \"是否显示网站公告\", example = \"true\")\n private Boolean showAnnouncement;\n\n @Schema(title = \"页面布局\", description =\"full:全屏,center:居中\", example = \"full\")\n private String layout;\n\n @Schema(title = \"移动端页面布局\", description =\"full:全屏,center:居中\", example = \"full\")\n private String mobileLayout;\n\n @Schema(title = \"移动端显示文件大小\", description = \"仅适用列表视图\", example = \"true\")\n private Boolean mobileShowSize;\n\n @Schema(title = \"是否显示生成直链功能（含直链和路径短链）\", example = \"true\")\n private Boolean showLinkBtn;\n\n @Schema(title = \"是否显示生成短链功能\", example = \"true\")\n private Boolean showShortLink;\n\n @Schema(title = \"是否显示生成路径链接功能\", example = \"true\")\n private Boolean showPathLink;\n\n @Schema(title = \"是否已初始化\", example = \"true\")\n private Boolean installed;\n\n @Schema(title = \"自定义视频文件后缀格式\")\n private String customVideoSuffix;\n\n @Schema(title = \"自定义图像文件后缀格式\")\n private String customImageSuffix;\n\n @Schema(title = \"自定义音频文件后缀格式\")\n private String customAudioSuffix;\n\n @Schema(title = \"自定义文本文件后缀格式\")\n private String customTextSuffix;\n\n @Schema(title = \"自定义Office后缀格式\")\n private String customOfficeSuffix;\n\n @Schema(title = \"自定义kkFileView后缀格式\")\n private String customKkFileViewSuffix;\n\n @Schema(title = \"直链地址前缀\")\n private String directLinkPrefix;\n\n @Schema(title = \"直链 Referer 防盗链类型\")\n private RefererTypeEnum refererType;\n\n @Schema(title = \"是否记录下载日志\", example = \"true\")\n private Boolean recordDownloadLog;\n\n @Schema(title = \"直链 Referer 是否允许为空\")\n private Boolean refererAllowEmpty;\n\n @Schema(title = \"直链 Referer 值\")\n private String refererValue;\n\n /**\n * 废弃的字段，改为使用 {@link #adminTwoFactorVerify} 和 {@link #loginVerifySecret} 代替\n */\n @Schema(title = \"管理员登陆验证方式，目前仅支持 2FA 认证或关闭\")\n @Deprecated\n private LoginVerifyModeEnum loginVerifyMode;\n\n @Schema(title = \"登陆验证 Secret\")\n private String loginVerifySecret;\n\n @Schema(title = \"是否启用登陆验证码\", example = \"true\")\n private Boolean loginImgVerify;\n\n @Schema(title = \"是否为管理员启用双因素认证\", example = \"true\")\n private Boolean adminTwoFactorVerify;\n\n @Schema(title = \"根目录是否显示所有存储源\", description =\"勾选则根目录显示所有存储源列表, 反之会自动显示第一个存储源的内容.\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n private Boolean rootShowStorage;\n\n @Schema(title = \"强制后端地址\", description =\"强制指定生成直链，短链，获取回调地址时的地址。\", example = \"http://xxx.example.com\")\n private String forceBackendAddress;\n\n @Schema(title = \"前端域名\", description =\"前端域名，前后端分离情况下需要配置.\", example = \"http://xxx.example.com\")\n private String frontDomain;\n\n @Schema(title = \"是否在前台显示登陆按钮\", example = \"true\")\n private Boolean showLogin;\n\n @Schema(title = \"安全登录入口\", description = \"用于隐藏默认登录地址的安全入口，不包含 '/'\", example = \"admin\")\n private String secureLoginEntry;\n\n @Schema(title = \"登录日志模式\", example = \"all\")\n private LoginLogModeEnum loginLogMode;\n\n @Schema(title = \"RAS Hex Key\", example = \"r2HKbzc1DfvOs5uHhLn7pA==\")\n private String rsaHexKey;\n\n @Schema(title = \"默认文件点击习惯\", example = \"click\")\n private FileClickModeEnum fileClickMode;\n\n @Schema(title = \"移动端默认文件点击习惯\", example = \"click\")\n private FileClickModeEnum mobileFileClickMode;\n\n @Schema(title = \"授权码\", example = \"e619510f-cdcd-f657-6c5e-2d12e9a28ae5\")\n private String authCode;\n\n @Schema(title = \"最大同时上传文件数\", example = \"5\")\n private Integer maxFileUploads;\n\n @Schema(title = \"onlyOffice 在线预览地址\", example = \"http://office.zfile.vip\")\n private String onlyOfficeUrl;\n\n @Schema(title = \"onlyOffice Secret\", example = \"X9rBGypwWE86Lca8e4Mo55iHFoiyh9ed\")\n private String onlyOfficeSecret;\n\n @Schema(title = \"kkFileView 在线预览地址\", example = \"http://kkfile.zfile.vip\")\n private String kkFileViewUrl;\n\n @Schema(title = \"kkFileView 预览方式\", example = \"iframe/newTab\")\n private String kkFileViewOpenMode;\n\n @Schema(title = \"启用 WebDAV\", example = \"true\")\n private Boolean webdavEnable;\n\n @Schema(title = \"WebDAV 服务器中转下载\", example = \"true\")\n private Boolean webdavProxy;\n\n @Schema(title = \"WebDAV 匿名用户访问\", example = \"true\")\n private Boolean webdavAllowAnonymous;\n\n @Schema(title = \"WebDAV 账号\", example = \"admin\")\n private String webdavUsername;\n\n @Schema(title = \"WebDAV 密码\", example = \"123456\")\n private String webdavPassword;\n\n @Schema(title = \"是否允许路径直链可直接访问\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n private Boolean allowPathLinkAnonAccess;\n\n @Schema(title = \"默认最大显示文件数\", example = \"1000\")\n private Integer maxShowSize;\n\n @Schema(title = \"每次加载更多文件数\", example = \"50\")\n private Integer loadMoreSize;\n\n @Schema(title = \"默认排序字段\", example = \"name\")\n private String defaultSortField;\n\n @Schema(title = \"默认排序方向\", example = \"asc\")\n private String defaultSortOrder;\n\n @Schema(title = \"站点 Home 名称\", example = \"xxx 的小站\")\n private String siteHomeName;\n\n @Schema(title = \"站点 Home Logo\", example = \"true\")\n private String siteHomeLogo;\n\n @Schema(title = \"站点 Logo 点击后链接\", example = \"https://www.zfile.vip\")\n private String siteHomeLogoLink;\n\n @Schema(title = \"站点 Logo 链接打开方式\", example = \"_blank\")\n private String siteHomeLogoTargetMode;\n\n @Schema(title = \"管理员页面点击 Logo 回到首页打开方式\", example = \"_blank\")\n private String siteAdminLogoTargetMode;\n\n @Schema(title = \"管理员页面点击版本号打开更新日志\", example = \"true\")\n private Boolean siteAdminVersionOpenChangeLog;\n\n @Schema(title = \"限制直链下载秒数\", example = \"_blank\")\n private Integer linkLimitSecond;\n\n @Schema(title = \"限制直链下载次数\", example = \"_blank\")\n private Integer linkDownloadLimit;\n\n @Schema(title = \"网站 favicon 图标地址\", example = \"https://www.example.com/favicon.ico\")\n private String faviconUrl;\n\n @Schema(title = \"短链过期时间设置\", example = \"[{value: 1, unit: \\\"day\\\"}, {value: 1, unit: \\\"week\\\"}, {value: 1, unit: \\\"month\\\"}, {value: 1, unit: \\\"year\\\"}]\")\n @JSONStringParse\n private List linkExpireTimes;\n\n @Schema(title = \"是否默认记住密码\", example = \"true\")\n private Boolean defaultSavePwd;\n\n @Schema(title = \"普通下载是否启用确认弹窗\", example = \"true\")\n private Boolean enableNormalDownloadConfirm;\n\n @Schema(title = \"打包下载是否启用确认弹窗\", example = \"true\")\n private Boolean enablePackageDownloadConfirm;\n\n @Schema(title = \"批量下载是否启用确认弹窗\", example = \"true\")\n private Boolean enableBatchDownloadConfirm;\n\n /**\n * 废弃的字段，不再使用悬浮菜单\n */\n @Deprecated\n @Schema(title = \"是否启用 hover 菜单\", example = \"true\")\n private Boolean enableHoverMenu;\n\n @Schema(title = \"访问 ip 黑名单\", example = \"162.13.1.0/24\\n192.168.1.1\")\n private String accessIpBlocklist;\n\n @Schema(title = \"访问 ua 黑名单\", example = \"Mozilla/5.0 (Linux; Android) AppleWebKit/537.36*\")\n private String accessUaBlocklist;\n\n @Schema(title = \"匿名用户首页显示内容\")\n private String guestIndexHtml;\n\n public String getAnnouncement() {\n return announcement == null ? \"\" : announcement;\n }\n\n public List getLinkExpireTimes() {\n if (linkExpireTimes == null) {\n LinkExpireDTO linkExpireDTO = new LinkExpireDTO();\n linkExpireDTO.setValue(1);\n linkExpireDTO.setUnit(\"d\");\n linkExpireDTO.setSeconds(86400L);\n linkExpireTimes = new ArrayList<>();\n linkExpireTimes.add(linkExpireDTO);\n }\n return linkExpireTimes;\n }\n\n public String getLayout() {\n return layout == null ? \"full\" : layout;\n }\n\n public String getMobileLayout() {\n return mobileLayout == null ? getLayout() : mobileLayout;\n }\n\n /**\n * 获取普通下载是否启用确认弹窗配置.\n *\n * @return 若为空则返回 true\n */\n public Boolean getEnableNormalDownloadConfirm() {\n return enableNormalDownloadConfirm == null ? Boolean.TRUE : enableNormalDownloadConfirm;\n }\n\n /**\n * 获取打包下载是否启用确认弹窗配置.\n *\n * @return 若为空则返回 true\n */\n public Boolean getEnablePackageDownloadConfirm() {\n return enablePackageDownloadConfirm == null ? Boolean.TRUE : enablePackageDownloadConfirm;\n }\n\n /**\n * 获取批量下载是否启用确认弹窗配置.\n *\n * @return 若为空则返回 true\n */\n public Boolean getEnableBatchDownloadConfirm() {\n return enableBatchDownloadConfirm == null ? Boolean.TRUE : enableBatchDownloadConfirm;\n }\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/entity/SystemConfig.java",
"content": "package im.zhaojun.zfile.module.config.model.entity;\n\nimport com.baomidou.mybatisplus.annotation.IdType;\nimport com.baomidou.mybatisplus.annotation.TableField;\nimport com.baomidou.mybatisplus.annotation.TableId;\nimport com.baomidou.mybatisplus.annotation.TableName;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport java.io.Serializable;\n\n/**\n * 系统设置 entity\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"系统设置\")\n@TableName(value = \"system_config\")\npublic class SystemConfig implements Serializable {\n\n public static final String DIRECT_LINK_PREFIX_NAME = \"directLinkPrefix\";\n\n public static final String SECURE_LOGIN_ENTRY_NAME = \"secureLoginEntry\";\n\n private static final long serialVersionUID = 1L;\n\n @TableId(value = \"id\", type = IdType.AUTO)\n @Schema(title = \"ID, 新增无需填写\", example = \"1\")\n private Integer id;\n\n\n @TableField(value = \"name\")\n @Schema(title = \"系统设置名称\", example = \"siteName\")\n private String name;\n\n\n @TableField(value = \"`value`\")\n @Schema(title = \"系统设置值\", example = \"ZFile 演示站\")\n private String value;\n\n\n @TableField(value = \"title\")\n @Schema(title = \"系统设置描述\", example = \"站点名称\")\n private String title;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/enums/FileClickModeEnum.java",
"content": "package im.zhaojun.zfile.module.config.model.enums;\n\nimport com.baomidou.mybatisplus.annotation.EnumValue;\nimport com.fasterxml.jackson.annotation.JsonValue;\nimport lombok.AllArgsConstructor;\nimport lombok.Getter;\n\n/**\n * 文件点击模式枚举, 用于控制文件是单击打开还是双击打开\n *\n * @author zhaojun\n */\n@Getter\n@AllArgsConstructor\npublic enum FileClickModeEnum {\n\n\t/**\n\t * 单击打开文件/文件夹\n\t */\n\tCLICK(\"click\"),\n\n\t/**\n\t * 双击打开文件/文件夹\n\t */\n\tDBCLICK(\"dbclick\");\n\n\t@EnumValue\n\t@JsonValue\n\tprivate final String value;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/request/UpdateAccessSettingRequest.java",
"content": "package im.zhaojun.zfile.module.config.model.request;\n\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\n/**\n * 站点访问控制请求参数类\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"站点访问控制参数类\")\npublic class UpdateAccessSettingRequest {\n\n @Schema(title = \"访问 ip 黑名单\", example = \"162.13.1.0/24\\n192.168.1.1\")\n private String accessIpBlocklist;\n\n @Schema(title = \"访问 ua 黑名单\", example = \"Mozilla/5.0 (Linux; Android) AppleWebKit/537.36*\")\n private String accessUaBlocklist;\n\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/request/UpdateLinkSettingRequest.java",
"content": "package im.zhaojun.zfile.module.config.model.request;\n\nimport im.zhaojun.zfile.module.config.model.dto.LinkExpireDTO;\nimport im.zhaojun.zfile.module.link.model.enums.RefererTypeEnum;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport jakarta.validation.constraints.NotBlank;\nimport lombok.Data;\n\nimport java.util.List;\n\n/**\n * 直链设置请求参数类\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"直链设置请求参数类\")\npublic class UpdateLinkSettingRequest {\n\n\t@Schema(title = \"是否记录下载日志\", example = \"true\")\n\tprivate Boolean recordDownloadLog;\n\n\t@Schema(title = \"直链 Referer 防盗链类型\")\n\tprivate RefererTypeEnum refererType;\n\n\t@Schema(title = \"直链 Referer 是否允许为空\")\n\tprivate Boolean refererAllowEmpty;\n\n\t@Schema(title = \"直链 Referer 值\")\n\tprivate String refererValue;\n\n\t@Schema(title = \"直链地址前缀\")\n\t@NotBlank(message = \"直链地址前缀不能为空\")\n\tprivate String directLinkPrefix;\n\n\t@Schema(title = \"是否显示生成直链功能（含直链和路径短链）\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean showLinkBtn;\n\n\t@Schema(title = \"是否显示生成短链功能\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean showShortLink;\n\n\t@Schema(title = \"是否显示生成路径链接功能\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean showPathLink;\n\n\t@Schema(title = \"是否允许路径直链可直接访问\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean allowPathLinkAnonAccess;\n\n\t@Schema(title = \"限制直链下载秒数\", example = \"_blank\")\n\tprivate Integer linkLimitSecond;\n\n\t@Schema(title = \"限制直链下载次数\", example = \"_blank\")\n\tprivate Integer linkDownloadLimit;\n\n\t@Schema(title = \"短链过期时间设置\", example = \"[{value: 1, unit: \\\"day\\\"}, {value: 1, unit: \\\"week\\\"}, {value: 1, unit: \\\"month\\\"}, {value: 1, unit: \\\"year\\\"}]\")\n\tprivate List linkExpireTimes;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/request/UpdateSecuritySettingRequest.java",
"content": "package im.zhaojun.zfile.module.config.model.request;\n\nimport im.zhaojun.zfile.module.user.model.enums.LoginLogModeEnum;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport jakarta.validation.constraints.Pattern;\nimport jakarta.validation.constraints.Size;\nimport lombok.Data;\n\n/**\n * 登陆安全设置请求参数类\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"登陆安全设置请求参数类\")\npublic class UpdateSecuritySettingRequest {\n\n\t@Schema(title = \"是否在前台显示登陆按钮\", example = \"true\")\n\tprivate Boolean showLogin;\n\n @Schema(title = \"安全登录入口\", description = \"仅允许字母、数字、短横线和下划线，长度不超过 32\", example = \"admin\")\n @Size(max = 32, message = \"安全登录入口长度不能超过 32 个字符\")\n @Pattern(regexp = \"^[A-Za-z0-9_-]*$\", message = \"安全登录入口只能包含字母、数字、短横线和下划线\")\n private String secureLoginEntry;\n\n\t@Schema(title = \"登录日志模式\", example = \"all\")\n\tprivate LoginLogModeEnum loginLogMode;\n\n\t@Schema(title = \"是否启用登陆验证码\", example = \"true\")\n\tprivate Boolean loginImgVerify;\n\n\t@Schema(title = \"是否为管理员启用双因素认证\", example = \"true\")\n\tprivate Boolean adminTwoFactorVerify;\n\n\t@Schema(title = \"2FA登陆验证 Secret\")\n\tprivate String loginVerifySecret;\n\n\t@Schema(title = \"匿名用户首页显示内容\")\n\tprivate String guestIndexHtml;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/request/UpdateSiteSettingRequest.java",
"content": "package im.zhaojun.zfile.module.config.model.request;\n\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport jakarta.validation.constraints.NotBlank;\n\n/**\n * 站点设置请求参数类\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"站点设置请求参数类\")\npublic class UpdateSiteSettingRequest {\n\n\t@Schema(title = \"站点名称\", requiredMode = Schema.RequiredMode.REQUIRED, example = \"ZFile Site Name\")\n\t@NotBlank(message = \"站点名称不能为空\")\n\tprivate String siteName;\n\n\t@Schema(title = \"强制后端地址\", description =\"强制指定生成直链，短链，获取回调地址时的地址。\", example = \"http://xxx.example.com\")\n\tprivate String forceBackendAddress;\n\n\t@Schema(title = \"前端域名\", description =\"前端域名，前后端分离情况下需要配置.\", example = \"http://xxx.example.com\")\n\tprivate String frontDomain;\n\n\t@Schema(title = \"头像地址\", example = \"https://cube.elemecdn.com/3/7c/3ea6beec64369c2642b92c6726f1epng.png\")\n\tprivate String avatar;\n\n\t@Schema(title = \"备案号\", example = \"冀ICP备12345678号-1\")\n\tprivate String icp;\n\n\t@Schema(title = \"授权码\", example = \"e619510f-cdcd-f657-6c5e-2d12e9a28ae5\")\n\tprivate String authCode;\n\n\t@Schema(title = \"最大同时上传文件数\", example = \"5\")\n\tprivate Integer maxFileUploads;\n\n\t@Schema(title = \"站点 Home 名称\", example = \"xxx 的小站\")\n\tprivate String siteHomeName;\n\n\t@Schema(title = \"站点 Home Logo\", example = \"true\")\n\tprivate String siteHomeLogo;\n\n\t@Schema(title = \"站点 Logo 点击后链接\", example = \"https://www.zfile.vip\")\n\tprivate String siteHomeLogoLink;\n\n\t@Schema(title = \"站点 Logo 链接打开方式\", example = \"_blank\")\n\tprivate String siteHomeLogoTargetMode;\n\n\t@Schema(title = \"网站 favicon 图标地址\", example = \"https://www.example.com/favicon.ico\")\n\tprivate String faviconUrl;\n\n\t@Schema(title = \"管理员页面点击 Logo 回到首页打开方式\", example = \"_blank\")\n\tprivate String siteAdminLogoTargetMode;\n\n\t@Schema(title = \"管理员页面点击版本号打开更新日志\", example = \"true\")\n\tprivate Boolean siteAdminVersionOpenChangeLog;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/request/UpdateUserNameAndPasswordRequest.java",
"content": "package im.zhaojun.zfile.module.config.model.request;\n\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport jakarta.validation.constraints.NotBlank;\n\n/**\n * 用户修改密码请求参数类\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"用户修改密码请求参数类\")\npublic class UpdateUserNameAndPasswordRequest {\n\n @Schema(title = \"用户名\", requiredMode = Schema.RequiredMode.REQUIRED, example = \"admin\")\n @NotBlank(message = \"用户名不能为空\")\n private String username;\n\n @Schema(title = \"密码\", requiredMode = Schema.RequiredMode.REQUIRED, example = \"123456\")\n @NotBlank(message = \"密码不能为空\")\n private String password;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/request/UpdateViewSettingRequest.java",
"content": "package im.zhaojun.zfile.module.config.model.request;\n\nimport im.zhaojun.zfile.module.config.model.enums.FileClickModeEnum;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\n/**\n * 显示设置请求参数类\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"显示设置请求参数类\")\npublic class UpdateViewSettingRequest {\n\n\t@Schema(title = \"根目录是否显示所有存储源\", description =\"勾选则根目录显示所有存储源列表, 反之会自动显示第一个存储源的内容.\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean rootShowStorage;\n\n\t@Schema(title = \"页面布局\", description =\"full:全屏,center:居中\", example = \"full\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate String layout;\n\n\t@Schema(title = \"移动端页面布局\", description =\"full:全屏,center:居中\", example = \"full\")\n\tprivate String mobileLayout;\n\n\t@Schema(title = \"移动端显示文件大小\", description = \"仅适用列表视图\", example = \"true\")\n\tprivate Boolean mobileShowSize;\n\n\t@Schema(title = \"列表尺寸\", description =\"large:大,default:中,small:小\", example = \"default\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate String tableSize;\n\n\t@Schema(title = \"自定义视频文件后缀格式\")\n\tprivate String customVideoSuffix;\n\n\t@Schema(title = \"自定义图像文件后缀格式\")\n\tprivate String customImageSuffix;\n\n\t@Schema(title = \"自定义音频文件后缀格式\")\n\tprivate String customAudioSuffix;\n\n\t@Schema(title = \"自定义文本文件后缀格式\")\n\tprivate String customTextSuffix;\n\n\t@Schema(title = \"自定义Office后缀格式\")\n\tprivate String customOfficeSuffix;\n\n\t@Schema(title = \"自定义kkFileView后缀格式\")\n\tprivate String customKkFileViewSuffix;\n\n\t@Schema(title = \"是否显示文档区\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean showDocument;\n\n\t@Schema(title = \"是否显示网站公告\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean showAnnouncement;\n\n\t@Schema(title = \"网站公告\", example = \"ZFile 网站公告\")\n\tprivate String announcement;\n\n\t@Schema(title = \"自定义 CSS\")\n\tprivate String customCss;\n\n\t@Schema(title = \"自定义 JS\")\n\tprivate String customJs;\n\n\t@Schema(title = \"默认文件点击习惯\", example = \"click\")\n\tprivate FileClickModeEnum fileClickMode;\n\n\t@Schema(title = \"移动端默认文件点击习惯\", example = \"click\")\n\tprivate FileClickModeEnum mobileFileClickMode;\n\n\t@Schema(title = \"onlyOffice 在线预览地址\", example = \"http://office.zfile.vip\")\n\tprivate String onlyOfficeUrl;\n\n\t@Schema(title = \"onlyOffice Secret\", example = \"X9rBGypwWE86Lca8e4Mo55iHFoiyh9ed\")\n\tprivate String onlyOfficeSecret;\n\n\t@Schema(title = \"kkFileView 在线预览地址\", example = \"http://kkfile.zfile.vip\")\n\tprivate String kkFileViewUrl;\n\n\t@Schema(title = \"kkFileView 预览方式\", example = \"iframe/newTab\")\n\tprivate String kkFileViewOpenMode;\n\n\t@Schema(title = \"默认最大显示文件数\", example = \"1000\")\n\tprivate Integer maxShowSize;\n\n\t@Schema(title = \"每次加载更多文件数\", example = \"50\")\n\tprivate Integer loadMoreSize;\n\n\t@Schema(title = \"默认排序字段\", example = \"name\")\n\tprivate String defaultSortField;\n\n\t@Schema(title = \"默认排序方向\", example = \"asc\")\n\tprivate String defaultSortOrder;\n\n\t@Schema(title = \"是否默认记住密码\", example = \"true\")\n\tprivate Boolean defaultSavePwd;\n\n\t@Schema(title = \"普通下载是否启用确认弹窗\", example = \"true\")\n\tprivate Boolean enableNormalDownloadConfirm;\n\n\t@Schema(title = \"打包下载是否启用确认弹窗\", example = \"true\")\n\tprivate Boolean enablePackageDownloadConfirm;\n\n\t@Schema(title = \"批量下载是否启用确认弹窗\", example = \"true\")\n\tprivate Boolean enableBatchDownloadConfirm;\n\n\t/**\n\t * 废弃的字段，不再使用悬浮菜单\n\t */\n\t@Deprecated\n\t@Schema(title = \"是否启用 hover 菜单\", example = \"true\")\n\tprivate Boolean enableHoverMenu;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/model/result/FrontSiteConfigResult.java",
"content": "package im.zhaojun.zfile.module.config.model.result;\n\nimport im.zhaojun.zfile.module.config.model.dto.LinkExpireDTO;\nimport im.zhaojun.zfile.module.config.model.enums.FileClickModeEnum;\nimport im.zhaojun.zfile.module.user.model.enums.LoginLogModeEnum;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport java.util.List;\n\n/**\n * 全局站点设置响应类\n *\n * @author zhaojun\n */\n@Data\n@Schema(title=\"全局站点设置响应类\")\npublic class FrontSiteConfigResult {\n\n\t@Schema(title = \"是否已初始化\", example = \"true\")\n\tprivate Boolean installed;\n\n\t@Schema(title = \"Debug 模式\", example = \"true\", description =\"开启 debug 模式后，可重置管理员密码\")\n\tprivate Boolean debugMode;\n\n\t@Schema(title = \"直链地址前缀\", example = \"true\", description =\"直链地址前缀, 如 http(s)://ip:port/${直链前缀}/path/filename\")\n\tprivate String directLinkPrefix;\n\n\t@Schema(title = \"站点名称\", example = \"ZFile Site Name\")\n\tprivate String siteName;\n\n\t@Schema(title = \"备案号\", example = \"冀ICP备12345678号-1\")\n\tprivate String icp;\n\n\t@Schema(title = \"页面布局\", description =\"full:全屏,center:居中\", example = \"full\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate String layout;\n\n\t@Schema(title = \"移动端页面布局\", description =\"full:全屏,center:居中\", example = \"full\")\n\tprivate String mobileLayout;\n\n\t@Schema(title = \"移动端显示文件大小\", description = \"仅适用列表视图\", example = \"true\")\n\tprivate Boolean mobileShowSize;\n\n\t@Schema(title = \"列表尺寸\", description =\"large:大,default:中,small:小\", example = \"default\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate String tableSize;\n\n\t@Schema(title = \"是否显示生成直链功能（含直链和路径短链）\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean showLinkBtn;\n\n\t@Schema(title = \"是否显示生成短链功能\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean showShortLink;\n\n\t@Schema(title = \"是否显示生成路径链接功能\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean showPathLink;\n\n\t@Schema(title = \"是否显示文档区\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean showDocument;\n\n\t@Schema(title = \"是否显示网站公告\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean showAnnouncement;\n\n\t@Schema(title = \"网站公告\", example = \"ZFile 网站公告\")\n\tprivate String announcement;\n\n\t@Schema(title = \"自定义 JS\")\n\tprivate String customJs;\n\n\t@Schema(title = \"自定义 CSS\")\n\tprivate String customCss;\n\n\t@Schema(title = \"自定义视频文件后缀格式\")\n\tprivate String customVideoSuffix;\n\n\t@Schema(title = \"自定义图像文件后缀格式\")\n\tprivate String customImageSuffix;\n\n\t@Schema(title = \"自定义音频文件后缀格式\")\n\tprivate String customAudioSuffix;\n\n\t@Schema(title = \"自定义文本文件后缀格式\")\n\tprivate String customTextSuffix;\n\n\t@Schema(title = \"自定义Office后缀格式\")\n\tprivate String customOfficeSuffix;\n\n\t@Schema(title = \"自定义kkFileView后缀格式\")\n\tprivate String customKkFileViewSuffix;\n\n\t@Schema(title = \"根目录是否显示所有存储源\", description =\"勾选则根目录显示所有存储源列表, 反之会自动显示第一个存储源的内容.\", example = \"true\", requiredMode = Schema.RequiredMode.REQUIRED)\n\tprivate Boolean rootShowStorage;\n\n\t@Schema(title = \"强制后端地址\", description =\"强制指定生成直链，短链，获取回调地址时的地址。\", example = \"http://xxx.example.com\")\n\tprivate String forceBackendAddress;\n\n\t@Schema(title = \"前端域名\", description =\"前端域名，前后端分离情况下需要配置.\", example = \"http://xxx.example.com\")\n\tprivate String frontDomain;\n\n\t@Schema(title = \"是否在前台显示登陆按钮\", example = \"true\")\n\tprivate Boolean showLogin;\n\n\t@Schema(title = \"登录日志模式\", example = \"all\")\n\tprivate LoginLogModeEnum loginLogMode;\n\n\t@Schema(title = \"默认文件点击习惯\", example = \"click\")\n\tprivate FileClickModeEnum fileClickMode;\n\n\t@Schema(title = \"移动端默认文件点击习惯\", example = \"click\")\n\tprivate FileClickModeEnum mobileFileClickMode;\n\n\t@Schema(title = \"最大同时上传文件数\", example = \"5\")\n\tprivate Integer maxFileUploads;\n\n\t@Schema(title = \"onlyOffice 在线预览地址\", example = \"http://office.zfile.vip\")\n\tprivate String onlyOfficeUrl;\n\n\t@Schema(title = \"kkFileView 在线预览地址\", example = \"http://kkfile.zfile.vip\")\n\tprivate String kkFileViewUrl;\n\n\t@Schema(title = \"kkFileView 预览方式\", example = \"iframe/newTab\")\n\tprivate String kkFileViewOpenMode;\n\n\t@Schema(title = \"默认最大显示文件数\", example = \"1000\")\n\tprivate Integer maxShowSize;\n\n\t@Schema(title = \"每次加载更多文件数\", example = \"50\")\n\tprivate Integer loadMoreSize;\n\n\t@Schema(title = \"默认排序字段\", example = \"name\")\n\tprivate String defaultSortField;\n\n\t@Schema(title = \"默认排序方向\", example = \"asc\")\n\tprivate String defaultSortOrder;\n\n\t@Schema(title = \"站点 Home 名称\", example = \"xxx 的小站\")\n\tprivate String siteHomeName;\n\n\t@Schema(title = \"站点 Home Logo\", example = \"true\")\n\tprivate String siteHomeLogo;\n\n\t@Schema(title = \"站点 Logo 点击后链接\", example = \"https://www.zfile.vip\")\n\tprivate String siteHomeLogoLink;\n\n\t@Schema(title = \"站点 Logo 链接打开方式\", example = \"_blank\")\n\tprivate String siteHomeLogoTargetMode;\n\n\t@Schema(title = \"短链过期时间设置\", example = \"[{value: 1, unit: \\\"day\\\"}, {value: 1, unit: \\\"week\\\"}, {value: 1, unit: \\\"month\\\"}, {value: 1, unit: \\\"year\\\"}]\")\n\tprivate List linkExpireTimes;\n\n\t@Schema(title = \"是否默认记住密码\", example = \"true\")\n\tprivate Boolean defaultSavePwd;\n\n\t@Schema(title = \"普通下载是否启用确认弹窗\", example = \"true\")\n\tprivate Boolean enableNormalDownloadConfirm;\n\n\t@Schema(title = \"打包下载是否启用确认弹窗\", example = \"true\")\n\tprivate Boolean enablePackageDownloadConfirm;\n\n\t@Schema(title = \"批量下载是否启用确认弹窗\", example = \"true\")\n\tprivate Boolean enableBatchDownloadConfirm;\n\n\t/**\n\t * 废弃的字段，不再使用悬浮菜单\n\t */\n\t@Deprecated\n\t@Schema(title = \"是否启用 hover 菜单\", example = \"true\")\n\tprivate Boolean enableHoverMenu;\n\n\t@Schema(title = \"是否是游客\", example = \"true\")\n\tprivate boolean guest;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/config/service/SystemConfigService.java",
"content": "package im.zhaojun.zfile.module.config.service;\n\nimport cn.hutool.core.convert.Convert;\nimport cn.hutool.core.convert.ConvertException;\nimport cn.hutool.core.net.url.UrlBuilder;\nimport cn.hutool.core.util.EnumUtil;\nimport cn.hutool.core.util.HexUtil;\nimport cn.hutool.core.util.ObjUtil;\nimport cn.hutool.crypto.SecureUtil;\nimport cn.hutool.crypto.symmetric.SymmetricAlgorithm;\nimport com.alibaba.fastjson2.JSONArray;\nimport com.alibaba.fastjson2.JSONObject;\nimport im.zhaojun.zfile.core.util.*;\nimport im.zhaojun.zfile.module.config.annotation.JSONStringParse;\nimport im.zhaojun.zfile.module.config.constant.SystemConfigConstant;\nimport im.zhaojun.zfile.module.config.event.SystemConfigModifyHandlerChain;\nimport im.zhaojun.zfile.module.config.mapper.SystemConfigMapper;\nimport im.zhaojun.zfile.module.config.model.dto.SystemConfigDTO;\nimport im.zhaojun.zfile.module.config.model.entity.SystemConfig;\nimport im.zhaojun.zfile.module.user.model.enums.LoginVerifyModeEnum;\nimport jakarta.annotation.Resource;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.aop.framework.AopContext;\nimport org.springframework.cache.Cache;\nimport org.springframework.cache.CacheManager;\nimport org.springframework.cache.annotation.CacheConfig;\nimport org.springframework.cache.annotation.CacheEvict;\nimport org.springframework.cache.annotation.Cacheable;\nimport org.springframework.stereotype.Service;\nimport org.springframework.transaction.annotation.Transactional;\n\nimport java.lang.reflect.Field;\nimport java.util.*;\n\nimport static im.zhaojun.zfile.module.config.service.SystemConfigService.CACHE_NAME;\n\n\n/**\n * 系统设置 Service\n *\n * @author zhaojun\n */\n@Slf4j\n@Service\n@CacheConfig(cacheNames = CACHE_NAME)\npublic class SystemConfigService {\n\n public static final String CACHE_NAME = \"systemConfig\";\n\n private static final String SERIAL_VERSION_UID_FIELD_NAME = \"serialVersionUID\";\n\n @Resource\n private SystemConfigMapper systemConfigMapper;\n\n @Resource\n private CacheManager cacheManager;\n\n @Resource\n private SystemConfigModifyHandlerChain systemConfigModifyHandlerChain;\n\n private final Class systemConfigClazz = SystemConfigDTO.class;\n\n public static final List ignoreFieldList = Arrays.asList(\"domain\");\n\n /**\n * 获取系统设置, 如果缓存中有, 则去缓存取, 没有则查询数据库并写入到缓存中.\n *\n * @return 系统设置\n */\n @Cacheable(key = \"'dto'\")\n public SystemConfigDTO getSystemConfig() {\n SystemConfigDTO systemConfigDTO = new SystemConfigDTO();\n List systemConfigList = systemConfigMapper.findAll();\n\n for (SystemConfig systemConfig : systemConfigList) {\n String key = systemConfig.getName();\n if (ignoreFieldList.contains(key)) {\n if (log.isTraceEnabled()) {\n log.trace(\"从数据库加载字段填充到 DTO 时，忽略字段: {}\", key);\n }\n continue;\n }\n try {\n Field field = systemConfigClazz.getDeclaredField(key);\n field.setAccessible(true);\n String strVal = systemConfig.getValue();\n Class fieldType = field.getType();\n\n Object convertVal;\n if (EnumUtil.isEnum(fieldType)) {\n convertVal = EnumConvertUtils.convertStrToEnum(fieldType, strVal);\n } else if (field.isAnnotationPresent(JSONStringParse.class)) {\n // 如果类是 Collection 类型, 则需要将 JSON 字符串转换为 List\n if (Collection.class.isAssignableFrom(fieldType)) {\n Class genericType = ClassUtils.getGenericType(field);\n convertVal = JSONArray.parseArray(strVal, genericType);\n } else {\n // 否则转换为普通对象\n convertVal = JSONObject.parseObject(strVal, fieldType);\n }\n } else {\n convertVal = Convert.convert(fieldType, strVal);\n }\n field.set(systemConfigDTO, convertVal);\n } catch (NoSuchFieldException | IllegalAccessException | ConvertException e) {\n log.error(\"通过反射, 将字段 {} 注入 SystemConfigDTO 时出现异常:\", key, e);\n }\n }\n\n return systemConfigDTO;\n }\n\n\n /**\n * 更新系统设置, 并清空缓存中的内容.\n *\n * @param systemConfigDTO\n * 系统设置 dto\n */\n @Transactional(rollbackFor = Exception.class)\n @CacheEvict(allEntries = true)\n public synchronized void updateSystemConfig(SystemConfigDTO systemConfigDTO) {\n // 获取更新前的值\n List systemConfigListInDb = systemConfigMapper.findAll();\n Map systemConfigMapInDb = CollectionUtils.toMap(systemConfigListInDb, null, SystemConfig::getName);\n\n // 存储更新后的值\n List updateSystemConfigList = new ArrayList<>();\n\n Field[] fields = systemConfigClazz.getDeclaredFields();\n for (Field field : fields) {\n // 获取数据库中的值对象\n String key = field.getName();\n if (SERIAL_VERSION_UID_FIELD_NAME.equals(key)) {\n continue;\n }\n SystemConfig systemConfig = systemConfigMapInDb.get(key);\n if (systemConfig != null) {\n field.setAccessible(true);\n Object val = null;\n\n try {\n val = field.get(systemConfigDTO);\n } catch (IllegalAccessException e) {\n log.error(\"通过反射, 从 SystemConfigDTO 获取字段 {} 时出现异常:\", key, e);\n }\n\n if (val != null) {\n // 如果是枚举类型, 则取 value 值.\n if (EnumUtil.isEnum(val)) {\n val = EnumConvertUtils.convertEnumToStr(val);\n } else if (field.isAnnotationPresent(JSONStringParse.class)) {\n val = JSONObject.toJSONString(val);\n }\n // 如果和原来的值一样, 则跳过\n String originVal = systemConfig.getValue();\n if (ObjUtil.equals(originVal, val)) {\n continue;\n }\n // 将更新后的值存到更新列表中\n SystemConfig updateSystemConfig = new SystemConfig();\n updateSystemConfig.setId(systemConfig.getId());\n updateSystemConfig.setName(systemConfig.getName());\n updateSystemConfig.setValue(Convert.toStr(val));\n updateSystemConfig.setTitle(systemConfig.getTitle());\n updateSystemConfigList.add(updateSystemConfig);\n }\n } else {\n log.warn(\"尝试保存系统配置表中不存在字段: {}\", key);\n }\n }\n\n updateSystemConfigList.forEach(systemConfigInForm -> {\n SystemConfig systemConfigInDb = systemConfigMapInDb.get(systemConfigInForm.getName());\n systemConfigModifyHandlerChain.execute(systemConfigInDb, systemConfigInForm);\n systemConfigMapper.updateById(systemConfigInForm);\n });\n }\n\n\n /**\n * 获取 AES Hex 格式密钥\n *\n * @return AES Hex 格式密钥\n */\n public synchronized String getAesHexKeyOrGenerate() {\n SystemConfigDTO systemConfigDTO = ((SystemConfigService)AopContext.currentProxy()).getSystemConfig();\n String aesHexKey = systemConfigDTO.getRsaHexKey();\n if (StringUtils.isEmpty(aesHexKey)) {\n byte[] key = SecureUtil.generateKey(SymmetricAlgorithm.AES.getValue()).getEncoded();\n aesHexKey = HexUtil.encodeHexStr(key);\n\n SystemConfig loginVerifyModeConfig = systemConfigMapper.findByName(SystemConfigConstant.AES_HEX_KEY);\n loginVerifyModeConfig.setValue(aesHexKey);\n systemConfigMapper.updateById(loginVerifyModeConfig);\n systemConfigDTO.setRsaHexKey(aesHexKey);\n\n Cache cache = cacheManager.getCache(CACHE_NAME);\n Optional.ofNullable(cache).ifPresent(cache1 -> cache1.put(\"dto\", systemConfigDTO));\n }\n return aesHexKey;\n }\n\n\n /**\n * 获取前端站点域名\n *\n * @return 前端站点域名\n */\n public String getFrontDomain() {\n SystemConfigDTO systemConfigDTO = ((SystemConfigService)AopContext.currentProxy()).getSystemConfig();\n return systemConfigDTO.getFrontDomain();\n }\n\n\n /**\n * 获取实际的前端站点域名\n *\n * @return 实际的前端站点域名\n */\n public String getRealFrontDomain() {\n SystemConfigDTO systemConfigDTO = ((SystemConfigService)AopContext.currentProxy()).getSystemConfig();\n return StringUtils.firstNonNull(systemConfigDTO.getFrontDomain(), getAxiosFromDomainOrSetting(), RequestHolder.getOriginAddress());\n }\n\n\n /**\n * 优先级：\n * 1. 如果设置了强制后端地址，则使用强制后端地址。\n * 2. 如果请求中有 axios-from 参数，则使用该参数。\n * 3. 如果没有强制后端地址和 axios-from 参数，则使用请求的服务器地址（如果经过多个代理，可能不是实际的后端地址）。\n *\n * @return 后端站点地址\n */\n public String getAxiosFromDomainOrSetting() {\n SystemConfigDTO systemConfigDTO = ((SystemConfigService)AopContext.currentProxy()).getSystemConfig();\n if (StringUtils.isNotBlank(systemConfigDTO.getForceBackendAddress())) {\n return systemConfigDTO.getForceBackendAddress();\n } else if (StringUtils.isNotEmpty(RequestHolder.getAxiosFrom())) {\n return RequestHolder.getAxiosFrom();\n } else {\n return RequestHolder.getRequestServerAddress();\n }\n }\n\n /**\n * 获取前端地址下的 401 页面地址.\n *\n * @return 前端地址下的 401 页面地址.\n *\n */\n public String getUnauthorizedUrl() {\n return getUnauthorizedUrl(null, null);\n }\n\n /**\n * 获取前端地址下的 401 页面地址. 可以指定 code 和 message.\n *\n * @param code\n * 指定错误码\n *\n * @param message\n * 指定错误信息\n *\n * @return 前端地址下的 401 页面地址.\n */\n public String getUnauthorizedUrl(String code, String message) {\n String url = StringUtils.concat(getRealFrontDomain(), \"/401\");\n UrlBuilder urlBuilder = UrlBuilder.of(url);\n if (StringUtils.isNotBlank(code)) {\n urlBuilder.addQuery(\"code\", code);\n }\n if (StringUtils.isNotBlank(message)) {\n urlBuilder.addQuery(\"message\", message);\n }\n return urlBuilder.build();\n }\n\n /**\n * 获取前端地址下的 403 页面地址.\n *\n * @return 前端地址下的 403 页面地址.\n *\n */\n public String getForbiddenUrl() {\n return getForbiddenUrl(null, null);\n }\n\n /**\n * 获取前端地址下的 403 页面地址. 可以指定 code 和 message.\n *\n * @param code\n * 指定错误码\n *\n * @param message\n * 指定错误信息\n *\n * @return 前端地址下的 403 页面地址.\n */\n public String getForbiddenUrl(String code, String message) {\n String url = StringUtils.concat(getRealFrontDomain(), \"/403\");\n UrlBuilder urlBuilder = UrlBuilder.of(url);\n if (StringUtils.isNotBlank(code)) {\n urlBuilder.addQuery(\"code\", code);\n }\n if (StringUtils.isNotBlank(message)) {\n urlBuilder.addQuery(\"message\", message);\n }\n return urlBuilder.build();\n }\n\n /**\n * 获取前端地址下的 404 页面地址.\n *\n * @return 前端地址下的 404 页面地址.\n *\n */\n public String getNotFoundUrl() {\n return getNotFoundUrl(null, null);\n }\n\n /**\n * 获取前端地址下的 404 页面地址. 可以指定 code 和 message.\n *\n * @param code\n * 指定错误码\n *\n * @param message\n * 指定错误信息\n *\n * @return 前端地址下的 404 页面地址.\n */\n public String getNotFoundUrl(String code, String message) {\n String url = StringUtils.concat(getRealFrontDomain(), \"/404\");\n UrlBuilder urlBuilder = UrlBuilder.of(url);\n if (StringUtils.isNotBlank(code)) {\n urlBuilder.addQuery(\"code\", code);\n }\n if (StringUtils.isNotBlank(message)) {\n urlBuilder.addQuery(\"message\", message);\n }\n return urlBuilder.build();\n }\n\n /**\n * 获取前端地址下的 500 页面地址. 可以指定 code 和 message.\n *\n * @param code\n * 指定错误码\n *\n * @param message\n * 指定错误信息\n *\n * @return 前端地址下的 500 页面地址.\n */\n public String getErrorPageUrl(String code, String message) {\n String url = StringUtils.concat(getRealFrontDomain(), \"/500\");\n UrlBuilder urlBuilder = UrlBuilder.of(url);\n if (StringUtils.isNotBlank(code)) {\n urlBuilder.addQuery(\"code\", code);\n }\n if (StringUtils.isNotBlank(message)) {\n urlBuilder.addQuery(\"message\", message);\n }\n return urlBuilder.build();\n }\n\n\n /**\n * 重置登录验证模式，去除所有登录额外验证方式.\n */\n public void resetLoginVerifyMode() {\n SystemConfigDTO systemConfigDTO = ((SystemConfigService)AopContext.currentProxy()).getSystemConfig();\n systemConfigDTO.setLoginImgVerify(false);\n systemConfigDTO.setAdminTwoFactorVerify(false);\n systemConfigDTO.setLoginVerifySecret(\"\");\n systemConfigDTO.setLoginVerifyMode(LoginVerifyModeEnum.OFF_MODE);\n ((SystemConfigService)AopContext.currentProxy()).updateSystemConfig(systemConfigDTO);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/filter/controller/StorageSourceFilterController.java",
"content": "package im.zhaojun.zfile.module.filter.controller;\n\nimport com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\nimport com.github.xiaoymin.knife4j.annotations.ApiSort;\nimport im.zhaojun.zfile.core.annotation.DemoDisable;\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.module.filter.model.entity.FilterConfig;\nimport im.zhaojun.zfile.module.filter.service.FilterConfigService;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.Parameter;\nimport io.swagger.v3.oas.annotations.enums.ParameterIn;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport org.springframework.web.bind.annotation.*;\n\nimport java.util.List;\n\n/**\n * 存储源过滤器维护接口\n *\n * @author zhaojun\n */\n@Tag(name = \"存储源模块-过滤文件\")\n@ApiSort(6)\n@RestController\n@RequestMapping(\"/admin\")\npublic class StorageSourceFilterController {\n\n @Resource\n private FilterConfigService filterConfigService;\n\n @ApiOperationSupport(order = 1)\n @Operation(summary = \"获取存储源过滤文件列表\", description =\"根据存储源 ID 获取存储源设置的过滤文件列表\")\n @Parameter(in = ParameterIn.PATH, name = \"storageId\", description = \"存储源 id\", required = true, schema = @Schema(type = \"integer\"))\n @GetMapping(\"/storage/{storageId}/filters\")\n public AjaxJson> getFilters(@PathVariable Integer storageId) {\n return AjaxJson.getSuccessData(filterConfigService.findByStorageId(storageId));\n }\n\n\n @ApiOperationSupport(order = 2)\n @Operation(summary = \"保存存储源过滤文件列表\", description =\"保存指定存储源 ID 设置的过滤文件列表\")\n @Parameter(in = ParameterIn.PATH, name = \"storageId\", description = \"存储源 id\", required = true, schema = @Schema(type = \"integer\"))\n @PostMapping(\"/storage/{storageId}/filters\")\n @DemoDisable\n public AjaxJson saveFilters(@PathVariable Integer storageId, @RequestBody List filter) {\n filterConfigService.batchSave(storageId, filter);\n return AjaxJson.getSuccess();\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/filter/mapper/FilterConfigMapper.java",
"content": "package im.zhaojun.zfile.module.filter.mapper;\n\nimport com.baomidou.mybatisplus.core.mapper.BaseMapper;\nimport im.zhaojun.zfile.module.filter.model.entity.FilterConfig;\nimport org.apache.ibatis.annotations.Mapper;\nimport org.apache.ibatis.annotations.Param;\n\nimport java.util.List;\n\n/**\n * 过滤器配置表 Mapper 接口\n *\n * @author zhaojun\n */\n@Mapper\npublic interface FilterConfigMapper extends BaseMapper {\n\n /**\n * 根据存储源 ID 获取存储源配置列表\n *\n * @param storageId\n * 存储源 ID\n *\n * @return 存储源过滤器配置列表\n */\n List findByStorageId(@Param(\"storageId\") Integer storageId);\n\n\n /**\n * 根据存储源 ID 删除过滤器配置\n *\n * @param storageId\n * 存储源 ID\n *\n * @return 删除条数\n */\n int deleteByStorageId(@Param(\"storageId\") Integer storageId);\n\n\n /**\n * 获取所有类型为禁止访问的过滤规则\n *\n * @param storageId\n * 存储 ID\n *\n * @return 禁止访问的过滤规则列表\n */\n List findByStorageIdAndInaccessible(@Param(\"storageId\")Integer storageId);\n\n\n /**\n * 获取所有类型为禁止下载的过滤规则\n *\n * @param storageId\n * 存储 ID\n *\n * @return 禁止下载的过滤规则列表\n */\n List findByStorageIdAndDisableDownload(@Param(\"storageId\")Integer storageId);\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/filter/model/entity/FilterConfig.java",
"content": "package im.zhaojun.zfile.module.filter.model.entity;\n\nimport com.baomidou.mybatisplus.annotation.IdType;\nimport com.baomidou.mybatisplus.annotation.TableField;\nimport com.baomidou.mybatisplus.annotation.TableId;\nimport com.baomidou.mybatisplus.annotation.TableName;\nimport com.fasterxml.jackson.annotation.JsonIgnore;\nimport im.zhaojun.zfile.module.filter.model.enums.FilterConfigHiddenModeEnum;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport java.io.Serializable;\n\n/**\n * 存储源过滤配置 entity\n *\n * @author zhaojun\n */\n@Data\n@Schema(title=\"存储源过滤配置\")\n@TableName(value = \"filter_config\")\npublic class FilterConfig implements Serializable {\n\n private static final long serialVersionUID = 1L;\n\n @TableId(value = \"id\", type = IdType.AUTO)\n @Schema(title = \"ID, 新增无需填写\", example = \"1\")\n private Integer id;\n\n\n @TableField(value = \"storage_id\")\n @Schema(title = \"存储源 ID\", requiredMode = Schema.RequiredMode.REQUIRED, example = \"1\")\n private Integer storageId;\n\n\n @TableField(value = \"expression\")\n @Schema(title = \"过滤表达式\", requiredMode = Schema.RequiredMode.REQUIRED, example = \"/*.png\")\n private String expression;\n\n\n @TableField(value = \"description\")\n @Schema(title = \"表达式描述\", requiredMode = Schema.RequiredMode.REQUIRED, example = \"用来辅助记忆表达式\")\n private String description;\n\n\n @TableField(value = \"mode\")\n @Schema(title = \"模式\", requiredMode = Schema.RequiredMode.REQUIRED, example = \"隐藏模式，仅隐藏: hidden, 隐藏后不可访问: inaccessible, 隐藏后不可下载: disable_download\")\n private FilterConfigHiddenModeEnum mode;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/filter/model/enums/FilterConfigHiddenModeEnum.java",
"content": "package im.zhaojun.zfile.module.filter.model.enums;\n\nimport com.baomidou.mybatisplus.annotation.EnumValue;\nimport com.fasterxml.jackson.annotation.JsonValue;\nimport lombok.AllArgsConstructor;\nimport lombok.Getter;\n\n/**\n * 文件夹隐藏模式枚举\n *\n * @author zhaojun\n */\n@Getter\n@AllArgsConstructor\npublic enum FilterConfigHiddenModeEnum {\n\n\t/**\n\t * 仅隐藏\n\t */\n\tHIDDEN(\"hidden\"),\n\n\t/**\n\t * 隐藏并不可访问 (针对目录)\n\t */\n\tINACCESSIBLE(\"inaccessible\"),\n\n\t/**\n\t * 隐藏并不可访问 (针对文件)\n\t */\n\tDISABLE_DOWNLOAD(\"disable_download\");\n\n\t@EnumValue\n\t@JsonValue\n\tprivate final String value;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/filter/service/FilterConfigService.java",
"content": "package im.zhaojun.zfile.module.filter.service;\n\nimport im.zhaojun.zfile.core.util.CollectionUtils;\nimport im.zhaojun.zfile.core.util.FileUtils;\nimport im.zhaojun.zfile.core.util.PatternMatcherUtils;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.module.filter.mapper.FilterConfigMapper;\nimport im.zhaojun.zfile.module.filter.model.entity.FilterConfig;\nimport im.zhaojun.zfile.module.storage.event.StorageSourceCopyEvent;\nimport im.zhaojun.zfile.module.storage.event.StorageSourceDeleteEvent;\nimport im.zhaojun.zfile.module.storage.model.enums.FileOperatorTypeEnum;\nimport im.zhaojun.zfile.module.user.service.UserStorageSourceService;\nimport jakarta.annotation.Resource;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.aop.framework.AopContext;\nimport org.springframework.beans.BeanUtils;\nimport org.springframework.cache.annotation.CacheConfig;\nimport org.springframework.cache.annotation.CacheEvict;\nimport org.springframework.cache.annotation.Cacheable;\nimport org.springframework.cache.annotation.Caching;\nimport org.springframework.context.event.EventListener;\nimport org.springframework.stereotype.Service;\nimport org.springframework.transaction.annotation.Transactional;\n\nimport java.util.List;\n\n/**\n * 存储源过滤规则 Service\n *\n * @author zhaojun\n */\n@Slf4j\n@Service\n@CacheConfig(cacheNames = \"filterConfig\")\npublic class FilterConfigService {\n\n @Resource\n private FilterConfigMapper filterConfigMapper;\n\n @Resource\n private UserStorageSourceService userStorageSourceService;\n\n /**\n * 根据存储源 ID 获取存储源配置列表\n *\n * @param storageId\n * 存储源 ID\n *\n * @return 存储源过滤规则配置列表\n */\n @Cacheable(key = \"'filter-base-' + #storageId\",\n condition = \"#storageId != null\")\n public List findByStorageId(Integer storageId) {\n return filterConfigMapper.findByStorageId(storageId);\n }\n\n\n /**\n * 获取所有类型为禁止访问的过滤规则\n *\n * @param storageId\n * 存储 ID\n *\n * @return 禁止访问的过滤规则列表\n */\n @Cacheable(key = \"'filter-inaccessible-' + #storageId\",\n condition = \"#storageId != null\")\n public List findByStorageIdAndInaccessible(Integer storageId) {\n return filterConfigMapper.findByStorageIdAndInaccessible(storageId);\n }\n\n\n /**\n * 获取所有类型为禁止下载的过滤规则\n *\n * @param storageId\n * 存储 ID\n *\n * @return 禁止下载的过滤规则列表\n */\n @Cacheable(key = \"'filter-disable-download-' + #storageId\",\n condition = \"#storageId != null\")\n public List findByStorageIdAndDisableDownload(Integer storageId) {\n return filterConfigMapper.findByStorageIdAndDisableDownload(storageId);\n }\n\n\n /**\n * 批量保存存储源过滤规则配置, 会先删除之前的所有配置(在事务中运行)\n *\n * @param storageId\n * 存储源 ID\n *\n * @param filterConfigList\n * 存储源过滤规则配置列表\n */\n @Transactional(rollbackFor = Exception.class)\n public void batchSave(Integer storageId, List filterConfigList) {\n ((FilterConfigService) AopContext.currentProxy()).deleteByStorageId(storageId);\n log.info(\"更新存储源 ID 为 {} 的过滤规则 {} 条\", storageId, filterConfigList.size());\n\n for (FilterConfig filterConfig : filterConfigList) {\n filterConfig.setId(null);\n filterConfig.setStorageId(storageId);\n filterConfigMapper.insert(filterConfig);\n\n if (log.isDebugEnabled()) {\n log.debug(\"新增过滤规则, 存储源 ID: {}, 表达式: {}, 描述: {}, 隐藏模式: {}\",\n filterConfig.getStorageId(), filterConfig.getExpression(),\n filterConfig.getDescription(), filterConfig.getMode().getValue());\n }\n }\n }\n\n\n /**\n * 根据存储源 ID 删除所有过滤规则配置\n *\n * @param storageId\n * 存储源 ID\n */\n @Caching(evict = {\n @CacheEvict(key = \"'filter-base-' + #storageId\"),\n @CacheEvict(key = \"'filter-inaccessible-' + #storageId\"),\n @CacheEvict(key = \"'filter-disable-download-' + #storageId\")\n })\n public int deleteByStorageId(Integer storageId) {\n int deleteSize = filterConfigMapper.deleteByStorageId(storageId);\n log.info(\"删除存储源 ID 为 {} 的过滤规则 {} 条\", storageId, deleteSize);\n return deleteSize;\n }\n\n /**\n * 监听存储源删除事件，根据存储源 id 删除相关的过滤条件设置\n *\n * @param storageSourceDeleteEvent\n * 存储源删除事件\n */\n @EventListener\n public void onStorageSourceDelete(StorageSourceDeleteEvent storageSourceDeleteEvent) {\n Integer storageId = storageSourceDeleteEvent.getId();\n int updateRows = ((FilterConfigService) AopContext.currentProxy()).deleteByStorageId(storageId);\n if (log.isDebugEnabled()) {\n log.debug(\"删除存储源 [id {}, name: {}, type: {}] 时，关联删除存储源过滤条件设置 {} 条\",\n storageId,\n storageSourceDeleteEvent.getName(),\n storageSourceDeleteEvent.getType().getDescription(),\n updateRows);\n }\n\n }\n\n /**\n * 判断访问的路径是否是不允许访问的\n *\n * @param storageId\n * 存储源 ID\n *\n * @param path\n * 请求路径\n *\n */\n public boolean checkFileIsInaccessible(Integer storageId, String path) {\n List filterConfigList = ((FilterConfigService) AopContext.currentProxy()).findByStorageIdAndInaccessible(storageId);\n return testPattern(storageId, filterConfigList, path);\n }\n\n\n /**\n * 指定存储源下的文件名称, 根据过滤表达式判断是否会显示, 如果符合任意一条表达式, 表示隐藏则返回 true, 反之表示不隐藏则返回 false.\n *\n * @param storageId\n * 存储源 ID\n *\n * @param fileName\n * 文件名\n *\n * @return 是否是隐藏文件夹\n */\n public boolean checkFileIsHidden(Integer storageId, String fileName) {\n List filterConfigList = ((FilterConfigService) AopContext.currentProxy()).findByStorageId(storageId);\n return testPattern(storageId, filterConfigList, fileName);\n }\n\n\n /**\n * 指定存储源下的文件名称, 根据过滤表达式判断文件名和所在路径是否禁止下载, 如果符合任意一条表达式, 则返回 true, 反之则返回 false.\n *\n * @param storageId\n * 存储源 ID\n *\n * @param fileName\n * 文件名\n *\n * @return 是否显示\n */\n public boolean checkFileIsDisableDownload(Integer storageId, String fileName) {\n List filterConfigList = ((FilterConfigService) AopContext.currentProxy()).findByStorageIdAndDisableDownload(storageId);\n String filePath = FileUtils.getParentPath(fileName);\n if (StringUtils.isEmpty(filePath)) {\n return testPattern(storageId, filterConfigList, fileName);\n } else {\n return testPattern(storageId, filterConfigList, fileName) || testPattern(storageId, filterConfigList, filePath);\n }\n }\n\n\n /**\n * 根据规则表达式和测试字符串进行匹配，如测试字符串和其中一个规则匹配上，则返回 true，反之返回 false。\n *\n * @param patternList\n * 规则列表\n *\n * @param test\n *\n * 测试字符串\n *\n * @return 是否显示\n */\n private boolean testPattern(Integer storageId, List patternList, String test) {\n // 如果规则列表为空, 则表示不需要过滤, 直接返回 false\n if (CollectionUtils.isEmpty(patternList)) {\n if (log.isDebugEnabled()) {\n log.debug(\"过滤规则列表为空, 存储源 ID: {}, 测试字符串: {}\", storageId, test);\n }\n return false;\n }\n\n // 判断是否需要忽略文件隐藏校验\n boolean isIgnoreHidden = userStorageSourceService.hasCurrentUserStorageOperatorPermission(storageId, FileOperatorTypeEnum.IGNORE_HIDDEN);\n if (isIgnoreHidden) {\n if (log.isDebugEnabled()) {\n log.debug(\"权限配置忽略过滤规则, 存储源 ID: {}, 测试字符串: {}\", storageId, test);\n }\n return false;\n }\n\n // 校验表达式\n for (FilterConfig filterConfig : patternList) {\n String expression = filterConfig.getExpression();\n\n if (StringUtils.isEmpty(expression)) {\n if (log.isDebugEnabled()) {\n log.debug(\"存储源 {} 过滤文件测试表达式: {}, 测试字符串: {}, 表达式为空，跳过该规则校验\", storageId, expression, test);\n }\n continue;\n }\n\n try {\n boolean match = PatternMatcherUtils.testCompatibilityGlobPattern(expression, test);\n\n if (log.isDebugEnabled()) {\n log.debug(\"存储源 {} 过滤文件测试表达式: {}, 测试字符串: {}, 匹配结果: {}\", storageId, expression, test, match);\n }\n\n if (match) {\n return true;\n }\n } catch (Exception e) {\n log.error(\"存储源 {} 过滤文件测试表达式: {}, 测试字符串: {}, 匹配异常，跳过该规则.\", storageId, expression, test, e);\n }\n }\n\n return false;\n }\n\n /**\n * 监听存储源复制事件, 复制存储源的过滤条件设置到新的存储源\n *\n * @param storageSourceCopyEvent\n * 存储源复制事件\n */\n @EventListener\n public void onStorageSourceCopy(StorageSourceCopyEvent storageSourceCopyEvent) {\n Integer fromId = storageSourceCopyEvent.getFromId();\n Integer newId = storageSourceCopyEvent.getNewId();\n\n List filterConfigList = ((FilterConfigService) AopContext.currentProxy()).findByStorageId(fromId);\n\n filterConfigList.forEach(filterConfig -> {\n FilterConfig newFilterConfig = new FilterConfig();\n BeanUtils.copyProperties(filterConfig, newFilterConfig);\n newFilterConfig.setId(null);\n newFilterConfig.setStorageId(newId);\n filterConfigMapper.insert(newFilterConfig);\n });\n\n log.info(\"复制存储源 ID 为 {} 的存储源过滤条件设置到存储源 ID 为 {} 成功, 共 {} 条\", fromId, newId, filterConfigList.size());\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/install/controller/InstallController.java",
"content": "package im.zhaojun.zfile.module.install.controller;\n\nimport com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\nimport im.zhaojun.zfile.core.annotation.DemoDisable;\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.module.install.model.request.InstallSystemRequest;\nimport im.zhaojun.zfile.module.install.service.InstallService;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport org.springframework.web.bind.annotation.*;\n\n/**\n * 系统初始化接口\n *\n * @author zhaojun\n */\n@Tag(name = \"初始化模块\")\n@RestController\n@RequestMapping(\"/api\")\npublic class InstallController {\n\n @Resource\n private InstallService installService;\n\n @GetMapping(\"/install/status\")\n @ApiOperationSupport(order = 1)\n @Operation(summary = \"获取系统初始化状态\", description = \"根据管理员用户名是否存在判断系统已初始化, 已初始化返回 true, 未初始化返回 false\")\n public AjaxJson isInstall() {\n return AjaxJson.getSuccessData(installService.getSystemIsInstalled());\n }\n\n @ApiOperationSupport(order = 2)\n @Operation(summary = \"初始化系统\", description = \"根据管理员用户名是否存在判断系统已初始化, 已初始化返回 true, 未初始化返回 false\")\n @PostMapping(\"/install\")\n @DemoDisable\n public AjaxJson install(@RequestBody InstallSystemRequest installSystemRequest) {\n installService.install(installSystemRequest);\n return AjaxJson.getSuccess();\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/install/model/request/InstallSystemRequest.java",
"content": "package im.zhaojun.zfile.module.install.model.request;\n\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\n/**\n * 系统初始化请求参数\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"系统初始化请求类\")\npublic class InstallSystemRequest {\n\n @Schema(title = \"站点名称\", example = \"ZFile Site Name\")\n private String siteName;\n\n @Schema(title = \"用户名\", example = \"admin\")\n private String username;\n\n @Schema(title = \"密码\", example = \"123456\")\n private String password;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/install/service/InstallService.java",
"content": "package im.zhaojun.zfile.module.install.service;\n\nimport im.zhaojun.zfile.core.exception.ErrorCode;\nimport im.zhaojun.zfile.core.exception.core.BizException;\nimport im.zhaojun.zfile.core.exception.core.SystemException;\nimport im.zhaojun.zfile.module.config.model.dto.SystemConfigDTO;\nimport im.zhaojun.zfile.module.install.model.request.InstallSystemRequest;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.user.service.UserService;\nimport jakarta.annotation.Resource;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.stereotype.Service;\nimport org.springframework.transaction.annotation.Transactional;\n\n@Service\n@Slf4j\npublic class InstallService {\n\n @Resource\n private SystemConfigService systemConfigService;\n\n @Resource\n private UserService userService;\n\n @Transactional(rollbackFor = Exception.class)\n public void install(InstallSystemRequest installSystemRequest) {\n if (getSystemIsInstalled()) {\n throw new BizException(ErrorCode.BIZ_SYSTEM_ALREADY_INIT);\n }\n\n boolean updateFlag = userService.initAdminUser(installSystemRequest.getUsername(),\n installSystemRequest.getPassword());\n if (!updateFlag) {\n throw new SystemException(ErrorCode.BIZ_SYSTEM_INIT_ERROR);\n }\n\n SystemConfigDTO systemConfigDTO = new SystemConfigDTO();\n systemConfigDTO.setSiteName(installSystemRequest.getSiteName());\n systemConfigDTO.setInstalled(true);\n systemConfigService.updateSystemConfig(systemConfigDTO);\n }\n\n /**\n * 获取系统是否已初始化\n *\n * @return 管理员名称\n */\n public Boolean getSystemIsInstalled() {\n return systemConfigService.getSystemConfig().getInstalled();\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/aspect/LinkRateLimiterAspect.java",
"content": "package im.zhaojun.zfile.module.link.aspect;\n\nimport cn.hutool.extra.servlet.JakartaServletUtil;\nimport im.zhaojun.zfile.core.exception.ErrorCode;\nimport im.zhaojun.zfile.core.exception.core.BizException;\nimport im.zhaojun.zfile.module.config.model.dto.SystemConfigDTO;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.link.cache.LinkRateLimiterCache;\nimport im.zhaojun.zfile.module.storage.annotation.LinkRateLimiter;\nimport jakarta.annotation.Resource;\nimport jakarta.servlet.http.HttpServletRequest;\nimport lombok.extern.slf4j.Slf4j;\nimport org.aspectj.lang.ProceedingJoinPoint;\nimport org.aspectj.lang.annotation.Around;\nimport org.aspectj.lang.annotation.Aspect;\nimport org.springframework.stereotype.Component;\n\nimport java.util.concurrent.atomic.AtomicInteger;\n\n/**\n * 校验直链访问频率.\n * \n * 校验所有标注了 {@link LinkRateLimiter} 的注解\n *\n * @author zhaojun\n */\n@Aspect\n@Component\n@Slf4j\npublic class LinkRateLimiterAspect {\n\n\t@Resource\n\tprivate HttpServletRequest httpServletRequest;\n\n\t@Resource\n\tprivate SystemConfigService systemConfigService;\n\n\t@Resource\n\tprivate LinkRateLimiterCache linkRateLimiterCache;\n\n\t/**\n\t * 校验直链访问频率.\n\t *\n\t * @param point\n\t * 连接点\n\t *\n\t * @return 方法运行结果\n\t */\n\t@Around(value = \"@annotation(im.zhaojun.zfile.module.storage.annotation.LinkRateLimiter)\")\n\tpublic Object around(ProceedingJoinPoint point) throws Throwable {\n\t\tSystemConfigDTO systemConfig = systemConfigService.getSystemConfig();\n\t\tInteger linkLimitSecond = systemConfig.getLinkLimitSecond();\n\t\tInteger linkDownloadLimit = systemConfig.getLinkDownloadLimit();\n\t\t// 如果未设置直链限制, 则不进行校验\n\t\tif (linkLimitSecond == null || linkDownloadLimit == null || linkLimitSecond == 0 || linkDownloadLimit == 0) {\n\t\t\treturn point.proceed();\n\t\t}\n\n\t\tString clientIP = JakartaServletUtil.getClientIP(httpServletRequest);\n\t\tif (linkRateLimiterCache.containsKey(clientIP)) {\n\t\t\tAtomicInteger atomicInteger = linkRateLimiterCache.get(clientIP, false);\n\t\t\tif (atomicInteger.incrementAndGet() > linkDownloadLimit) {\n\t\t\t\tthrow new BizException(ErrorCode.BIZ_ACCESS_TOO_FREQUENT);\n\t\t\t}\n\t\t} else {\n\t\t\tlinkRateLimiterCache.put(clientIP, new AtomicInteger(1), linkLimitSecond * 1000);\n\t\t}\n\n\t\treturn point.proceed();\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/aspect/RefererCheckAspect.java",
"content": "package im.zhaojun.zfile.module.link.aspect;\n\nimport im.zhaojun.zfile.core.util.CollectionUtils;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.module.config.model.dto.SystemConfigDTO;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.link.model.enums.RefererTypeEnum;\nimport im.zhaojun.zfile.module.storage.annotation.RefererCheck;\nimport jakarta.annotation.Resource;\nimport jakarta.servlet.http.HttpServletRequest;\nimport jakarta.servlet.http.HttpServletResponse;\nimport lombok.extern.slf4j.Slf4j;\nimport org.aspectj.lang.ProceedingJoinPoint;\nimport org.aspectj.lang.annotation.Around;\nimport org.aspectj.lang.annotation.Aspect;\nimport org.springframework.http.HttpHeaders;\nimport org.springframework.stereotype.Component;\nimport org.springframework.util.AntPathMatcher;\n\nimport java.util.Collection;\nimport java.util.List;\n\n/**\n * 校验 referer 防盗链.\n *
\n * 校验所有标注了 {@link RefererCheck} 的注解\n *\n * @author zhaojun\n */\n@Aspect\n@Component\n@Slf4j\npublic class RefererCheckAspect {\n\n\t@Resource\n\tprivate HttpServletRequest httpServletRequest;\n\n\t@Resource\n\tprivate HttpServletResponse httpServletResponse;\n\n\t@Resource\n\tprivate SystemConfigService systemConfigService;\n\n\tprivate final AntPathMatcher pathMatcher = new AntPathMatcher();\n\n\t/**\n\t * 校验 referer 防盗链.\n\t *\n\t * @param point\n\t * 连接点\n\t *\n\t * @return 方法运行结果\n\t */\n\t@Around(value = \"@annotation(im.zhaojun.zfile.module.storage.annotation.RefererCheck)\")\n\tpublic Object around(ProceedingJoinPoint point) throws Throwable {\n\t\t// 获取配置的 referer 类型\n\t\tSystemConfigDTO systemConfig = systemConfigService.getSystemConfig();\n\t\tRefererTypeEnum refererType = systemConfig.getRefererType();\n\n\t\t// 如果未开启 referer 防盗链则跳过.\n\t\tif (refererType == RefererTypeEnum.OFF) {\n\t\t\treturn point.proceed();\n\t\t}\n\n\t\t// 获取当前请求 referer\n\t\tString referer = httpServletRequest.getHeader(HttpHeaders.REFERER);\n\t\tString requestUrl = httpServletRequest.getRequestURI();\n\n\t\t// 获取 Forbidden 页面地址\n\t\tString forbiddenUrl = systemConfigService.getForbiddenUrl();\n\n\t\t// 如果 referer 不允许为空，且当前 referer 为空，则校验\n\t\tBoolean refererAllowEmpty = systemConfig.getRefererAllowEmpty();\n\t\tif (!refererAllowEmpty && StringUtils.isEmpty(referer)) {\n\t\t\tlog.warn(\"请求路径 {}, referer 不允许为空，当前请求 referer 为空，禁止访问.\", requestUrl);\n\t\t\thttpServletResponse.sendRedirect(forbiddenUrl);\n\t\t\treturn null;\n\t\t} else if (refererAllowEmpty && StringUtils.isEmpty(referer)) { // 如果 referer 允许为空，且当前 referer 为空，则跳过校验\n\t\t\treturn point.proceed();\n\t\t}\n\n\t\t// 获取允许的 referer 地址\n\t\tString refererValue = systemConfig.getRefererValue();\n\t\tList refererValueList = StringUtils.split(refererValue, StringUtils.LF);\n\n\t\t// 如果是白名单模式，则校验当前 referer, 如果未在允许的列表中，则禁止访问.\n\t\tif (refererType == RefererTypeEnum.WHITE_LIST && containsPathMatcher(refererValueList, referer) == null) {\n\t\t\tlog.warn(\"请求路径 {}, referer 为白名单模式，当前请求 referer {} 未在白名单中，禁止访问.\", requestUrl, referer);\n\t\t\thttpServletResponse.sendRedirect(forbiddenUrl);\n\t\t\treturn null;\n\t\t}\n\n\t\t// 如果是黑名单模式，则校验当前 referer 是否在列表中，则禁止访问.\n\t\tif (refererType == RefererTypeEnum.BLACK_LIST && containsPathMatcher(refererValueList, referer) != null) {\n\t\t\tlog.warn(\"请求路径 {}, referer 为黑名单模式，当前请求 referer {} 在黑名单中，禁止访问.\", requestUrl, referer);\n\n\t\t\thttpServletResponse.sendRedirect(forbiddenUrl);\n\t\t\treturn null;\n\t\t}\n\n\t\treturn point.proceed();\n\t}\n\n\t/**\n\t * 校验 value 是否在 Ant 表达式列表中.\n\t *\n\t * @param patternList\n\t * Ant 表达式列表\n\t *\n\t * @param value\n\t * 要校验的值\n\t *\n\t * @return 返回匹配到的规则项，如果没有匹配到，则返回 null.\n\t */\n\tpublic String containsPathMatcher(Collection patternList, String value) {\n\t\tif (CollectionUtils.isEmpty(patternList)) {\n\t\t\treturn null;\n\t\t}\n\n\t\tfor (String pattern : patternList) {\n\t\t\tif (pathMatcher.match(pattern, value)) {\n\t\t\t\treturn pattern;\n\t\t\t}\n\t\t}\n\n\t\treturn null;\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/cache/LinkRateLimiterCache.java",
"content": "package im.zhaojun.zfile.module.link.cache;\n\nimport cn.hutool.cache.impl.CacheObj;\nimport cn.hutool.cache.impl.TimedCache;\nimport im.zhaojun.zfile.module.link.model.dto.CacheInfo;\nimport org.springframework.stereotype.Service;\n\nimport java.util.ArrayList;\nimport java.util.Iterator;\nimport java.util.List;\nimport java.util.concurrent.atomic.AtomicInteger;\n\n/**\n * 直链/短链访问频率限制缓存\n */\n@Service\npublic class LinkRateLimiterCache {\n\n /**\n * cache 在 put 时不指定 timeout, 则使用默认的 timeout. (单位: 毫秒)\n */\n public static final Integer DEFAULT_TIME_OUT = 60_000;\n\n private final TimedCache timedCache = new TimedCache<>(DEFAULT_TIME_OUT);\n\n public boolean containsKey(String key) {\n return timedCache.containsKey(key);\n }\n\n public AtomicInteger get(String key, boolean isUpdateLastAccess) {\n return timedCache.get(key, isUpdateLastAccess);\n }\n\n public void put(String key, AtomicInteger object, long timeout) {\n timedCache.put(key, object, timeout);\n }\n\n public List> getCacheInfo() {\n List> cacheInfoList = new ArrayList<>();\n\n Iterator> cacheObjIterator = timedCache.cacheObjIterator();\n while (cacheObjIterator.hasNext()) {\n CacheObj next = cacheObjIterator.next();\n CacheInfo cacheInfo = new CacheInfo<>();\n cacheInfo.setKey(next.getKey());\n cacheInfo.setValue(next.getValue());\n cacheInfo.setTtl(next.getTtl());\n cacheInfo.setExpiredTime(next.getExpiredTime());\n cacheInfoList.add(cacheInfo);\n }\n\n return cacheInfoList;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/controller/DirectLinkController.java",
"content": "package im.zhaojun.zfile.module.link.controller;\n\nimport com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\nimport com.github.xiaoymin.knife4j.annotations.ApiSort;\nimport im.zhaojun.zfile.core.exception.ErrorCode;\nimport im.zhaojun.zfile.core.exception.core.BizException;\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.core.util.SpringMvcUtils;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.module.config.model.entity.SystemConfig;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.link.model.request.BatchGenerateLinkRequest;\nimport im.zhaojun.zfile.module.link.model.result.BatchGenerateLinkResponse;\nimport im.zhaojun.zfile.module.link.service.DynamicDirectLinkPrefixService;\nimport im.zhaojun.zfile.module.link.service.LinkDownloadService;\nimport im.zhaojun.zfile.module.storage.annotation.StoragePermissionCheck;\nimport im.zhaojun.zfile.module.storage.context.StorageSourceContext;\nimport im.zhaojun.zfile.module.storage.model.enums.FileOperatorTypeEnum;\nimport im.zhaojun.zfile.module.storage.service.base.AbstractBaseFileService;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport jakarta.validation.Valid;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.boot.context.event.ApplicationReadyEvent;\nimport org.springframework.context.event.EventListener;\nimport org.springframework.http.ResponseEntity;\nimport org.springframework.stereotype.Controller;\nimport org.springframework.web.bind.annotation.PathVariable;\nimport org.springframework.web.bind.annotation.PostMapping;\nimport org.springframework.web.bind.annotation.RequestBody;\nimport org.springframework.web.bind.annotation.ResponseBody;\nimport org.springframework.web.servlet.mvc.method.RequestMappingInfo;\n\nimport java.io.IOException;\nimport java.lang.reflect.Method;\nimport java.util.ArrayList;\nimport java.util.List;\n\n/**\n * 短链接口\n *\n * @author zhaojun\n */\n@Tag(name = \"短链\")\n@ApiSort(5)\n@Controller\n@Slf4j\npublic class DirectLinkController {\n\n @Resource\n private LinkDownloadService linkDownloadService;\n\n @Resource\n private SystemConfigService systemConfigService;\n\n @Resource\n private DynamicDirectLinkPrefixService dynamicDirectLinkPrefixService;\n\n public static final String DIRECT_LINK_SUFFIX_PATH = \"/{storageKey}/**\";\n\n @EventListener(ApplicationReadyEvent.class)\n public void init() throws NoSuchMethodException {\n String directLinkPrefix = systemConfigService.getSystemConfig().getDirectLinkPrefix();\n Method directLinkMethod = DirectLinkController.class.getMethod(\"directLink\", String.class);\n RequestMappingInfo requestMappingInfo = RequestMappingInfo.paths(directLinkPrefix + DIRECT_LINK_SUFFIX_PATH).build();\n dynamicDirectLinkPrefixService.registerMappingHandlerMapping(SystemConfig.DIRECT_LINK_PREFIX_NAME, requestMappingInfo, this, directLinkMethod);\n }\n\n /**\n * 路径直链处理方法，会根据 URL 中的存储源 key 和文件路径, 获取到文件，判断文件是否有短链，没有则生成，然后跳转到短链.\n *\n * @param storageKey\n * 存储源 key\n */\n public ResponseEntity directLink(@PathVariable(\"storageKey\") String storageKey) throws IOException {\n // 获取直链全路径\n String filePath = SpringMvcUtils.getExtractPathWithinPattern();\n\n // 如果路径不是以 / 开头, 则补充上\n if (StringUtils.isNotEmpty(filePath) && filePath.charAt(0) != StringUtils.SLASH_CHAR) {\n filePath = StringUtils.SLASH + filePath;\n }\n\n return linkDownloadService.handlerDirectLink(storageKey, filePath);\n }\n\n @PostMapping(\"/api/path-link/batch/generate\")\n @ResponseBody\n @ApiOperationSupport(order = 1)\n @Operation(summary = \"生成路径直链\", description =\"对指定存储源的某文件路径生成路径直链\")\n @StoragePermissionCheck(action = FileOperatorTypeEnum.LINK)\n public AjaxJson> generatorShortLink(@RequestBody @Valid BatchGenerateLinkRequest batchGenerateLinkRequest) {\n List result = new ArrayList<>();\n\n // 获取站点域名\n String serverAddress = systemConfigService.getAxiosFromDomainOrSetting();\n String directLinkPrefix = systemConfigService.getSystemConfig().getDirectLinkPrefix();\n\n String storageKey = batchGenerateLinkRequest.getStorageKey();\n\n AbstractBaseFileService baseFileService = StorageSourceContext.getByStorageKey(storageKey);\n if (baseFileService == null) {\n throw new BizException(ErrorCode.BIZ_STORAGE_NOT_FOUND);\n }\n String currentUserBasePath = baseFileService.getCurrentUserBasePath();\n\n for (String path : batchGenerateLinkRequest.getPaths()) {\n // 拼接全路径地址.\n String fullPath = StringUtils.concat(serverAddress, directLinkPrefix, storageKey, currentUserBasePath, path);\n result.add(new BatchGenerateLinkResponse(fullPath));\n }\n return AjaxJson.getSuccessData(result);\n }\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/controller/ShortLinkController.java",
"content": "package im.zhaojun.zfile.module.link.controller;\n\nimport com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\nimport com.github.xiaoymin.knife4j.annotations.ApiSort;\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.link.model.entity.ShortLink;\nimport im.zhaojun.zfile.module.link.model.request.BatchGenerateLinkRequest;\nimport im.zhaojun.zfile.module.link.model.result.BatchGenerateLinkResponse;\nimport im.zhaojun.zfile.module.link.service.LinkDownloadService;\nimport im.zhaojun.zfile.module.link.service.ShortLinkService;\nimport im.zhaojun.zfile.module.storage.annotation.StoragePermissionCheck;\nimport im.zhaojun.zfile.module.storage.context.StorageSourceContext;\nimport im.zhaojun.zfile.module.storage.model.enums.FileOperatorTypeEnum;\nimport im.zhaojun.zfile.module.storage.service.StorageSourceService;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.Parameter;\nimport io.swagger.v3.oas.annotations.enums.ParameterIn;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport jakarta.validation.Valid;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.http.ResponseEntity;\nimport org.springframework.stereotype.Controller;\nimport org.springframework.web.bind.annotation.*;\n\nimport java.io.IOException;\nimport java.util.ArrayList;\nimport java.util.List;\n\n/**\n * 短链接口\n *\n * @author zhaojun\n */\n@Tag(name = \"直短链模块\")\n@ApiSort(5)\n@Controller\n@Slf4j\npublic class ShortLinkController {\n\n @Resource\n private SystemConfigService systemConfigService;\n\n @Resource\n private ShortLinkService shortLinkService;\n\n @Resource\n private StorageSourceService storageSourceService;\n\n @Resource\n private LinkDownloadService linkDownloadService;\n\n @PostMapping(\"/api/short-link/batch/generate\")\n @ResponseBody\n @ApiOperationSupport(order = 1)\n @Operation(summary = \"生成短链\", description =\"对指定存储源的某文件路径生成短链\")\n @StoragePermissionCheck(action = FileOperatorTypeEnum.SHORT_LINK)\n public AjaxJson> generatorShortLink(@RequestBody @Valid BatchGenerateLinkRequest batchGenerateLinkRequest) {\n List result = new ArrayList<>();\n\n // 获取站点域名\n String domain = systemConfigService.getAxiosFromDomainOrSetting();\n Long expireTime = batchGenerateLinkRequest.getExpireTime();\n String storageKey = batchGenerateLinkRequest.getStorageKey();\n Integer storageId = storageSourceService.findIdByKey(storageKey);\n\n for (String path : batchGenerateLinkRequest.getPaths()) {\n // 拼接全路径地址.\n String currentUserBasePath = StorageSourceContext.getByStorageId(storageId).getCurrentUserBasePath();\n String fullPath = StringUtils.concat(currentUserBasePath, path);\n ShortLink shortLink = shortLinkService.generatorShortLink(storageId, fullPath, expireTime);\n String shortUrl = StringUtils.removeDuplicateSlashes(domain + \"/s/\" + shortLink.getShortKey());\n result.add(new BatchGenerateLinkResponse(shortUrl));\n }\n return AjaxJson.getSuccessData(result);\n }\n\n\n @GetMapping(\"/s/{key}\")\n @ApiOperationSupport(order = 2)\n @Operation(summary = \"跳转短链\", description =\"根据短链 key 跳转（302 重定向）到对应的直链.\")\n @Parameter(in = ParameterIn.PATH, name = \"key\", description = \"短链 key\", required = true, schema = @Schema(type = \"string\"))\n public ResponseEntity parseShortKey(@PathVariable String key) throws IOException {\n return linkDownloadService.handlerShortLink(key);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/controller/ShortLinkManagerController.java",
"content": "package im.zhaojun.zfile.module.link.controller;\n\nimport cn.hutool.core.io.IoUtil;\nimport cn.hutool.core.util.ObjUtil;\nimport cn.hutool.poi.excel.ExcelUtil;\nimport cn.hutool.poi.excel.ExcelWriter;\nimport com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;\nimport com.baomidou.mybatisplus.core.metadata.OrderItem;\nimport com.baomidou.mybatisplus.extension.plugins.pagination.Page;\nimport com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\nimport com.github.xiaoymin.knife4j.annotations.ApiSort;\nimport im.zhaojun.zfile.core.annotation.DemoDisable;\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.link.cache.LinkRateLimiterCache;\nimport im.zhaojun.zfile.module.link.convert.ShortLinkConvert;\nimport im.zhaojun.zfile.module.link.model.dto.CacheInfo;\nimport im.zhaojun.zfile.module.link.model.entity.ShortLink;\nimport im.zhaojun.zfile.module.link.model.request.BatchDeleteRequest;\nimport im.zhaojun.zfile.module.link.model.request.QueryShortLinkLogRequest;\nimport im.zhaojun.zfile.module.link.model.request.ShortLinkResult;\nimport im.zhaojun.zfile.module.link.service.ShortLinkService;\nimport im.zhaojun.zfile.module.storage.model.entity.StorageSource;\nimport im.zhaojun.zfile.module.storage.service.StorageSourceService;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.Parameter;\nimport io.swagger.v3.oas.annotations.enums.ParameterIn;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport jakarta.servlet.ServletOutputStream;\nimport jakarta.servlet.http.HttpServletResponse;\nimport jakarta.validation.constraints.NotNull;\nimport org.springframework.http.MediaType;\nimport org.springframework.stereotype.Controller;\nimport org.springframework.web.bind.annotation.*;\n\nimport java.io.IOException;\nimport java.util.HashMap;\nimport java.util.List;\nimport java.util.Map;\nimport java.util.Objects;\nimport java.util.concurrent.atomic.AtomicInteger;\nimport java.util.stream.Collectors;\nimport java.util.stream.Stream;\n\n/**\n * 直链管理接口\n *\n * @author zhaojun\n */\n@Tag(name = \"直链管理\")\n@ApiSort(7)\n@Controller\n@RequestMapping(\"/admin\")\npublic class ShortLinkManagerController {\n\n @Resource\n private SystemConfigService systemConfigService;\n\n @Resource\n private ShortLinkService shortLinkService;\n\n @Resource\n private StorageSourceService storageSourceService;\n\n @Resource\n private ShortLinkConvert shortLinkConvert;\n\n @Resource\n private LinkRateLimiterCache linkRateLimiterCache;\n\n\n @ApiOperationSupport(order = 1)\n @GetMapping(\"/link/list\")\n @Operation(summary = \"搜索短链\")\n @ResponseBody\n public AjaxJson> list(QueryShortLinkLogRequest queryObj) {\n Page resultPage = getShortLinkResultPage(queryObj);\n\n String serverAddress = systemConfigService.getAxiosFromDomainOrSetting();\n\n resultPage.getRecords().forEach(shortLinkResult -> {\n shortLinkResult.setShortLink(StringUtils.concat(serverAddress, \"s\", shortLinkResult.getShortKey()));\n });\n\n return AjaxJson.getPageData(resultPage.getTotal(), resultPage.getRecords());\n }\n\n @ApiOperationSupport(order = 2)\n @DeleteMapping(\"/link/delete/{id}\")\n @Operation(summary = \"删除短链\")\n @Parameter(in = ParameterIn.PATH, name = \"id\", description = \"短链 id\", required = true, schema = @Schema(type = \"integer\"))\n @ResponseBody\n @DemoDisable\n public AjaxJson deleteById(@PathVariable Integer id) {\n shortLinkService.removeById(id);\n return AjaxJson.getSuccess();\n }\n\n\n @ApiOperationSupport(order = 3)\n @PostMapping(\"/link/delete/batch\")\n @ResponseBody\n @Operation(summary = \"批量删除短链\")\n @DemoDisable\n public AjaxJson batchDelete(@RequestBody BatchDeleteRequest batchDeleteRequest) {\n shortLinkService.removeBatchByIds(batchDeleteRequest.getIds());\n return AjaxJson.getSuccess();\n }\n\n @ApiOperationSupport(order = 4)\n @GetMapping(\"/link/export\")\n @ResponseBody\n @Operation(summary = \"导出短链\")\n public void exportExcel(QueryShortLinkLogRequest queryObj, HttpServletResponse response) throws IOException {\n Page shortLinkResultPage = getShortLinkResultPage(queryObj);\n\n ExcelWriter writer = ExcelUtil.getWriter(true);\n writer.addHeaderAlias(\"id\", \"ID\");\n writer.addHeaderAlias(\"storageName\", \"存储源名称\");\n writer.addHeaderAlias(\"storageTypeStr\", \"存储源类型\");\n writer.addHeaderAlias(\"shortKey\", \"短链 key\");\n writer.addHeaderAlias(\"url\", \"文件路径\");\n writer.addHeaderAlias(\"createDate\", \"创建时间\");\n writer.addHeaderAlias(\"expireDate\", \"过期时间\");\n writer.setOnlyAlias(true);\n\n writer.write(shortLinkResultPage.getRecords(), true);\n writer.setColumnWidth(0, 8);\n writer.setColumnWidth(1, 30);\n writer.setColumnWidth(2, 15);\n writer.setColumnWidth(3, 15);\n writer.setColumnWidth(4, 50);\n writer.setColumnWidth(5, 15);\n writer.setColumnWidth(6, 15);\n\n response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);\n\n ServletOutputStream out=response.getOutputStream();\n\n writer.flush(out, true);\n writer.close();\n IoUtil.close(out);\n }\n\n\n @ApiOperationSupport(order = 5)\n @GetMapping(\"/link/limit/info\")\n @ResponseBody\n @Operation(summary = \"获取直链访问限制信息\")\n public AjaxJson>> getLinkLimitInfo() {\n return AjaxJson.getSuccessData(linkRateLimiterCache.getCacheInfo());\n }\n\n @NotNull\n private Page getShortLinkResultPage(QueryShortLinkLogRequest queryObj) {\n // 分页和排序\n boolean asc = Objects.equals(queryObj.getOrderDirection(), \"asc\");\n OrderItem orderItem = asc ? OrderItem.asc(queryObj.getOrderBy()) : OrderItem.desc(queryObj.getOrderBy());\n Page pages = new Page(queryObj.getPage(), queryObj.getLimit())\n .addOrder(orderItem);\n\n // 搜索条件\n LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper()\n .eq(StringUtils.isNotEmpty(queryObj.getStorageId()), ShortLink::getStorageId, queryObj.getStorageId())\n .like(StringUtils.isNotEmpty(queryObj.getKey()), ShortLink::getShortKey, queryObj.getKey())\n .like(StringUtils.isNotEmpty(queryObj.getUrl()), ShortLink::getUrl, queryObj.getUrl())\n .ge(ObjUtil.isNotEmpty(queryObj.getDateFrom()), ShortLink::getCreateDate, queryObj.getDateFrom())\n .le(ObjUtil.isNotEmpty(queryObj.getDateTo()), ShortLink::getCreateDate, queryObj.getDateTo());\n\n // 执行查询\n Page selectResult = shortLinkService.selectPage(pages, queryWrapper);\n\n // 转换为结果集\n Map cache = new HashMap<>();\n Stream shortLinkResultList = selectResult.getRecords().stream().map(shortLink -> {\n Integer shortLinkStorageId = shortLink.getStorageId();\n\n StorageSource storageSource = cache.getOrDefault(shortLinkStorageId, storageSourceService.findById(shortLinkStorageId));\n cache.put(shortLinkStorageId, storageSource);\n return shortLinkConvert.entityToResultList(shortLink, storageSource);\n });\n\n Page resultPage = new Page<>();\n resultPage.setTotal(selectResult.getTotal());\n resultPage.setRecords(shortLinkResultList.collect(Collectors.toList()));\n return resultPage;\n }\n\n @ApiOperationSupport(order = 6)\n @DeleteMapping(\"/link/deleteExpireLink\")\n @Operation(summary = \"删除过期短链\")\n @ResponseBody\n @DemoDisable\n public AjaxJson deleteExpireLink() {\n int updateRows = shortLinkService.deleteExpireLink();\n return AjaxJson.getSuccessData(updateRows);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/convert/ShortLinkConvert.java",
"content": "package im.zhaojun.zfile.module.link.convert;\n\nimport im.zhaojun.zfile.module.link.model.entity.ShortLink;\nimport im.zhaojun.zfile.module.storage.model.entity.StorageSource;\nimport im.zhaojun.zfile.module.link.model.request.ShortLinkResult;\nimport org.mapstruct.Mapper;\nimport org.mapstruct.Mapping;\nimport org.mapstruct.ReportingPolicy;\nimport org.springframework.stereotype.Component;\n\n/**\n * 直链实体类器\n *\n * @author zhaojun\n */\n@Component\n@Mapper(componentModel = \"spring\", unmappedTargetPolicy = ReportingPolicy.IGNORE)\npublic interface ShortLinkConvert {\n\n\t@Mapping(source = \"shortLink.id\", target = \"id\")\n\t@Mapping(source = \"storageSource.name\", target = \"storageName\")\n\t@Mapping(source = \"storageSource.type\", target = \"storageType\")\n\tShortLinkResult entityToResultList(ShortLink shortLink, StorageSource storageSource);\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/dto/DynamicRegisterMappingHandlerDTO.java",
"content": "package im.zhaojun.zfile.module.link.dto;\n\nimport lombok.AllArgsConstructor;\nimport lombok.Data;\nimport org.springframework.web.servlet.mvc.method.RequestMappingInfo;\n\nimport java.lang.reflect.Method;\n\n@Data\n@AllArgsConstructor\npublic class DynamicRegisterMappingHandlerDTO {\n\n private RequestMappingInfo requestMappingInfo;\n\n private Object object;\n\n private Method method;\n\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/event/DeleteExpireLinkEvent.java",
"content": "package im.zhaojun.zfile.module.link.event;\n\nimport lombok.Data;\n\n@Data\npublic class DeleteExpireLinkEvent {\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/mapper/ShortLinkMapper.java",
"content": "package im.zhaojun.zfile.module.link.mapper;\n\nimport com.baomidou.mybatisplus.core.mapper.BaseMapper;\nimport im.zhaojun.zfile.module.link.model.entity.ShortLink;\nimport jakarta.annotation.Nullable;\nimport org.apache.ibatis.annotations.Mapper;\nimport org.apache.ibatis.annotations.Param;\n\nimport java.util.Date;\n\n/**\n * 短链接配置表 Mapper 接口\n *\n * @author zhaojun\n */\n@Mapper\npublic interface ShortLinkMapper extends BaseMapper {\n\n /**\n * 根据短链接 key 查询短链接\n *\n * @param key\n * 短链接 key\n *\n * @return 短链接信息\n */\n ShortLink findByKey(@Param(\"key\")String key);\n\n\n\t/**\n\t * 根据存储源 ID 删除所有数据\n\t *\n\t * @param \tstorageId\n\t * \t\t\t存储源 ID\n\t */\n\tint deleteByStorageId(@Param(\"storageId\") Integer storageId);\n\n\t/**\n\t * 根据存储源 ID 和 URL 查询短链接\n\t */\n\tShortLink findByStorageIdAndUrl(@Param(\"storageId\") Integer storageId,\n\t\t\t\t\t\t\t\t\t@Param(\"url\") String url,\n\t\t\t\t\t\t\t\t\t@Nullable @Param(\"expireDate\") Date expireDate);\n\n\t/**\n\t * 删除过期的短链接\n\t *\n\t * @return 删除的行数\n\t */\n\tint deleteExpireLink();\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/dto/CacheInfo.java",
"content": "package im.zhaojun.zfile.module.link.model.dto;\n\nimport lombok.Data;\n\nimport java.util.Date;\n\n@Data\npublic class CacheInfo {\n\n private K key;\n\n private V value;\n\n private Date expiredTime;\n\n private Long ttl;\n\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/entity/ShortLink.java",
"content": "package im.zhaojun.zfile.module.link.model.entity;\n\nimport cn.hutool.core.date.DateUtil;\nimport com.baomidou.mybatisplus.annotation.IdType;\nimport com.baomidou.mybatisplus.annotation.TableField;\nimport com.baomidou.mybatisplus.annotation.TableId;\nimport com.baomidou.mybatisplus.annotation.TableName;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport java.io.Serializable;\nimport java.util.Date;\n\n/**\n * 短链信息 entity\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"短链信息\")\n@TableName(value = \"short_link\")\npublic class ShortLink implements Serializable {\n\n private static final long serialVersionUID = 1L;\n\n /**\n * 永久直链失效时间为 -1\n */\n public static final Long PERMANENT_EXPIRE_TIME = -1L;\n\n /**\n * 永久直链失效日期为 9999-12-31\n */\n public static final Date PERMANENT_EXPIRE_DATE = DateUtil.parseDate(\"9999-12-31\");\n\n @TableId(value = \"id\", type = IdType.AUTO)\n @Schema(title = \"ID, 新增无需填写\", example = \"1\")\n private Integer id;\n\n\n @TableField(value = \"storage_id\")\n @Schema(title = \"存储源 ID\", example = \"1\")\n private Integer storageId;\n\n\n @TableField(value = \"short_key\")\n @Schema(title = \"短链 key\", example = \"voldd3\")\n private String shortKey;\n\n\n @TableField(value = \"url\")\n @Schema(title = \"短链 url\", example = \"/directlink/1/test02.png\")\n private String url;\n\n\n @TableField(value = \"create_date\")\n @Schema(title = \"创建时间\", example = \"2021-11-22 10:05\")\n private Date createDate;\n\n\n @TableField(value = \"expire_date\")\n @Schema(title = \"过期时间\", example = \"2021-11-23 10:05\")\n private Date expireDate;\n\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/enums/RefererTypeEnum.java",
"content": "package im.zhaojun.zfile.module.link.model.enums;\n\nimport com.baomidou.mybatisplus.annotation.EnumValue;\nimport com.fasterxml.jackson.annotation.JsonValue;\nimport lombok.AllArgsConstructor;\nimport lombok.Getter;\n\n/**\n * Referer 防盗链类型枚举\n *\n * @author zhaojun\n */\n@Getter\n@AllArgsConstructor\npublic enum RefererTypeEnum {\n\n\t/**\n\t * 不启用 Referer 防盗链\n\t */\n\tOFF(\"off\"),\n\n\t/**\n\t * 启用白名单模式\n\t */\n\tWHITE_LIST(\"white_list\"),\n\n\t/**\n\t * 启用黑名单模式\n\t */\n\tBLACK_LIST(\"black_list\");\n\n\t@EnumValue\n\t@JsonValue\n\tprivate final String value;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/request/BatchDeleteRequest.java",
"content": "package im.zhaojun.zfile.module.link.model.request;\n\nimport lombok.Data;\n\nimport java.util.List;\n\n/**\n * @author zhaojun\n */\n@Data\npublic class BatchDeleteRequest {\n\t\n\tprivate List ids;\n\t\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/request/BatchGenerateLinkRequest.java",
"content": "package im.zhaojun.zfile.module.link.model.request;\n\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport jakarta.validation.constraints.NotBlank;\nimport jakarta.validation.constraints.NotEmpty;\nimport jakarta.validation.constraints.NotNull;\nimport java.util.List;\n\n/**\n * 批量生成直链请求类\n * @author zhaojun\n */\n@Data\n@Schema(description = \"批量生成直链请求类\")\npublic class BatchGenerateLinkRequest {\n\t\n\t@NotBlank(message = \"存储源 key 不能为空\")\n\tprivate String storageKey;\n\t\n\t@NotEmpty(message = \"生成的文件路径不能为空\")\n\tprivate List paths;\n\n\t/**\n\t * 有效期, 单位: 秒\n\t */\n\t@NotNull(message = \"过期时间不能为空\")\n\tprivate Long expireTime;\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/request/QueryDownloadLogRequest.java",
"content": "package im.zhaojun.zfile.module.link.model.request;\n\nimport cn.hutool.core.date.DateUtil;\nimport im.zhaojun.zfile.core.model.request.PageQueryRequest;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\nimport lombok.EqualsAndHashCode;\nimport org.springframework.format.annotation.DateTimeFormat;\n\nimport java.util.Date;\nimport java.util.List;\n\n/**\n * 查询下载日志请求参数\n *\n * @author zhaojun\n */\n@Data\n@EqualsAndHashCode(callSuper = true)\npublic class QueryDownloadLogRequest extends PageQueryRequest {\n\n\t@Schema(title=\"文件路径\")\n\tprivate String path;\n\n\t@Schema(title=\"存储源 key\")\n\tprivate String storageKey;\n\n\t@Schema(title=\"链接类型\")\n\tprivate String linkType;\n\n\t@Schema(title=\"短链 key\")\n\tprivate String shortKey;\n\n\t@Schema(title=\"访问时间\")\n\t@DateTimeFormat(pattern = \"yyyy-MM-dd HH:mm:ss\")\n\tprivate List searchDate;\n\n\t@Schema(title=\"访问 ip\")\n\tprivate String ip;\n\n\t@Schema(title=\"访问 user_agent\")\n\tprivate String userAgent;\n\n\t@Schema(title=\"访问 referer\")\n\tprivate String referer;\n\t\n\t@Schema(title=\"排序字段\")\n\tprivate String orderBy = \"create_time\";\n\n\tpublic Date getDateFrom() {\n\t\tif (searchDate == null || searchDate.isEmpty()) {\n\t\t\treturn null;\n\t\t}\n\t\treturn DateUtil.beginOfDay(searchDate.getFirst());\n\t}\n\n\tpublic Date getDateTo() {\n\t\tif (searchDate == null || searchDate.isEmpty()) {\n\t\t\treturn null;\n\t\t}\n\t\treturn DateUtil.endOfDay(searchDate.getLast());\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/request/QueryLoginLogRequest.java",
"content": "package im.zhaojun.zfile.module.link.model.request;\n\nimport cn.hutool.core.date.DateUtil;\nimport im.zhaojun.zfile.core.model.request.PageQueryRequest;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\nimport lombok.EqualsAndHashCode;\nimport org.springframework.format.annotation.DateTimeFormat;\n\nimport java.util.Date;\nimport java.util.List;\n\n/**\n * 查询下载日志请求参数\n *\n * @author zhaojun\n */\n@Data\n@EqualsAndHashCode(callSuper = true)\npublic class QueryLoginLogRequest extends PageQueryRequest {\n\n\t@Schema(title=\"用户名\")\n\tprivate String username;\n\n\t@Schema(title=\"密码\")\n\tprivate String password;\n\n\t@Schema(title=\"IP\")\n\tprivate String ip;\n\n\t@Schema(title=\"User-Agent\")\n\tprivate String userAgent;\n\n\t@Schema(title=\"来源\")\n\tprivate String referer;\n\n\t@Schema(title=\"登录结果\")\n\tprivate String result;\n\n\t@Schema(title=\"访问时间\")\n\t@DateTimeFormat(pattern = \"yyyy-MM-dd HH:mm:ss\")\n\tprivate List searchDate;\n\n\t@Schema(title=\"排序字段\")\n\tprivate String orderBy = \"create_time\";\n\n\tpublic Date getDateFrom() {\n\t\tif (searchDate == null) {\n\t\t\treturn null;\n\t\t}\n\t\treturn DateUtil.beginOfDay(searchDate.getFirst());\n\t}\n\n\tpublic Date getDateTo() {\n\t\tif (searchDate == null) {\n\t\t\treturn null;\n\t\t}\n\t\treturn DateUtil.endOfDay(searchDate.getLast());\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/request/QueryShortLinkLogRequest.java",
"content": "package im.zhaojun.zfile.module.link.model.request;\n\nimport cn.hutool.core.date.DateUtil;\nimport im.zhaojun.zfile.core.model.request.PageQueryRequest;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\nimport lombok.EqualsAndHashCode;\nimport org.springframework.format.annotation.DateTimeFormat;\n\nimport java.util.Date;\nimport java.util.List;\n\n/**\n * @author zhaojun\n */\n@Data\n@EqualsAndHashCode(callSuper = true)\npublic class QueryShortLinkLogRequest extends PageQueryRequest {\n\t\n\t@Schema(title=\"短链 key\")\n\tprivate String key;\n\t\n\t@Schema(title=\"存储源 id\")\n\tprivate String storageId;\n\t\n\t@Schema(title=\"短链文件路径\")\n\tprivate String url;\n\n\t@Schema(title=\"访问时间\")\n\t@DateTimeFormat(pattern = \"yyyy-MM-dd HH:mm:ss\")\n\tprivate List searchDate;\n\n\tpublic Date getDateFrom() {\n\t\tif (searchDate == null) {\n\t\t\treturn null;\n\t\t}\n\t\treturn DateUtil.beginOfDay(searchDate.getFirst());\n\t}\n\n\tpublic Date getDateTo() {\n\t\tif (searchDate == null) {\n\t\t\treturn null;\n\t\t}\n\t\treturn DateUtil.endOfDay(searchDate.getLast());\n\t}\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/request/ShortLinkResult.java",
"content": "package im.zhaojun.zfile.module.link.model.request;\n\nimport com.fasterxml.jackson.annotation.JsonIgnore;\nimport im.zhaojun.zfile.module.storage.model.enums.StorageTypeEnum;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport java.util.Date;\n\n/**\n * 短链结果类\n *\n * @author zhaojun\n */\n@Data\npublic class ShortLinkResult {\n\n\t@Schema(title = \"短链 id\", example = \"1\")\n\tprivate Integer id;\n\n\t@Schema(title = \"存储源名称\", example = \"我的本地存储\")\n\tprivate String storageName;\n\n\t@Schema(title = \"存储源类型\")\n\tprivate StorageTypeEnum storageType;\n\n\t@JsonIgnore\n\tprivate String storageTypeStr;\n\n\tpublic String getStorageTypeStr() {\n\t\treturn storageType.getDescription();\n\t}\n\n\t@Schema(title = \"短链 key\", example = \"voldd3\")\n\tprivate String shortKey;\n\n\t@Schema(title = \"文件 url\", example = \"/directlink/1/test02.png\")\n\tprivate String url;\n\n\t@Schema(title = \"创建时间\", example = \"2021-11-22 10:05\")\n\tprivate Date createDate;\n\n\t@Schema(title = \"过期时间\", example = \"2021-11-23 10:05\")\n\tprivate Date expireDate;\n\n\t@Schema(title=\"短链地址\")\n\tprivate String shortLink;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/request/ShortLinkSearchRequest.java",
"content": "package im.zhaojun.zfile.module.link.model.request;\n\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport jakarta.validation.constraints.NotBlank;\n\n/**\n * 短链接搜索请求参数\n *\n * @author zhaojun\n */\n@Data\n@Schema(description = \"搜索存储源中文件请求类\")\npublic class ShortLinkSearchRequest {\n\n @Schema(title = \"存储源 id\", requiredMode = Schema.RequiredMode.REQUIRED, example = \"1\")\n @NotBlank(message = \"存储源 id 不能为空\")\n private Integer storageId;\n\n @Schema(title = \"存储源 key\", example = \"local\")\n private String key;\n\n @Schema(title = \"文件 url/路径\", example = \"/a\")\n private String url;\n\n @Schema(title = \"开始时间\", example = \"2022-01-01 00:00:00\")\n private String dateFrom;\n\n @Schema(title = \"结束时间\", example = \"2022-12-31 23:59:59\")\n private String dateTo;\n\n @Schema(title = \"页码\", example = \"1\")\n private Integer page;\n\n @Schema(title = \"每页数量\", example = \"10\")\n private Integer limit;\n\n @Schema(title = \"排序字段\", example = \"id\")\n private String orderBy;\n\n @Schema(title = \"排序方式\", example = \"desc\")\n private String orderDirection;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/model/result/BatchGenerateLinkResponse.java",
"content": "package im.zhaojun.zfile.module.link.model.result;\n\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.AllArgsConstructor;\nimport lombok.Data;\n\n/**\n * @author zhaojun\n */\n@Data\n@Schema(description = \"批量生成直链结果类\")\n@AllArgsConstructor\npublic class BatchGenerateLinkResponse {\n\t\n\tprivate String address;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/service/DynamicDirectLinkPrefixService.java",
"content": "package im.zhaojun.zfile.module.link.service;\n\nimport im.zhaojun.zfile.module.link.dto.DynamicRegisterMappingHandlerDTO;\nimport jakarta.annotation.Resource;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.stereotype.Service;\nimport org.springframework.web.servlet.mvc.method.RequestMappingInfo;\nimport org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;\n\nimport java.lang.reflect.Method;\nimport java.util.Map;\nimport java.util.concurrent.ConcurrentHashMap;\n\n/**\n * 动态请求映射服务，用于在线注册、修改、注销 @RequestMapping 注解的方法.\n *\n * @author zhaojun\n */\n@Slf4j\n@Service\npublic class DynamicDirectLinkPrefixService {\n\n @Resource\n private RequestMappingHandlerMapping requestMappingHandlerMapping;\n\n public static final Map REGISTER_MAPPING = new ConcurrentHashMap<>();\n\n public void registerMappingHandlerMapping(String key, RequestMappingInfo requestMappingInfo, Object controllerObj, Method directLinkMethod) {\n requestMappingHandlerMapping.registerMapping(requestMappingInfo, controllerObj, directLinkMethod);\n REGISTER_MAPPING.put(key, new DynamicRegisterMappingHandlerDTO(requestMappingInfo, controllerObj, directLinkMethod));\n\n }\n\n public void updateRegisterMappingHandler(String key, RequestMappingInfo requestMappingInfo) {\n synchronized (key.intern()) {\n DynamicRegisterMappingHandlerDTO dynamicRegisterMappingHandlerDTO = REGISTER_MAPPING.get(key);\n if (dynamicRegisterMappingHandlerDTO != null) {\n requestMappingHandlerMapping.unregisterMapping(dynamicRegisterMappingHandlerDTO.getRequestMappingInfo());\n requestMappingHandlerMapping.registerMapping(requestMappingInfo, dynamicRegisterMappingHandlerDTO.getObject(), dynamicRegisterMappingHandlerDTO.getMethod());\n REGISTER_MAPPING.put(key, new DynamicRegisterMappingHandlerDTO(requestMappingInfo, dynamicRegisterMappingHandlerDTO.getObject(), dynamicRegisterMappingHandlerDTO.getMethod()));\n }\n }\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/service/LinkDownloadService.java",
"content": "package im.zhaojun.zfile.module.link.service;\n\nimport cn.hutool.core.date.DateTime;\nimport cn.hutool.core.date.DateUtil;\nimport cn.hutool.core.io.FileUtil;\nimport im.zhaojun.zfile.core.exception.ErrorCode;\nimport im.zhaojun.zfile.core.exception.biz.InvalidStorageSourceBizException;\nimport im.zhaojun.zfile.core.exception.core.ErrorPageBizException;\nimport im.zhaojun.zfile.core.exception.status.ForbiddenAccessException;\nimport im.zhaojun.zfile.core.exception.status.NotFoundAccessException;\nimport im.zhaojun.zfile.core.util.FileUtils;\nimport im.zhaojun.zfile.core.util.HttpUtil;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.module.config.model.dto.SystemConfigDTO;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.filter.service.FilterConfigService;\nimport im.zhaojun.zfile.module.link.model.entity.ShortLink;\nimport im.zhaojun.zfile.module.log.model.entity.DownloadLog;\nimport im.zhaojun.zfile.module.log.service.DownloadLogService;\nimport im.zhaojun.zfile.module.storage.annotation.LinkRateLimiter;\nimport im.zhaojun.zfile.module.storage.annotation.RefererCheck;\nimport im.zhaojun.zfile.module.storage.context.StorageSourceContext;\nimport im.zhaojun.zfile.module.storage.model.entity.StorageSource;\nimport im.zhaojun.zfile.module.storage.service.StorageSourceService;\nimport im.zhaojun.zfile.module.storage.service.base.AbstractBaseFileService;\nimport jakarta.annotation.Resource;\nimport lombok.extern.slf4j.Slf4j;\nimport org.apache.commons.lang3.BooleanUtils;\nimport org.springframework.http.ContentDisposition;\nimport org.springframework.http.HttpHeaders;\nimport org.springframework.http.MediaType;\nimport org.springframework.http.ResponseEntity;\nimport org.springframework.stereotype.Service;\n\nimport java.io.IOException;\nimport java.nio.charset.StandardCharsets;\nimport java.util.HashSet;\nimport java.util.Set;\n\n/**\n * @author zhaojun\n */\n@Slf4j\n@Service\npublic class LinkDownloadService {\n\n @Resource\n private StorageSourceService storageSourceService;\n\n @Resource\n private DownloadLogService downloadLogService;\n\n @Resource\n private SystemConfigService systemConfigService;\n\n @Resource\n private ShortLinkService shortLinkService;\n\n @Resource\n private FilterConfigService filterConfigService;\n\n private final Set expireKeySet = new HashSet<>();\n\n @RefererCheck\n @LinkRateLimiter\n public ResponseEntity handlerDirectLink(String storageKey, String filePath) {\n // 检查系统是否允许直链\n SystemConfigDTO systemConfigDTO = systemConfigService.getSystemConfig();\n if (BooleanUtils.isNotTrue(systemConfigDTO.getShowPathLink())) {\n throw new ForbiddenAccessException(ErrorCode.BIZ_DIRECT_LINK_NOT_ALLOWED);\n }\n return handlerDownloadGetUrl(storageKey, filePath, null, DownloadLog.DOWNLOAD_TYPE_DIRECT_LINK);\n }\n\n @RefererCheck\n @LinkRateLimiter\n public ResponseEntity handlerShortLink(String shortKey) throws IOException {\n // 从缓存中判断是否短链是否过期\n if (expireKeySet.contains(shortKey)) {\n throw new ForbiddenAccessException(ErrorCode.BIZ_SHORT_LINK_EXPIRED);\n }\n\n // 判断是否允许生成短链.\n SystemConfigDTO systemConfigDTO = systemConfigService.getSystemConfig();\n if (BooleanUtils.isNotTrue(systemConfigDTO.getShowShortLink())) {\n throw new ForbiddenAccessException(ErrorCode.BIZ_SHORT_LINK_NOT_ALLOWED);\n }\n\n // 判断短链是否存在\n ShortLink shortLink = shortLinkService.findByKey(shortKey);\n if (shortLink == null) {\n throw new NotFoundAccessException(ErrorCode.BIZ_SHORT_LINK_NOT_FOUNT);\n }\n\n // 判断短链是否过期\n if (shortLink.getExpireDate() != null) {\n DateTime now = DateUtil.date();\n boolean isExpire = now.isAfter(shortLink.getExpireDate());\n if (isExpire) {\n expireKeySet.add(shortKey);\n throw new ForbiddenAccessException(ErrorCode.BIZ_SHORT_LINK_EXPIRED);\n }\n }\n\n // 获取实际文件路径，下载并记录日志\n Integer storageId = shortLink.getStorageId();\n String storageKey = storageSourceService.findStorageKeyById(storageId);\n String filePath = shortLink.getUrl();\n return handlerDownloadGetUrl(storageKey, filePath, shortKey, DownloadLog.DOWNLOAD_TYPE_SHORT_LINK);\n }\n\n /**\n * 处理指定存储源的下载请求\n *\n * @param storageKey\n * 存储源 key\n *\n * @param filePath\n * 文件路径\n *\n * @param shortKey\n * 短链接 key\n *\n * @param downloadType\n * 下载类型, 直链下载(directLink)或短链下载(shortLink)\n */\n private ResponseEntity handlerDownloadGetUrl(String storageKey, String filePath, String shortKey, String downloadType) {\n String fileAlias = StringUtils.equals(downloadType, DownloadLog.DOWNLOAD_TYPE_DIRECT_LINK) ? filePath : shortKey;\n\n // 获取存储源 Service\n AbstractBaseFileService fileService;\n try {\n fileService = StorageSourceContext.getByStorageKey(storageKey);\n } catch (InvalidStorageSourceBizException e) {\n throw new ErrorPageBizException(\"无效的或初始化失败的存储源 [\" + storageKey + \"] 文件 [\" + fileAlias + \"] 下载链接异常, 无法下载.\", e);\n }\n\n if (fileService == null) {\n throw new ErrorPageBizException(\"未找到存储源 [\" + storageKey + \"] 文件 [\" + fileAlias + \"] 下载链接异常, 无法下载.\");\n }\n\n StorageSource storageSource = storageSourceService.findByStorageKey(storageKey);\n Boolean enable = storageSource.getEnable();\n if (!enable) {\n throw new ErrorPageBizException(\"未启用的存储源 [\" + storageKey + \"] 文件 [\" + fileAlias + \"] 下载链接异常, 无法下载.\");\n }\n\n // 检查是否访问了禁止下载的目录\n if (filterConfigService.checkFileIsDisableDownload(storageSource.getId(), filePath)) {\n // 获取 Forbidden 页面地址\n return ResponseEntity.status(302)\n .header(HttpHeaders.CACHE_CONTROL, \"no-cache, no-store, must-revalidate, private\")\n .header(HttpHeaders.PRAGMA, \"no-cache\")\n .header(HttpHeaders.EXPIRES, \"0\")\n .header(HttpHeaders.LOCATION, systemConfigService.getForbiddenUrl())\n .build();\n }\n\n // 获取文件下载链接\n String downloadUrl;\n try {\n downloadUrl = fileService.getDownloadUrl(filePath);\n } catch (NotFoundAccessException e) {\n throw e;\n } catch (Exception e) {\n throw new ErrorPageBizException(\"获取存储源 [\" + storageKey + \"] 文件 [\" + fileAlias + \"] 下载链接异常, 无法下载.\", e);\n }\n\n // 判断下载链接是否为空\n if (StringUtils.isEmpty(downloadUrl)) {\n throw new ErrorPageBizException(\"获取存储源 [\" + storageKey + \"] 文件 [\" + fileAlias + \"] 下载链接为空, 无法下载.\");\n }\n\n // 记录下载日志.\n SystemConfigDTO systemConfig = systemConfigService.getSystemConfig();\n Boolean recordDownloadLog = systemConfig.getRecordDownloadLog();\n if (BooleanUtils.isTrue(recordDownloadLog)) {\n DownloadLog downloadLog = new DownloadLog(downloadType, filePath, storageKey, shortKey);\n downloadLogService.save(downloadLog);\n }\n\n // 判断下载链接是否为 m3u8 格式, 如果是则返回 m3u8 内容.\n if (StringUtils.equalsIgnoreCase(FileUtil.extName(filePath), \"m3u8\")) {\n String textContent = HttpUtil.getTextContent(downloadUrl);\n HttpHeaders headers = new HttpHeaders();\n headers.setContentType(MediaType.parseMediaType(\"application/vnd.apple.mpegurl;charset=utf-8\"));\n ContentDisposition contentDisposition = ContentDisposition\n .builder(\"attachment\")\n .filename(FileUtils.getName(filePath), StandardCharsets.UTF_8)\n .build();\n headers.setContentDisposition(contentDisposition);\n\n return ResponseEntity.ok()\n .headers(headers)\n .body(textContent);\n }\n\n return ResponseEntity.status(302)\n .header(HttpHeaders.CACHE_CONTROL, \"no-cache, no-store, must-revalidate, private\")\n .header(HttpHeaders.PRAGMA, \"no-cache\")\n .header(HttpHeaders.EXPIRES, \"0\")\n .header(HttpHeaders.LOCATION, downloadUrl)\n .build();\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/link/service/ShortLinkService.java",
"content": "package im.zhaojun.zfile.module.link.service;\n\nimport cn.hutool.core.util.RandomUtil;\nimport com.baomidou.mybatisplus.core.conditions.Wrapper;\nimport com.baomidou.mybatisplus.extension.plugins.pagination.Page;\nimport im.zhaojun.zfile.core.exception.ErrorCode;\nimport im.zhaojun.zfile.core.exception.core.BizException;\nimport im.zhaojun.zfile.module.config.model.dto.LinkExpireDTO;\nimport im.zhaojun.zfile.module.config.model.dto.SystemConfigDTO;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.link.event.DeleteExpireLinkEvent;\nimport im.zhaojun.zfile.module.link.mapper.ShortLinkMapper;\nimport im.zhaojun.zfile.module.link.model.entity.ShortLink;\nimport im.zhaojun.zfile.module.storage.event.StorageSourceDeleteEvent;\nimport jakarta.annotation.Resource;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.aop.framework.AopContext;\nimport org.springframework.cache.annotation.CacheConfig;\nimport org.springframework.cache.annotation.CacheEvict;\nimport org.springframework.cache.annotation.Cacheable;\nimport org.springframework.context.ApplicationEventPublisher;\nimport org.springframework.context.event.EventListener;\nimport org.springframework.lang.Nullable;\nimport org.springframework.stereotype.Service;\nimport org.springframework.transaction.annotation.Transactional;\n\nimport java.util.Date;\nimport java.util.List;\nimport java.util.Objects;\n\n/**\n * 短链 Service\n *\n * @author zhaojun\n */\n@Service\n@Slf4j\n@CacheConfig(cacheNames = \"shortLink\")\npublic class ShortLinkService {\n\n @Resource\n private ShortLinkMapper shortLinkMapper;\n\n @Resource\n private SystemConfigService systemConfigService;\n\n @Resource\n private ApplicationEventPublisher applicationEventPublisher;\n\n /**\n * 根据短链接 key 查询短链接\n *\n * @param key\n * 短链接 key\n *\n * @return 短链接信息\n */\n @Cacheable(key = \"#key\", unless = \"#result == null\", condition = \"#key != null\")\n public ShortLink findByKey(String key) {\n return shortLinkMapper.findByKey(key);\n }\n\n /**\n * 根据存储源 ID 和 URL 查询短链接\n *\n * @param storageId\n * 存储源 ID\n *\n * @param url\n * 文件路径\n *\n * @return 短链接信息\n */\n public @Nullable ShortLink findByStorageIdAndUrl(Integer storageId, String url, @Nullable Date expireDate) {\n return shortLinkMapper.findByStorageIdAndUrl(storageId, url, expireDate);\n }\n\n\n /**\n * 为存储源指定路径生成短链接, 保证生成的短连接 key 是不同的 (如果是永久链接, 则不会重新生成)\n *\n * @param storageId\n * 存储源 id\n *\n * @param fullPath\n * 存储源路径\n *\n * @return 生成后的短链接信息\n */\n public ShortLink generatorShortLink(Integer storageId, String fullPath, Long expireTime) {\n boolean validate = checkExpireDateIsValidate(expireTime);\n if (!validate) {\n throw new BizException(ErrorCode.BIZ_EXPIRE_TIME_ILLEGAL);\n }\n\n // 永久链接不在重复生成\n if (Objects.equals(expireTime, ShortLink.PERMANENT_EXPIRE_TIME)) {\n ShortLink shortLink = findByStorageIdAndUrl(storageId, fullPath, ShortLink.PERMANENT_EXPIRE_DATE);\n if (shortLink != null) {\n return shortLink;\n }\n }\n\n ShortLink shortLink;\n String randomKey;\n int generateCount = 0;\n do {\n // 获取短链\n randomKey = RandomUtil.randomString(6);\n shortLink = ((ShortLinkService) AopContext.currentProxy()).findByKey(randomKey);\n generateCount++;\n } while (shortLink != null);\n\n shortLink = new ShortLink();\n shortLink.setStorageId(storageId);\n shortLink.setUrl(fullPath);\n shortLink.setCreateDate(new Date());\n shortLink.setShortKey(randomKey);\n\n if (expireTime == -1) {\n shortLink.setExpireDate(ShortLink.PERMANENT_EXPIRE_DATE);\n } else {\n shortLink.setExpireDate(new Date(System.currentTimeMillis() + expireTime * 1000L));\n }\n\n\n if (log.isDebugEnabled()) {\n log.debug(\"生成直/短链: 存储源 ID: {}, 文件路径: {}, 短链 key {}, 随机生成直链冲突次数: {}\",\n shortLink.getStorageId(), shortLink.getUrl(), shortLink.getShortKey(), generateCount);\n }\n\n shortLinkMapper.insert(shortLink);\n return shortLink;\n }\n\n @CacheEvict(allEntries = true)\n public int deleteExpireLink() {\n applicationEventPublisher.publishEvent(new DeleteExpireLinkEvent());\n int deleteSize = shortLinkMapper.deleteExpireLink();\n log.info(\"删除过期直/短链 {} 条\", deleteSize);\n return deleteSize;\n }\n\n @CacheEvict(allEntries = true)\n public void removeById(Integer id) {\n log.info(\"删除 id 为 {} 的直/短链\", id);\n shortLinkMapper.deleteById(id);\n }\n\n @Transactional(rollbackFor = Exception.class)\n @CacheEvict(allEntries = true)\n public void removeBatchByIds(List ids) {\n log.info(\"批量删除直/短链，id 集合为 {}\", ids);\n shortLinkMapper.deleteBatchIds(ids);\n }\n\n @CacheEvict(allEntries = true)\n public int deleteByStorageId(Integer storageId) {\n int deleteSize = shortLinkMapper.deleteByStorageId(storageId);\n log.info(\"删除存储源 ID 为 {} 的短链 {} 条\", storageId, deleteSize);\n return deleteSize;\n }\n\n /**\n * 监听存储源删除事件，根据存储源 id 删除相关的短链\n *\n * @param storageSourceDeleteEvent\n * 存储源删除事件\n */\n @EventListener\n public void onStorageSourceDelete(StorageSourceDeleteEvent storageSourceDeleteEvent) {\n Integer storageId = storageSourceDeleteEvent.getId();\n int updateRows = ((ShortLinkService) AopContext.currentProxy()).deleteByStorageId(storageId);\n if (log.isDebugEnabled()) {\n log.debug(\"删除存储源 [id {}, name: {}, type: {}] 时，关联删除存储源短链 {} 条\",\n storageId,\n storageSourceDeleteEvent.getName(),\n storageSourceDeleteEvent.getType().getDescription(),\n updateRows);\n }\n }\n\n public Page selectPage(Page pages, Wrapper queryWrapper) {\n return shortLinkMapper.selectPage(pages, queryWrapper);\n }\n\n private boolean checkExpireDateIsValidate(Long expires) {\n SystemConfigDTO systemConfig = systemConfigService.getSystemConfig();\n\n List linkExpireTimeList = systemConfig.getLinkExpireTimes();\n\n for (LinkExpireDTO linkExpireDTO : linkExpireTimeList) {\n if (linkExpireDTO.getSeconds().equals(expires)) {\n return true;\n }\n }\n return false;\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/log/controller/DownloadLogManagerController.java",
"content": "package im.zhaojun.zfile.module.log.controller;\n\nimport cn.hutool.core.util.ObjUtil;\nimport com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;\nimport com.baomidou.mybatisplus.core.metadata.OrderItem;\nimport com.baomidou.mybatisplus.extension.plugins.pagination.Page;\nimport com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\nimport com.github.xiaoymin.knife4j.annotations.ApiSort;\nimport im.zhaojun.zfile.core.annotation.DemoDisable;\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.module.config.model.dto.SystemConfigDTO;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.link.model.request.BatchDeleteRequest;\nimport im.zhaojun.zfile.module.link.model.request.QueryDownloadLogRequest;\nimport im.zhaojun.zfile.module.log.convert.DownloadLogConvert;\nimport im.zhaojun.zfile.module.log.model.entity.DownloadLog;\nimport im.zhaojun.zfile.module.log.model.result.DownloadLogResult;\nimport im.zhaojun.zfile.module.log.service.DownloadLogService;\nimport im.zhaojun.zfile.module.storage.model.entity.StorageSource;\nimport im.zhaojun.zfile.module.storage.service.StorageSourceService;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.Parameter;\nimport io.swagger.v3.oas.annotations.enums.ParameterIn;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport org.springframework.stereotype.Controller;\nimport org.springframework.web.bind.annotation.*;\n\nimport java.util.HashMap;\nimport java.util.List;\nimport java.util.Map;\nimport java.util.Objects;\nimport java.util.stream.Stream;\n\n/**\n * 直链下载日志接口\n *\n * @author zhaojun\n */\n@Tag(name = \"直链日志管理\")\n@ApiSort(7)\n@Controller\n@RequestMapping(\"/admin/download/log\")\npublic class DownloadLogManagerController {\n\n @Resource\n private StorageSourceService storageSourceService;\n\n @Resource\n private DownloadLogConvert downloadLogConvert;\n\n @Resource\n private DownloadLogService downloadLogService;\n\n @Resource\n private SystemConfigService systemConfigService;\n\n @ApiOperationSupport(order = 1)\n @GetMapping(\"/list\")\n @Operation(summary = \"直链下载日志\")\n @ResponseBody\n public AjaxJson> list(QueryDownloadLogRequest queryDownloadLogRequest) {\n // 分页和排序\n boolean asc = Objects.equals(queryDownloadLogRequest.getOrderDirection(), \"asc\");\n OrderItem orderItem = asc ? OrderItem.asc(queryDownloadLogRequest.getOrderBy()) : OrderItem.desc(queryDownloadLogRequest.getOrderBy());\n Page pages = new Page(queryDownloadLogRequest.getPage(), queryDownloadLogRequest.getLimit())\n .addOrder(orderItem);\n\n LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper()\n .eq(StringUtils.isNotEmpty(queryDownloadLogRequest.getStorageKey()), DownloadLog::getStorageKey, queryDownloadLogRequest.getStorageKey())\n .like(StringUtils.isNotEmpty(queryDownloadLogRequest.getPath()), DownloadLog::getPath, queryDownloadLogRequest.getPath())\n .isNotNull(\"shortLink\".equals(queryDownloadLogRequest.getLinkType()), DownloadLog::getShortKey)\n .isNull(\"directLink\".equals(queryDownloadLogRequest.getLinkType()), DownloadLog::getShortKey)\n .like(StringUtils.isNotEmpty(queryDownloadLogRequest.getShortKey()), DownloadLog::getShortKey, queryDownloadLogRequest.getShortKey())\n .like(StringUtils.isNotEmpty(queryDownloadLogRequest.getIp()), DownloadLog::getIp, queryDownloadLogRequest.getIp())\n .like(StringUtils.isNotEmpty(queryDownloadLogRequest.getReferer()), DownloadLog::getReferer, queryDownloadLogRequest.getReferer())\n .like(StringUtils.isNotEmpty(queryDownloadLogRequest.getUserAgent()), DownloadLog::getUserAgent, queryDownloadLogRequest.getUserAgent())\n .ge(ObjUtil.isNotEmpty(queryDownloadLogRequest.getDateFrom()), DownloadLog::getCreateTime, queryDownloadLogRequest.getDateFrom())\n .le(ObjUtil.isNotEmpty(queryDownloadLogRequest.getDateTo()), DownloadLog::getCreateTime, queryDownloadLogRequest.getDateTo());\n\n Page selectResult = downloadLogService.selectPage(pages, queryWrapper);\n\n Map cache = new HashMap<>();\n\n String serverAddress = systemConfigService.getAxiosFromDomainOrSetting();\n SystemConfigDTO systemConfig = systemConfigService.getSystemConfig();\n String directLinkPrefix = systemConfig.getDirectLinkPrefix();\n\n Stream shortLinkResultList = selectResult.getRecords().stream().map(model -> {\n String storageKey = model.getStorageKey();\n\n StorageSource storageSource = cache.computeIfAbsent(storageKey, (key) -> storageSourceService.findByStorageKey(key));\n DownloadLogResult downloadLogResult = downloadLogConvert.entityToResultList(model, storageSource);\n\n if (StringUtils.isNotBlank(downloadLogResult.getShortKey())) {\n downloadLogResult.setShortLink(StringUtils.concat(serverAddress, \"s\", downloadLogResult.getShortKey()));\n } else {\n downloadLogResult.setPathLink(StringUtils.concat(serverAddress, directLinkPrefix, downloadLogResult.getStorageKey(), downloadLogResult.getPath()));\n }\n\n return downloadLogResult;\n });\n return AjaxJson.getPageData(selectResult.getTotal(), shortLinkResultList);\n }\n\n\n @ApiOperationSupport(order = 2)\n @DeleteMapping(\"/delete/{id}\")\n @Operation(summary = \"删除直链\")\n @Parameter(in = ParameterIn.PATH, name = \"id\", description = \"直链 id\", required = true, schema = @Schema(type = \"integer\"))\n @ResponseBody\n @DemoDisable\n public AjaxJson deleteById(@PathVariable Integer id) {\n downloadLogService.removeById(id);\n return AjaxJson.getSuccess();\n }\n\n\n @ApiOperationSupport(order = 3)\n @PostMapping(\"/delete/batch\")\n @ResponseBody\n @Operation(summary = \"批量删除直链\")\n @DemoDisable\n public AjaxJson batchDelete(@RequestBody BatchDeleteRequest batchDeleteRequest) {\n List ids = batchDeleteRequest.getIds();\n downloadLogService.removeBatchByIds(ids);\n return AjaxJson.getSuccess();\n }\n\n @ApiOperationSupport(order = 4)\n @PostMapping(\"/delete/batch/query\")\n @ResponseBody\n @Operation(summary = \"根据查询条件批量删除直链\")\n @DemoDisable\n public AjaxJson batchDeleteBySearchParams(@RequestBody QueryDownloadLogRequest queryDownloadLogRequest) {\n\n LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper()\n .eq(StringUtils.isNotEmpty(queryDownloadLogRequest.getStorageKey()), DownloadLog::getStorageKey, queryDownloadLogRequest.getStorageKey())\n .like(StringUtils.isNotEmpty(queryDownloadLogRequest.getPath()), DownloadLog::getPath, queryDownloadLogRequest.getPath())\n .like(StringUtils.isNotEmpty(queryDownloadLogRequest.getShortKey()), DownloadLog::getShortKey, queryDownloadLogRequest.getShortKey())\n .like(StringUtils.isNotEmpty(queryDownloadLogRequest.getIp()), DownloadLog::getIp, queryDownloadLogRequest.getIp())\n .like(StringUtils.isNotEmpty(queryDownloadLogRequest.getReferer()), DownloadLog::getReferer, queryDownloadLogRequest.getReferer())\n .like(StringUtils.isNotEmpty(queryDownloadLogRequest.getUserAgent()), DownloadLog::getUserAgent, queryDownloadLogRequest.getUserAgent())\n .ge(ObjUtil.isNotEmpty(queryDownloadLogRequest.getDateFrom()), DownloadLog::getCreateTime, queryDownloadLogRequest.getDateFrom())\n .le(ObjUtil.isNotEmpty(queryDownloadLogRequest.getDateTo()), DownloadLog::getCreateTime, queryDownloadLogRequest.getDateTo());\n\n downloadLogService.deleteByQueryWrapper(queryWrapper);\n return AjaxJson.getSuccess();\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/log/controller/LoginLogController.java",
"content": "package im.zhaojun.zfile.module.log.controller;\n\nimport cn.hutool.core.util.ObjUtil;\nimport com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;\nimport com.baomidou.mybatisplus.core.metadata.OrderItem;\nimport com.baomidou.mybatisplus.extension.plugins.pagination.Page;\nimport com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\nimport com.github.xiaoymin.knife4j.annotations.ApiSort;\nimport im.zhaojun.zfile.core.util.AjaxJson;\nimport im.zhaojun.zfile.core.util.StringUtils;\nimport im.zhaojun.zfile.module.link.model.request.QueryLoginLogRequest;\nimport im.zhaojun.zfile.module.log.model.entity.LoginLog;\nimport im.zhaojun.zfile.module.log.service.LoginLogService;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport org.springframework.stereotype.Controller;\nimport org.springframework.web.bind.annotation.GetMapping;\nimport org.springframework.web.bind.annotation.RequestMapping;\nimport org.springframework.web.bind.annotation.ResponseBody;\n\nimport java.util.List;\nimport java.util.Objects;\n\n/**\n * 用户登录日志接口\n *\n * @author zhaojun\n */\n@Tag(name = \"登录日志管理\")\n@ApiSort(7)\n@Controller\n@RequestMapping(\"/admin/login/log\")\npublic class LoginLogController {\n \n @Resource\n private LoginLogService loginLogService;\n\n @ApiOperationSupport(order = 1)\n @GetMapping(\"/list\")\n @Operation(summary = \"登录日志列表\")\n @ResponseBody\n public AjaxJson> list(QueryLoginLogRequest queryLoginLogRequest) {\n // 分页和排序\n boolean asc = Objects.equals(queryLoginLogRequest.getOrderDirection(), \"asc\");\n OrderItem orderItem = asc ? OrderItem.asc(queryLoginLogRequest.getOrderBy()) : OrderItem.desc(queryLoginLogRequest.getOrderBy());\n Page pages = new Page(queryLoginLogRequest.getPage(), queryLoginLogRequest.getLimit())\n .addOrder(orderItem);\n\n LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper()\n .like(StringUtils.isNotEmpty(queryLoginLogRequest.getUsername()), LoginLog::getUsername, queryLoginLogRequest.getUsername())\n .like(StringUtils.isNotEmpty(queryLoginLogRequest.getPassword()), LoginLog::getPassword, queryLoginLogRequest.getPassword())\n .like(StringUtils.isNotEmpty(queryLoginLogRequest.getIp()), LoginLog::getIp, queryLoginLogRequest.getIp())\n .like(StringUtils.isNotEmpty(queryLoginLogRequest.getUserAgent()), LoginLog::getUserAgent, queryLoginLogRequest.getUserAgent())\n .like(StringUtils.isNotEmpty(queryLoginLogRequest.getReferer()), LoginLog::getReferer, queryLoginLogRequest.getReferer())\n .like(StringUtils.isNotEmpty(queryLoginLogRequest.getResult()), LoginLog::getResult, queryLoginLogRequest.getResult())\n .ge(ObjUtil.isNotEmpty(queryLoginLogRequest.getDateFrom()), LoginLog::getCreateTime, queryLoginLogRequest.getDateFrom())\n .le(ObjUtil.isNotEmpty(queryLoginLogRequest.getDateTo()), LoginLog::getCreateTime, queryLoginLogRequest.getDateTo());\n\n Page selectResult = loginLogService.selectPage(pages, queryWrapper);\n return AjaxJson.getPageData(selectResult.getTotal(), selectResult.getRecords());\n }\n \n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/log/convert/DownloadLogConvert.java",
"content": "package im.zhaojun.zfile.module.log.convert;\n\nimport im.zhaojun.zfile.module.log.model.entity.DownloadLog;\nimport im.zhaojun.zfile.module.storage.model.entity.StorageSource;\nimport im.zhaojun.zfile.module.log.model.result.DownloadLogResult;\nimport org.mapstruct.Mapper;\nimport org.mapstruct.Mapping;\nimport org.springframework.stereotype.Component;\n\n/**\n * 下载日志实体类转换器\n *\n * @author zhaojun\n */\n@Component\n@Mapper(componentModel = \"spring\")\npublic interface DownloadLogConvert {\n\n\t@Mapping(source = \"downloadLog.id\", target = \"id\")\n\t@Mapping(source = \"storageSource.name\", target = \"storageName\")\n\t@Mapping(source = \"storageSource.type\", target = \"storageType\")\n\tDownloadLogResult entityToResultList(DownloadLog downloadLog, StorageSource storageSource);\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/log/mapper/DownloadLogMapper.java",
"content": "package im.zhaojun.zfile.module.log.mapper;\n\nimport com.baomidou.mybatisplus.core.mapper.BaseMapper;\nimport im.zhaojun.zfile.module.log.model.entity.DownloadLog;\nimport org.apache.ibatis.annotations.Mapper;\n\n/**\n * 下载日志 Mapper 接口\n *\n * @author zhaojun\n */\n@Mapper\npublic interface DownloadLogMapper extends BaseMapper {\n\n\t/**\n\t * 根据存储源 KEY 删除所有数据\n\t *\n\t * @param \tstorageKey\n\t * \t\t\t存储源 KEY\n\t */\n\tint deleteByStorageKey(String storageKey);\n\n\n\t/**\n\t * 删除过期的短链下载日志\n\t *\n\t * @return\t删除的行数\n\t */\n\tint deleteExpireShortLinkLog();\n\t\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/log/mapper/LoginLogMapper.java",
"content": "package im.zhaojun.zfile.module.log.mapper;\n\nimport com.baomidou.mybatisplus.core.mapper.BaseMapper;\nimport im.zhaojun.zfile.module.log.model.entity.LoginLog;\nimport org.apache.ibatis.annotations.Mapper;\n\n@Mapper\npublic interface LoginLogMapper extends BaseMapper {\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/log/model/entity/DownloadLog.java",
"content": "package im.zhaojun.zfile.module.log.model.entity;\n\nimport cn.hutool.extra.servlet.JakartaServletUtil;\nimport com.baomidou.mybatisplus.annotation.IdType;\nimport com.baomidou.mybatisplus.annotation.TableField;\nimport com.baomidou.mybatisplus.annotation.TableId;\nimport com.baomidou.mybatisplus.annotation.TableName;\nimport im.zhaojun.zfile.core.util.RequestHolder;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.servlet.http.HttpServletRequest;\nimport lombok.Data;\nimport lombok.NoArgsConstructor;\nimport org.springframework.http.HttpHeaders;\n\nimport java.io.Serializable;\nimport java.util.Date;\n\n/**\n * 文件下载日志 entity\n *\n * @author zhaojun\n */\n@Data\n@Tag(name =\"文件下载日志\")\n@TableName(value = \"`download_log`\")\n@NoArgsConstructor\npublic class DownloadLog implements Serializable {\n\n public static final String DOWNLOAD_TYPE_DIRECT_LINK = \"directLink\";\n\n public static final String DOWNLOAD_TYPE_SHORT_LINK = \"shortLink\";\n\n private static final long serialVersionUID = 1L;\n\n @TableId(value = \"id\", type = IdType.INPUT)\n @Schema(title = \"ID, 新增无需填写\", example = \"1\")\n private Integer id;\n\n\n @TableField(value = \"`download_type`\")\n @Schema(title=\"下载类型\", example = \"directLink\", allowableValues = \"directLink, shortLink\")\n private String downloadType;\n\n\n @TableField(value = \"`path`\")\n @Schema(title=\"文件路径\")\n private String path;\n\n\n @TableField(value = \"`storage_key`\")\n @Schema(title=\"存储源 key\")\n private String storageKey;\n\n\n @TableField(value = \"`create_time`\")\n @Schema(title=\"访问时间\")\n private Date createTime;\n\n\n @TableField(value = \"`ip`\")\n @Schema(title=\"访问 ip\")\n private String ip;\n\n\n @TableField(value = \"short_key\")\n @Schema(title = \"短链 key\", example = \"voldd3\")\n private String shortKey;\n\n\n @TableField(value = \"`user_agent`\")\n @Schema(title=\"访问 user_agent\")\n private String userAgent;\n\n\n @TableField(value = \"`referer`\")\n @Schema(title=\"访问 referer\")\n private String referer;\n\n public DownloadLog(String downloadType, String path, String storageKey, String shortKey) {\n this.downloadType = downloadType;\n this.path = path;\n this.storageKey = storageKey;\n this.shortKey = shortKey;\n this.createTime = new Date();\n HttpServletRequest request = RequestHolder.getRequest();\n this.ip = JakartaServletUtil.getClientIP(request);\n this.referer = request.getHeader(HttpHeaders.REFERER);\n this.userAgent = request.getHeader(HttpHeaders.USER_AGENT);\n }\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/log/model/entity/LoginLog.java",
"content": "package im.zhaojun.zfile.module.log.model.entity;\n\nimport com.baomidou.mybatisplus.annotation.*;\nimport lombok.Data;\n\nimport java.io.Serializable;\nimport java.util.Date;\n\n@Data\n@TableName(value = \"login_log\")\npublic class LoginLog implements Serializable {\n\n private static final long serialVersionUID = 1L;\n\n @TableId(value = \"id\", type = IdType.INPUT)\n private Integer id;\n\n @TableField(value = \"username\")\n private String username;\n\n @TableField(value = \"`password`\")\n private String password;\n\n @TableField(value = \"create_time\", fill = FieldFill.INSERT)\n private Date createTime;\n\n @TableField(value = \"ip\")\n private String ip;\n\n @TableField(value = \"user_agent\")\n private String userAgent;\n\n @TableField(value = \"referer\")\n private String referer;\n\n @TableField(value = \"`result`\")\n private String result;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/log/model/result/DownloadLogResult.java",
"content": "package im.zhaojun.zfile.module.log.model.result;\n\nimport im.zhaojun.zfile.module.storage.model.enums.StorageTypeEnum;\nimport io.swagger.v3.oas.annotations.media.Schema;\nimport lombok.Data;\n\nimport java.util.Date;\n\n/**\n * 下载日志结果类\n *\n * @author zhaojun\n */\n@Data\npublic class DownloadLogResult {\n\n\t@Schema(title=\"\")\n\tprivate Integer id;\n\n\t@Schema(title=\"文件路径\")\n\tprivate String path;\n\n\t@Schema(title = \"存储源类型\")\n\tprivate StorageTypeEnum storageType;\n\t\n\t@Schema(title = \"存储源名称\", example = \"我的本地存储\")\n\tprivate String storageName;\n\n\t@Schema(title = \"存储源Key\", example = \"local\")\n\tprivate String storageKey;\n\n\t@Schema(title=\"访问时间\")\n\tprivate Date createTime;\n\n\t@Schema(title=\"访问 ip\")\n\tprivate String ip;\n\t\n\t@Schema(title = \"短链 Key\")\n\tprivate String shortKey;\n\n\t@Schema(title=\"访问 user_agent\")\n\tprivate String userAgent;\n\n\t@Schema(title=\"访问 referer\")\n\tprivate String referer;\n\n\t@Schema(title=\"短链地址\")\n\tprivate String shortLink;\n\n\t@Schema(title=\"直链地址\")\n\tprivate String pathLink;\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/log/service/DownloadLogService.java",
"content": "package im.zhaojun.zfile.module.log.service;\n\nimport com.baomidou.mybatisplus.core.conditions.Wrapper;\nimport com.baomidou.mybatisplus.extension.plugins.pagination.Page;\nimport im.zhaojun.zfile.module.link.event.DeleteExpireLinkEvent;\nimport im.zhaojun.zfile.module.log.mapper.DownloadLogMapper;\nimport im.zhaojun.zfile.module.log.model.entity.DownloadLog;\nimport im.zhaojun.zfile.module.storage.event.StorageSourceDeleteEvent;\nimport jakarta.annotation.Resource;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.aop.framework.AopContext;\nimport org.springframework.context.event.EventListener;\nimport org.springframework.stereotype.Service;\nimport org.springframework.transaction.annotation.Transactional;\n\nimport java.util.List;\n\n/**\n * 下载日志 Service\n *\n * @author zhaojun\n */\n@Slf4j\n@Service\npublic class DownloadLogService {\n\n\t@Resource\n\tprivate DownloadLogMapper downloadLogMapper;\n\n\tpublic void save(DownloadLog downloadLog) {\n\t\tdownloadLogMapper.insert(downloadLog);\n\t}\n\n\tpublic Page selectPage(Page pages, Wrapper queryWrapper) {\n\t\treturn downloadLogMapper.selectPage(pages, queryWrapper);\n\t}\n\n\tpublic void removeById(Integer id) {\n\t\tdownloadLogMapper.deleteById(id);\n\t}\n\n\t@Transactional(rollbackFor = Exception.class)\n\tpublic void removeBatchByIds(List ids) {\n\t\tdownloadLogMapper.deleteBatchIds(ids);\n\t}\n\n\tpublic void deleteByQueryWrapper(Wrapper queryWrapper) {\n\t\tdownloadLogMapper.delete(queryWrapper);\n\t}\n\n\tpublic int deleteByStorageKey(String storageKey) {\n\t\tint deleteSize = downloadLogMapper.deleteByStorageKey(storageKey);\n\t\tlog.info(\"删除存储源 ID 为 {} 的直/短链下载日志 {} 条\", storageKey, deleteSize);\n\t\treturn deleteSize;\n\t}\n\n\t/**\n\t * 监听存储源删除事件，根据存储源 id 删除相关的下载日志\n\t *\n\t * @param storageSourceDeleteEvent\n\t * 存储源删除事件\n\t */\n\t@EventListener\n\tpublic void onStorageSourceDelete(StorageSourceDeleteEvent storageSourceDeleteEvent) {\n\t\tString storageKey = storageSourceDeleteEvent.getKey();\n\t\tint updateRows = ((DownloadLogService) AopContext.currentProxy()).deleteByStorageKey(storageKey);\n\t\tif (log.isDebugEnabled()) {\n\t\t\tlog.debug(\"删除存储源 [id {}, key: {}, name: {}, type: {}] 时，关联删除存储源直/短链下载日志 {} 条\",\n\t\t\t\t\tstorageSourceDeleteEvent.getId(),\n\t\t\t\t\tstorageKey,\n\t\t\t\t\tstorageSourceDeleteEvent.getName(),\n\t\t\t\t\tstorageSourceDeleteEvent.getType().getDescription(),\n\t\t\t\t\tupdateRows);\n\t\t}\n\t}\n\n\t/**\n\t * 删除过期下载日志\n\t *\n\t * @return 删除的条数\n\t */\n\tpublic int deleteExpireShortLinkLog() {\n\t\treturn downloadLogMapper.deleteExpireShortLinkLog();\n\t}\n\n\t@EventListener(classes = DeleteExpireLinkEvent.class)\n\tpublic void deleteExpireShortLinkLog(DeleteExpireLinkEvent event) {\n\t\tint updateRows = deleteExpireShortLinkLog();\n\t\tlog.info(\"删除过期短链关联删除日志 {} 条\", updateRows);\n\t}\n\n}"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/log/service/LoginLogService.java",
"content": "package im.zhaojun.zfile.module.log.service;\n\nimport com.baomidou.mybatisplus.core.conditions.Wrapper;\nimport com.baomidou.mybatisplus.extension.plugins.pagination.Page;\nimport im.zhaojun.zfile.module.log.mapper.LoginLogMapper;\nimport im.zhaojun.zfile.module.log.model.entity.LoginLog;\nimport org.springframework.stereotype.Service;\n\nimport jakarta.annotation.Resource;\n\n@Service\npublic class LoginLogService {\n\n @Resource\n private LoginLogMapper loginLogMapper;\n\n public void save(LoginLog loginLog) {\n loginLogMapper.insert(loginLog);\n }\n\n public Page selectPage(Page pages, Wrapper queryWrapper) {\n return loginLogMapper.selectPage(pages, queryWrapper);\n }\n}\n"
},
{
"path": "src/main/java/im/zhaojun/zfile/module/onlyoffice/controller/OnlyOfficeController.java",
"content": "package im.zhaojun.zfile.module.onlyoffice.controller;\n\nimport cn.dev33.satoken.stp.StpUtil;\nimport cn.hutool.core.collection.CollUtil;\nimport cn.hutool.core.util.StrUtil;\nimport cn.hutool.jwt.JWTUtil;\nimport com.alibaba.fastjson2.JSONObject;\nimport com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\nimport im.zhaojun.zfile.module.onlyoffice.model.OnlyOfficeCallback;\nimport im.zhaojun.zfile.core.exception.biz.InvalidStorageSourceBizException;\nimport im.zhaojun.zfile.core.exception.core.BizException;\nimport im.zhaojun.zfile.core.util.*;\nimport im.zhaojun.zfile.module.onlyoffice.model.OnlyOfficeFile;\nimport im.zhaojun.zfile.module.config.model.dto.SystemConfigDTO;\nimport im.zhaojun.zfile.module.config.service.SystemConfigService;\nimport im.zhaojun.zfile.module.storage.annotation.CheckPassword;\nimport im.zhaojun.zfile.module.storage.context.StorageSourceContext;\nimport im.zhaojun.zfile.module.storage.model.enums.FileOperatorTypeEnum;\nimport im.zhaojun.zfile.module.storage.model.request.base.FileItemRequest;\nimport im.zhaojun.zfile.module.storage.model.result.FileItemResult;\nimport im.zhaojun.zfile.module.storage.service.StorageSourceService;\nimport im.zhaojun.zfile.module.storage.service.base.AbstractBaseFileService;\nimport im.zhaojun.zfile.module.storage.service.base.AbstractProxyTransferService;\nimport im.zhaojun.zfile.module.user.model.entity.User;\nimport im.zhaojun.zfile.module.user.service.UserStorageSourceService;\nimport io.swagger.v3.oas.annotations.Operation;\nimport io.swagger.v3.oas.annotations.tags.Tag;\nimport jakarta.annotation.Resource;\nimport jakarta.validation.Valid;\nimport lombok.extern.slf4j.Slf4j;\nimport org.apache.commons.lang3.tuple.Pair;\nimport org.springframework.web.bind.annotation.PostMapping;\nimport org.springframework.web.bind.annotation.RequestBody;\nimport org.springframework.web.bind.annotation.RequestMapping;\nimport org.springframework.web.bind.annotation.RestController;\n\nimport java.beans.Beans;\nimport java.io.InputStream;\nimport java.net.URI;\nimport java.net.URL;\nimport java.net.URLConnection;\nimport java.nio.charset.StandardCharsets;\nimport java.util.List;\nimport java.util.concurrent.locks.ReentrantLock;\n\n@Slf4j\n@Tag(name = \"OnlyOffice 相关接口\")\n@RestController\n@RequestMapping(\"/onlyOffice\")\npublic class OnlyOfficeController {\n\n @Resource\n private SystemConfigService systemConfigService;\n\n @Resource\n private StorageSourceService storageSourceService;\n\n @Resource\n private UserStorageSourceService userStorageSourceService;\n\n private static final String CALLBACK_ERROR_MSG = \"{\\\"error\\\":1}\";\n\n private static final String CALLBACK_SUCCESS_MSG = \"{\\\"error\\\":0}\";\n\n public static final List SUPPORTED_STATUS = List.of(2, 3, 6, 7);\n\n @ApiOperationSupport(order = 3)\n @Operation(summary = \"OnlyOffice 预览文件\", description = \"根据传入的文件信息, 生成 OnlyOffice 预览所需的 JSON 数据.\")\n @PostMapping(\"/config/token\")\n @CheckPassword(storageKeyFieldExpression = \"[0].storageKey\",\n pathFieldExpression = \"[0].path\",\n pathIsDirectory = false,\n passwordFieldExpression = \"[0].password\")\n public AjaxJson getPreviewFileJSONInfo(@Valid @RequestBody FileItemRequest fileItemRequest) {\n // 根据存储策略获取文件信息(下载地址), 会校验权限.\n Pair pair = getFileInfo(fileItemRequest);\n FileItemResult fileInfo = pair.getKey();\n Boolean hasUploadPermission = pair.getRight();\n\n // 为 OnlyOffice 获取或生成文件 Key.\n OnlyOfficeFile onlyOfficeFile = new OnlyOfficeFile(fileItemRequest.getStorageKey(), fileItemRequest.getPath());\n String key = OnlyOfficeKeyCacheUtils.getKeyOrPutNew(onlyOfficeFile, 3000);\n\n JSONObject onlyOfficePayload = createOnlyOfficePayload(fileInfo, key, hasUploadPermission);\n return AjaxJson.getSuccessData(onlyOfficePayload);\n }\n\n private Pair
![\"ZFile\"](\"/img/logo-zfile.png\"){kind=logo}
\n \n ![\"last](\"https://shields.io/github/last-commit/zfile-dev/zfile.svg?style=flat-square\"/)
\n ![\"downloads\"](\"https://shields.io/github/downloads/zfile-dev/zfile/total?style=flat-square\"/)
\n ![\"release](\"https://shields.io/github/v/release/zfile-dev/zfile?style=flat-square\"/)
\n ![\"commit](\"https://shields.io/github/commit-activity/y/zfile-dev/zfile?style=flat-square\"/)
\n ![\"open](\"https://shields.io/github/issues/zfile-dev/zfile?style=flat-square\"/)
\n ![\"closed](\"https://shields.io/github/issues-closed-raw/zfile-dev/zfile?style=flat-square\"/)
\n ![\"forks\"](\"https://shields.io/github/forks/zfile-dev/zfile?style=flat-square\"/)
\n ![\"stars\"](\"https://shields.io/github/stars/zfile-dev/zfile?style=flat-square\"/)
\n ![\"watchers\"](\"https://shields.io/github/watchers/zfile-dev/zfile?style=flat-square\"/)
\n ![\"gitcode\"](\"https://gitcode.com/zfile-dev/zfile/star/badge.svg\"/)
\n![\"赞助我\"](\"https://cdn.jun6.net/2021/03/27/152704e91f13d.png\")
\n\n## Star History\n\n[](https://star-history.com/#zfile-dev/zfile&Date)\n"
},
{
"path": "pom.xml",
"content": "\n