mirror of
https://github.com/RatzzFatzz/MKVAudioSubtitleChanger.git
synced 2026-02-11 02:05:56 +01:00
[IMPL] path supports dir and files now
This commit is contained in:
RatzzFatzz
@@ -1,87 +1,21 @@
|
|||||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
||||||
import config.MKVToolProperties;
|
import config.MKVToolProperties;
|
||||||
import lombok.extern.log4j.Log4j2;
|
import lombok.extern.log4j.Log4j2;
|
||||||
import model.FileAttribute;
|
import query.QueryBuilder;
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.io.InputStream;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.Scanner;
|
import java.util.Scanner;
|
||||||
|
|
||||||
@Log4j2
|
@Log4j2
|
||||||
public class Main {
|
public class Main {
|
||||||
private static ObjectMapper mapper = new ObjectMapper();
|
|
||||||
|
|
||||||
public static void main(String[] args) {
|
public static void main(String[] args) {
|
||||||
MKVToolProperties.getInstance().defineMKVToolNixPath();
|
MKVToolProperties.getInstance().defineMKVToolNixPath();
|
||||||
Scanner input = new Scanner(System.in);
|
Scanner input = new Scanner(System.in);
|
||||||
log.info("Please enter path to file");
|
log.info("Please enter path to file");
|
||||||
String path = input.nextLine();
|
String path = input.nextLine();
|
||||||
updateAttributes(path, queryAttributes(path));
|
QueryBuilder queryBuilder = new QueryBuilder();
|
||||||
|
queryBuilder.executeUpdateOnAllFiles(path);
|
||||||
}
|
}
|
||||||
|
|
||||||
public static List<FileAttribute> queryAttributes(String path) {
|
|
||||||
Map<String, Object> jsonMap;
|
|
||||||
List<FileAttribute> fileAttributes = new ArrayList<>();
|
|
||||||
try(InputStream inputStream = Runtime.getRuntime().exec("\"" + MKVToolProperties.getInstance().getMkvmergePath() + "\" --identify --identification-format json \"" + path + "\"").getInputStream()){
|
|
||||||
jsonMap = mapper.readValue(inputStream, Map.class);
|
|
||||||
List<Map<String, Object>> tracks = (List<Map<String, Object>>) jsonMap.get("tracks");
|
|
||||||
|
|
||||||
for(Map<String, Object> attribute : tracks){
|
|
||||||
if(! "video".equals(attribute.get("type"))){
|
|
||||||
Map<String, Object> properties = (Map<String, Object>) attribute.get("properties");
|
|
||||||
fileAttributes.add(new FileAttribute(
|
|
||||||
(int) properties.get("number"),
|
|
||||||
(String) properties.get("language"),
|
|
||||||
(String) properties.get("track_name"),
|
|
||||||
(Boolean) properties.get("default_track"),
|
|
||||||
(Boolean) properties.get("forced_track"),
|
|
||||||
(String) attribute.get("type")));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}catch(IOException e){
|
|
||||||
log.error("File could not be found or loaded!");
|
|
||||||
}
|
|
||||||
return fileAttributes;
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void updateAttributes(String path, List<FileAttribute> fileAttributes) {
|
|
||||||
if(fileAttributes.size() > 2){
|
|
||||||
StringBuffer stringBuffer = new StringBuffer("\"");
|
|
||||||
stringBuffer.append(MKVToolProperties.getInstance().getMkvpropeditPath());
|
|
||||||
stringBuffer.append("\" \"");
|
|
||||||
stringBuffer.append(path);
|
|
||||||
stringBuffer.append("\" ");
|
|
||||||
|
|
||||||
int c = 0;
|
|
||||||
int d = 0;
|
|
||||||
|
|
||||||
for(FileAttribute attributes : fileAttributes){
|
|
||||||
if(attributes.isDefaultTrack() && "audio".equals(attributes.getType())){
|
|
||||||
stringBuffer.append("--edit track:" + attributes.getId() + " --set flag-default=0 ");
|
|
||||||
}
|
|
||||||
if(attributes.isDefaultTrack() && "subtitles".equals(attributes.getType())){
|
|
||||||
stringBuffer.append("--edit track:" + attributes.getId() + " --set flag-default=0 ");
|
|
||||||
}
|
|
||||||
if("jpn".equals(attributes.getLanguage()) && "audio".equals(attributes.getType()) && c == 0){
|
|
||||||
c++;
|
|
||||||
stringBuffer.append("--edit track:" + attributes.getId() + " --set flag-default=1 ");
|
|
||||||
}
|
|
||||||
if("eng".equals(attributes.getLanguage()) && "subtitles".equals(attributes.getType()) && d == 0){
|
|
||||||
d++;
|
|
||||||
stringBuffer.append("--edit track:" + attributes.getId() + " --set flag-default=1 ");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
try{
|
|
||||||
Runtime.getRuntime().exec(stringBuffer.toString());
|
|
||||||
}catch(IOException e){
|
|
||||||
log.error("Couldn't make changes to file");
|
|
||||||
|
|
||||||
}
|
}
|
||||||
}else{
|
|
||||||
log.info("There were not enough lines provided to make any changes to the file");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
112
src/main/java/query/QueryBuilder.java
Normal file
112
src/main/java/query/QueryBuilder.java
Normal file
@@ -0,0 +1,112 @@
|
|||||||
|
package query;
|
||||||
|
|
||||||
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||||
|
import config.MKVToolProperties;
|
||||||
|
import lombok.extern.log4j.Log4j2;
|
||||||
|
import model.FileAttribute;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.nio.file.Files;
|
||||||
|
import java.nio.file.Path;
|
||||||
|
import java.nio.file.Paths;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
import java.util.stream.Stream;
|
||||||
|
|
||||||
|
@Log4j2
|
||||||
|
public class QueryBuilder {
|
||||||
|
private ObjectMapper mapper = new ObjectMapper();
|
||||||
|
|
||||||
|
public QueryBuilder() {
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean executeUpdateOnAllFiles(String path) {
|
||||||
|
List<String> allFilePaths = getAllFilesFromDirectory(path);
|
||||||
|
if(allFilePaths == null){
|
||||||
|
log.error("Couldn't process path!");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
for(String filePath : allFilePaths){
|
||||||
|
updateAttributes(filePath, queryAttributes(filePath));
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
private List<String> getAllFilesFromDirectory(String path) {
|
||||||
|
try(Stream<Path> paths = Files.walk(Paths.get(path))){
|
||||||
|
return paths
|
||||||
|
.filter(Files::isRegularFile)
|
||||||
|
.map(file -> file.toAbsolutePath().toString())
|
||||||
|
.collect(Collectors.toList());
|
||||||
|
}catch(IOException e){
|
||||||
|
log.error("Couldn't find file or directory!", e);
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
private List<FileAttribute> queryAttributes(String path) {
|
||||||
|
Map<String, Object> jsonMap;
|
||||||
|
List<FileAttribute> fileAttributes = new ArrayList<>();
|
||||||
|
try(InputStream inputStream = Runtime.getRuntime().exec("\"" + MKVToolProperties.getInstance().getMkvmergePath() + "\" --identify --identification-format json \"" + path + "\"").getInputStream()){
|
||||||
|
jsonMap = mapper.readValue(inputStream, Map.class);
|
||||||
|
List<Map<String, Object>> tracks = (List<Map<String, Object>>) jsonMap.get("tracks");
|
||||||
|
|
||||||
|
for(Map<String, Object> attribute : tracks){
|
||||||
|
if(! "video".equals(attribute.get("type"))){
|
||||||
|
Map<String, Object> properties = (Map<String, Object>) attribute.get("properties");
|
||||||
|
fileAttributes.add(new FileAttribute(
|
||||||
|
(int) properties.get("number"),
|
||||||
|
(String) properties.get("language"),
|
||||||
|
(String) properties.get("track_name"),
|
||||||
|
(Boolean) properties.get("default_track"),
|
||||||
|
(Boolean) properties.get("forced_track"),
|
||||||
|
(String) attribute.get("type")));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}catch(IOException e){
|
||||||
|
log.error("File could not be found or loaded!");
|
||||||
|
}
|
||||||
|
return fileAttributes;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void updateAttributes(String path, List<FileAttribute> fileAttributes) {
|
||||||
|
if(fileAttributes.size() > 2){
|
||||||
|
StringBuffer stringBuffer = new StringBuffer("\"");
|
||||||
|
stringBuffer.append(MKVToolProperties.getInstance().getMkvpropeditPath());
|
||||||
|
stringBuffer.append("\" \"");
|
||||||
|
stringBuffer.append(path);
|
||||||
|
stringBuffer.append("\" ");
|
||||||
|
|
||||||
|
int c = 0;
|
||||||
|
int d = 0;
|
||||||
|
|
||||||
|
for(FileAttribute attributes : fileAttributes){
|
||||||
|
if(attributes.isDefaultTrack() && "audio".equals(attributes.getType())){
|
||||||
|
stringBuffer.append("--edit track:" + attributes.getId() + " --set flag-default=0 ");
|
||||||
|
}
|
||||||
|
if(attributes.isDefaultTrack() && "subtitles".equals(attributes.getType())){
|
||||||
|
stringBuffer.append("--edit track:" + attributes.getId() + " --set flag-default=0 ");
|
||||||
|
}
|
||||||
|
if("jpn".equals(attributes.getLanguage()) && "audio".equals(attributes.getType()) && c == 0){
|
||||||
|
c++;
|
||||||
|
stringBuffer.append("--edit track:" + attributes.getId() + " --set flag-default=1 ");
|
||||||
|
}
|
||||||
|
if("eng".equals(attributes.getLanguage()) && "subtitles".equals(attributes.getType()) && d == 0){
|
||||||
|
d++;
|
||||||
|
stringBuffer.append("--edit track:" + attributes.getId() + " --set flag-default=1 ");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
try{
|
||||||
|
Runtime.getRuntime().exec(stringBuffer.toString());
|
||||||
|
}catch(IOException e){
|
||||||
|
log.error("Couldn't make changes to file");
|
||||||
|
|
||||||
|
}
|
||||||
|
}else{
|
||||||
|
log.info("There were not enough lines provided to make any changes to the file");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user