hatter/tiny-encrypt-java
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat works

Browse Source
This commit is contained in:
Hatter Jiang
2021-04-16 01:30:25 +08:00
parent b0acd62ccc
commit c5a372374b
3 changed files with 23 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/me/hatter/tools/tinyencrypt/TinyEncryptArgs.java
View File

@@ -21,6 +21,9 @@ public class TinyEncryptArgs {
@CommandLine.Option(names = {"--config"}, description = "Encrypt config")
File config;
@CommandLine.Option(names = {"--remove-file"}, description = "Remove origin config")
boolean removeFile = false;
@CommandLine.Parameters(paramLabel = "FILE", description = "Encrypt or Decrypt files")
File[] files;

13
src/main/java/me/hatter/tools/tinyencrypt/TinyEncryptMain.java
View File

@@ -105,10 +105,19 @@ public class TinyEncryptMain {
int index = 1;
for (File f : tinyEncryptArgs.files) {
log.info("Start processing file: " + f + ", " + index + " of " + total);
if (!f.isFile()) {
log.info("Skip not a file: " + f);
continue;
}
boolean result;
if (tinyEncryptArgs.encrypt) {
EncryptedFileUtil.encryptFile(config, tinyEncryptArgs.key, f, tinyEncryptArgs.comment);
result = EncryptedFileUtil.encryptFile(config, tinyEncryptArgs.key, f, tinyEncryptArgs.comment);
} else {
EncryptedFileUtil.decryptFile(config, f);
result = EncryptedFileUtil.decryptFile(config, f);
}
if (result && tinyEncryptArgs.removeFile) {
log.info("Remove file: " + f);
f.delete();
}
index++;
}

14
src/main/java/me/hatter/tools/tinyencrypt/encrypt/EncryptedFileUtil.java
View File

@@ -21,15 +21,15 @@ public class EncryptedFileUtil {
BCUtil.init();
}
public static void decryptFile(TinyEncryptConfig config, File file) {
public static boolean decryptFile(TinyEncryptConfig config, File file) {
File decFile = getDecryptFile(file);
if (decFile == null) {
log.warn("File is not tinyenc file, skip: " + decFile);
return;
return false;
}
if (decFile.exists()) {
log.warn("File exists, skip: " + decFile);
return;
return false;
}
try {
try (FileInputStream fis = new FileInputStream(file)) {
@@ -49,17 +49,19 @@ public class EncryptedFileUtil {
}
}
log.info("Decrypt file success: " + file);
return true;
} catch (Exception e) {
log.error("Decrypt file filed: " + file + ", reason: " + e.getMessage());
log.debug("Decrypt file filed: " + file + ", reason: " + e.getMessage(), e);
return false;
}
}
public static void encryptFile(TinyEncryptConfig config, String keyName, File file, String comment) {
public static boolean encryptFile(TinyEncryptConfig config, String keyName, File file, String comment) {
File encFile = getEncryptFile(file);
if (encFile.exists()) {
log.warn("File exists, skip: " + encFile);
return;
return false;
}
try {
try (FileInputStream fis = new FileInputStream(file)) {
@@ -78,9 +80,11 @@ public class EncryptedFileUtil {
}
}
log.info("Encrypt file success: " + file);
return true;
} catch (Exception e) {
log.error("Encrypt file filed: " + file + ", reason: " + e.getMessage());
log.debug("Encrypt file filed: " + file + ", reason: " + e.getMessage(), e);
return false;
}
}