refactor: 重构 APT 配置读取类。

mybatis-flex-processor/src/main/java/com/mybatisflex/processor/config/MybatisFlexConfig.java

@@ -0,0 +1,92 @@
+/*
+ *  Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com).
+ *  <p>
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *  <p>
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *  <p>
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+
+package com.mybatisflex.processor.config;
+
+import javax.annotation.processing.Filer;
+import javax.tools.FileObject;
+import javax.tools.StandardLocation;
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.util.Properties;
+
+/**
+ * Mybatis Flex 生成配置。
+ *
+ * @author 王帅
+ * @since 2023-06-22
+ */
+public class MybatisFlexConfig {
+
+    /**
+     * 配置文件名。
+     */
+    private static final String MYBATIS_FLEX = "mybatis-flex.properties";
+
+    /**
+     * mybatis-flex.properties
+     */
+    protected final Properties properties = new Properties();
+
+    public MybatisFlexConfig(Filer filer) {
+        InputStream inputStream = null;
+        try {
+            FileObject propertiesFileObject = filer.getResource(StandardLocation.CLASS_OUTPUT, "", MYBATIS_FLEX);
+
+            File propertiesFile = new File(propertiesFileObject.toUri());
+
+            if (propertiesFile.exists()) {
+                inputStream = propertiesFileObject.openInputStream();
+            } else if (getClass().getClassLoader().getResource(MYBATIS_FLEX) != null) {
+                inputStream = getClass().getClassLoader().getResourceAsStream(MYBATIS_FLEX);
+            } else {
+                File pomXmlFile = new File(propertiesFile.getParentFile().getParentFile().getParentFile(), "pom.xml");
+                if (pomXmlFile.exists()) {
+                    propertiesFile = new File(pomXmlFile.getParentFile(), "src/main/resources/mybatis-flex.properties");
+                }
+            }
+
+            if (inputStream == null && propertiesFile.exists()) {
+                inputStream = Files.newInputStream(propertiesFile.toPath());
+            }
+
+            if (inputStream != null) {
+                try (InputStreamReader reader = new InputStreamReader(inputStream, StandardCharsets.UTF_8)) {
+                    properties.load(reader);
+                }
+            }
+        } catch (Exception ignored) {
+            // do nothing here.
+        } finally {
+            if (inputStream != null) {
+                try {
+                    inputStream.close();
+                } catch (IOException ignored) {
+                    // do nothing here.
+                }
+            }
+        }
+    }
+
+    public String get(ConfigurationKey key) {
+        return properties.getProperty(key.getConfigKey(), key.getDefaultValue());
+    }
+
+}