1
0
mirror of https://github.com/octoleo/plantuml.git synced 2024-06-06 02:10:53 +00:00

Merge pull request #665 from matthew16550/SourceFileReaderAbstract

Extract common code to SourceFileReaderAbstract constructor
This commit is contained in:
arnaudroques 2021-09-12 12:33:12 +02:00 committed by GitHub
commit 061ce99bc5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 20 additions and 32 deletions

View File

@ -38,12 +38,10 @@ package net.sourceforge.plantuml;
import java.io.File; import java.io.File;
import java.io.FileNotFoundException; import java.io.FileNotFoundException;
import java.io.IOException; import java.io.IOException;
import java.io.Reader;
import java.util.Collections; import java.util.Collections;
import java.util.List; import java.util.List;
import net.sourceforge.plantuml.preproc.Defines; import net.sourceforge.plantuml.preproc.Defines;
import net.sourceforge.plantuml.preproc.FileWithSuffix;
import net.sourceforge.plantuml.security.SFile; import net.sourceforge.plantuml.security.SFile;
public class SourceFileReader extends SourceFileReaderAbstract implements ISourceFileReader { public class SourceFileReader extends SourceFileReaderAbstract implements ISourceFileReader {
@ -70,11 +68,7 @@ public class SourceFileReader extends SourceFileReaderAbstract implements ISourc
public SourceFileReader(Defines defines, final File file, File outputDirectory, List<String> config, String charset, public SourceFileReader(Defines defines, final File file, File outputDirectory, List<String> config, String charset,
FileFormatOption fileFormatOption) throws IOException { FileFormatOption fileFormatOption) throws IOException {
this.file = file; super(file, fileFormatOption, defines, config, charset);
this.fileFormatOption = fileFormatOption;
if (file.exists() == false) {
throw new IllegalArgumentException();
}
FileSystem.getInstance().setCurrentDir(SFile.fromFile(file.getAbsoluteFile().getParentFile())); FileSystem.getInstance().setCurrentDir(SFile.fromFile(file.getAbsoluteFile().getParentFile()));
if (outputDirectory == null) { if (outputDirectory == null) {
outputDirectory = file.getAbsoluteFile().getParentFile(); outputDirectory = file.getAbsoluteFile().getParentFile();
@ -86,9 +80,6 @@ public class SourceFileReader extends SourceFileReaderAbstract implements ISourc
} }
this.outputDirectory = outputDirectory; this.outputDirectory = outputDirectory;
final Reader reader = getReader(charset);
builder = new BlockUmlBuilder(config, charset, defines, reader,
SFile.fromFile(file.getAbsoluteFile().getParentFile()), FileWithSuffix.getFileName(file));
} }
private File getDirIfDirectory(String newName) throws FileNotFoundException { private File getDirIfDirectory(String newName) throws FileNotFoundException {

View File

@ -53,6 +53,7 @@ import java.util.Set;
import net.sourceforge.plantuml.api.ImageDataSimple; import net.sourceforge.plantuml.api.ImageDataSimple;
import net.sourceforge.plantuml.core.Diagram; import net.sourceforge.plantuml.core.Diagram;
import net.sourceforge.plantuml.error.PSystemError; import net.sourceforge.plantuml.error.PSystemError;
import net.sourceforge.plantuml.preproc.Defines;
import net.sourceforge.plantuml.preproc.FileWithSuffix; import net.sourceforge.plantuml.preproc.FileWithSuffix;
import net.sourceforge.plantuml.security.SFile; import net.sourceforge.plantuml.security.SFile;
import net.sourceforge.plantuml.security.SecurityUtils; import net.sourceforge.plantuml.security.SecurityUtils;
@ -68,6 +69,19 @@ public abstract class SourceFileReaderAbstract implements ISourceFileReader {
private boolean checkMetadata; private boolean checkMetadata;
private boolean noerror; private boolean noerror;
public SourceFileReaderAbstract(File file, FileFormatOption fileFormatOption, Defines defines, List<String> config, String charset)
throws IOException {
if (!file.exists()) {
throw new IllegalArgumentException();
}
this.file = file;
this.fileFormatOption = fileFormatOption;
this.builder = new BlockUmlBuilder(config, charset, defines, getReader(charset),
SFile.fromFile(file.getAbsoluteFile().getParentFile()), FileWithSuffix.getFileName(file));
}
public void setCheckMetadata(boolean checkMetadata) { public void setCheckMetadata(boolean checkMetadata) {
this.checkMetadata = checkMetadata; this.checkMetadata = checkMetadata;
} }

View File

@ -40,18 +40,12 @@ import java.io.IOException;
import java.util.List; import java.util.List;
import net.sourceforge.plantuml.preproc.Defines; import net.sourceforge.plantuml.preproc.Defines;
import net.sourceforge.plantuml.preproc.FileWithSuffix;
import net.sourceforge.plantuml.security.SFile;
public class SourceFileReaderCopyCat extends SourceFileReaderAbstract implements ISourceFileReader { public class SourceFileReaderCopyCat extends SourceFileReaderAbstract implements ISourceFileReader {
public SourceFileReaderCopyCat(Defines defines, final File file, File outputDirectory, List<String> config, public SourceFileReaderCopyCat(Defines defines, final File file, File outputDirectory, List<String> config,
String charset, FileFormatOption fileFormatOption) throws IOException { String charset, FileFormatOption fileFormatOption) throws IOException {
this.file = file; super(file, fileFormatOption, defines, config, charset);
this.fileFormatOption = fileFormatOption;
if (file.exists() == false) {
throw new IllegalArgumentException();
}
final String path = file.getParentFile().getPath(); final String path = file.getParentFile().getPath();
// System.err.println("SourceFileReaderCopyCat::path=" + path); // System.err.println("SourceFileReaderCopyCat::path=" + path);
// System.err.println("SourceFileReaderCopyCat::outputDirectory=" + // System.err.println("SourceFileReaderCopyCat::outputDirectory=" +
@ -63,8 +57,6 @@ public class SourceFileReaderCopyCat extends SourceFileReaderAbstract implements
// System.err.println("SourceFileReaderCopyCat=" + // System.err.println("SourceFileReaderCopyCat=" +
// this.outputDirectory.getPath() + " " // this.outputDirectory.getPath() + " "
// + this.outputDirectory.getAbsolutePath()); // + this.outputDirectory.getAbsolutePath());
builder = new BlockUmlBuilder(config, charset, defines, getReader(charset),
SFile.fromFile(file.getAbsoluteFile().getParentFile()), FileWithSuffix.getFileName(file));
} }
@Override @Override

View File

@ -5,12 +5,12 @@
* (C) Copyright 2009-2020, Arnaud Roques * (C) Copyright 2009-2020, Arnaud Roques
* *
* Project Info: http://plantuml.com * Project Info: http://plantuml.com
* *
* If you like this project or if you find it useful, you can support us at: * If you like this project or if you find it useful, you can support us at:
* *
* http://plantuml.com/patreon (only 1$ per month!) * http://plantuml.com/patreon (only 1$ per month!)
* http://plantuml.com/paypal * http://plantuml.com/paypal
* *
* This file is part of PlantUML. * This file is part of PlantUML.
* *
* PlantUML is free software; you can redistribute it and/or modify it * PlantUML is free software; you can redistribute it and/or modify it
@ -40,24 +40,15 @@ import java.io.IOException;
import java.util.List; import java.util.List;
import net.sourceforge.plantuml.preproc.Defines; import net.sourceforge.plantuml.preproc.Defines;
import net.sourceforge.plantuml.preproc.FileWithSuffix;
import net.sourceforge.plantuml.security.SFile; import net.sourceforge.plantuml.security.SFile;
public class SourceFileReaderHardFile extends SourceFileReaderAbstract implements ISourceFileReader { public class SourceFileReaderHardFile extends SourceFileReaderAbstract implements ISourceFileReader {
public SourceFileReaderHardFile(Defines defines, final File file, File outputFile, List<String> config, public SourceFileReaderHardFile(Defines defines, final File file, File outputFile, List<String> config,
String charset, FileFormatOption fileFormatOption) throws IOException { String charset, FileFormatOption fileFormatOption) throws IOException {
this.file = file; super(file, fileFormatOption, defines, config, charset);
this.fileFormatOption = fileFormatOption;
this.outputFile = outputFile; this.outputFile = outputFile;
if (file.exists() == false) {
throw new IllegalArgumentException();
}
FileSystem.getInstance().setCurrentDir(SFile.fromFile(file.getAbsoluteFile().getParentFile())); FileSystem.getInstance().setCurrentDir(SFile.fromFile(file.getAbsoluteFile().getParentFile()));
final SFile parentFile = SFile.fromFile(file.getAbsoluteFile().getParentFile());
builder = new BlockUmlBuilder(config, charset, defines, getReader(charset), parentFile,
FileWithSuffix.getFileName(file));
} }
@Override @Override