private File saveSketchInTempFolder() throws IOException {
File tempFolder = FileUtils.createTempFolder("arduino_modified_sketch_");
FileUtils.copy(sketch.getFolder(), tempFolder);
for (SketchFile file :
Stream.of(sketch.getFiles()).filter(SketchFile::isModified).collect(Collectors.toList())) {
Files.write(
Paths.get(tempFolder.getAbsolutePath(), file.getFileName()),
file.getProgram().getBytes());
}
return Paths.get(tempFolder.getAbsolutePath(), sketch.getPrimaryFile().getFileName()).toFile();
}
/**
* Preprocess and compile all the code for this sketch.
*
* <p>In an advanced program, the returned class name could be different, which is why the
* className is set based on the return value. A compilation error will burp up a RunnerException.
*
* @return null if compilation failed, main class name if not
*/
public String build(boolean verbose, boolean save)
throws RunnerException, PreferencesMapException, IOException {
// run the preprocessor
editor.status.progressUpdate(20);
ensureExistence();
CompilerProgressListener progressListener = editor.status::progressUpdate;
boolean deleteTemp = false;
File pathToSketch = sketch.getPrimaryFile().getFile();
if (sketch.isModified()) {
// If any files are modified, make a copy of the sketch with the changes
// saved, so arduino-builder will see the modifications.
pathToSketch = saveSketchInTempFolder();
deleteTemp = true;
}
try {
return new Compiler(pathToSketch, sketch).build(progressListener, save);
} finally {
// Make sure we clean up any temporary sketch copy
if (deleteTemp) FileUtils.recursiveDelete(pathToSketch.getParentFile());
}
}
/** Save all code in the current sketch. */
public boolean save() throws IOException {
// make sure the user didn't hide the sketch folder
ensureExistence();
if (isReadOnly(
BaseNoGui.librariesIndexer.getInstalledLibraries(), BaseNoGui.getExamplesPath())) {
Base.showMessage(
tr("Sketch is read-only"),
tr(
"Some files are marked \"read-only\", so you'll\n"
+ "need to re-save this sketch to another location."));
return saveAs();
}
// rename .pde files to .ino
List<SketchFile> oldFiles = new ArrayList<>();
for (SketchFile file : sketch.getFiles()) {
if (file.isExtension(Sketch.OLD_SKETCH_EXTENSIONS)) oldFiles.add(file);
}
if (oldFiles.size() > 0) {
if (PreferencesData.get("editor.update_extension") == null) {
Object[] options = {tr("OK"), tr("Cancel")};
int result =
JOptionPane.showOptionDialog(
editor,
tr(
"In Arduino 1.0, the default file extension has changed\n"
+ "from .pde to .ino. New sketches (including those created\n"
+ "by \"Save-As\") will use the new extension. The extension\n"
+ "of existing sketches will be updated on save, but you can\n"
+ "disable this in the Preferences dialog.\n"
+ "\n"
+ "Save sketch and update its extension?"),
tr(".pde -> .ino"),
JOptionPane.OK_CANCEL_OPTION,
JOptionPane.QUESTION_MESSAGE,
null,
options,
options[0]);
if (result != JOptionPane.OK_OPTION) return false; // save cancelled
PreferencesData.setBoolean("editor.update_extension", true);
}
if (PreferencesData.getBoolean("editor.update_extension")) {
// Do rename of all .pde files to new .ino extension
for (SketchFile file : oldFiles) {
File newName =
FileUtils.replaceExtension(file.getFile(), Sketch.DEFAULT_SKETCH_EXTENSION);
file.renameTo(newName.getName());
}
}
}
sketch.save();
return true;
}
public static File getBuildFolder(Sketch data) throws IOException {
File buildFolder;
if (PreferencesData.get("build.path") != null) {
buildFolder = BaseNoGui.absoluteFile(PreferencesData.get("build.path"));
Files.createDirectories(buildFolder.toPath());
} else {
buildFolder =
FileUtils.createTempFolder("build", DigestUtils.md5Hex(data.getMainFilePath()) + ".tmp");
DeleteFilesOnShutdown.add(buildFolder);
}
return buildFolder;
}
/**
* Add a file to the sketch.
*
* <p>Supported code files will be copied to the sketch folder. All other files will be copied to
* the "data" folder (which is created if it does not exist yet).
*
* @return true if successful.
*/
public boolean addFile(File sourceFile) {
String filename = sourceFile.getName();
File destFile = null;
boolean isData = false;
boolean replacement = false;
if (FileUtils.hasExtension(sourceFile, Sketch.EXTENSIONS)) {
destFile = new File(sketch.getFolder(), filename);
} else {
sketch.prepareDataFolder();
destFile = new File(sketch.getDataFolder(), filename);
isData = true;
}
// check whether this file already exists
if (destFile.exists()) {
Object[] options = {tr("OK"), tr("Cancel")};
String prompt = I18n.format(tr("Replace the existing version of {0}?"), filename);
int result =
JOptionPane.showOptionDialog(
editor,
prompt,
tr("Replace"),
JOptionPane.YES_NO_OPTION,
JOptionPane.QUESTION_MESSAGE,
null,
options,
options[0]);
if (result == JOptionPane.YES_OPTION) {
replacement = true;
} else {
return false;
}
}
// If it's a replacement, delete the old file first,
// otherwise case changes will not be preserved.
// http://dev.processing.org/bugs/show_bug.cgi?id=969
if (replacement) {
boolean muchSuccess = destFile.delete();
if (!muchSuccess) {
Base.showWarning(
tr("Error adding file"),
I18n.format(tr("Could not delete the existing ''{0}'' file."), filename),
null);
return false;
}
}
// make sure they aren't the same file
if (isData && sourceFile.equals(destFile)) {
Base.showWarning(
tr("You can't fool me"),
tr(
"This file has already been copied to the\n"
+ "location from which where you're trying to add it.\n"
+ "I ain't not doin nuthin'."),
null);
return false;
}
// in case the user is "adding" the code in an attempt
// to update the sketch's tabs
if (!sourceFile.equals(destFile)) {
try {
Base.copyFile(sourceFile, destFile);
} catch (IOException e) {
Base.showWarning(
tr("Error adding file"),
I18n.format(tr("Could not add ''{0}'' to the sketch."), filename),
e);
return false;
}
}
if (!isData) {
int tabIndex;
if (replacement) {
tabIndex = editor.findTabIndex(destFile);
editor.getTabs().get(tabIndex).reload();
} else {
SketchFile sketchFile;
try {
sketchFile = sketch.addFile(destFile.getName());
editor.addTab(sketchFile, null);
} catch (IOException e) {
// This does not pass on e, to prevent showing a backtrace for
// "normal" errors.
Base.showWarning(tr("Error"), e.getMessage(), null);
return false;
}
tabIndex = editor.findTabIndex(sketchFile);
}
editor.selectTab(tabIndex);
}
return true;
}
/**
* Handles 'Save As' for a sketch.
*
* <p>This basically just duplicates the current sketch folder to a new location, and then calls
* 'Save'. (needs to take the current state of the open files and save them to the new folder..
* but not save over the old versions for the old sketch..)
*
* <p>Also removes the previously-generated .class and .jar files, because they can cause trouble.
*/
protected boolean saveAs() throws IOException {
// get new name for folder
FileDialog fd = new FileDialog(editor, tr("Save sketch folder as..."), FileDialog.SAVE);
if (isReadOnly(BaseNoGui.librariesIndexer.getInstalledLibraries(), BaseNoGui.getExamplesPath())
|| isUntitled()) {
// default to the sketchbook folder
fd.setDirectory(BaseNoGui.getSketchbookFolder().getAbsolutePath());
} else {
// default to the parent folder of where this was
// on macs a .getParentFile() method is required
fd.setDirectory(sketch.getFolder().getParentFile().getAbsolutePath());
}
String oldName = sketch.getName();
fd.setFile(oldName);
fd.setVisible(true);
String newParentDir = fd.getDirectory();
String newName = fd.getFile();
// user canceled selection
if (newName == null) return false;
newName = SketchController.checkName(newName);
File newFolder = new File(newParentDir, newName);
// check if the paths are identical
if (newFolder.equals(sketch.getFolder())) {
// just use "save" here instead, because the user will have received a
// message (from the operating system) about "do you want to replace?"
return save();
}
// check to see if the user is trying to save this sketch inside itself
try {
String newPath = newFolder.getCanonicalPath() + File.separator;
String oldPath = sketch.getFolder().getCanonicalPath() + File.separator;
if (newPath.indexOf(oldPath) == 0) {
Base.showWarning(
tr("How very Borges of you"),
tr(
"You cannot save the sketch into a folder\n"
+ "inside itself. This would go on forever."),
null);
return false;
}
} catch (IOException e) {
// ignore
}
// if the new folder already exists, then need to remove
// its contents before copying everything over
// (user will have already been warned)
if (newFolder.exists()) {
FileUtils.recursiveDelete(newFolder);
}
// in fact, you can't do this on windows because the file dialog
// will instead put you inside the folder, but it happens on osx a lot.
try {
sketch.saveAs(newFolder);
} catch (IOException e) {
// This does not pass on e, to prevent showing a backtrace for "normal"
// errors.
Base.showWarning(tr("Error"), e.getMessage(), null);
}
// Name changed, rebuild the sketch menus
// editor.sketchbook.rebuildMenusAsync();
editor.base.rebuildSketchbookMenus();
editor.header.rebuild();
// Make sure that it's not an untitled sketch
setUntitled(false);
// let Editor know that the save was successful
return true;
}
/**
* This is called upon return from entering a new file name. (that is, from either newCode or
* renameCode after the prompt) This code is almost identical for both the newCode and renameCode
* cases, so they're kept merged except for right in the middle where they diverge.
*/
protected void nameCode(String newName) {
// make sure the user didn't hide the sketch folder
ensureExistence();
newName = newName.trim();
if (newName.equals("")) return;
if (newName.charAt(0) == '.') {
Base.showWarning(tr("Problem with rename"), tr("The name cannot start with a period."), null);
return;
}
FileUtils.SplitFile split = FileUtils.splitFilename(newName);
if (split.extension.equals("")) split.extension = Sketch.DEFAULT_SKETCH_EXTENSION;
if (!Sketch.EXTENSIONS.contains(split.extension)) {
String msg = I18n.format(tr("\".{0}\" is not a valid extension."), split.extension);
Base.showWarning(tr("Problem with rename"), msg, null);
return;
}
// Sanitize name
split.basename = BaseNoGui.sanitizeName(split.basename);
newName = split.join();
if (renamingCode) {
SketchFile current = editor.getCurrentTab().getSketchFile();
if (current.isPrimary()) {
if (!split.extension.equals(Sketch.DEFAULT_SKETCH_EXTENSION)) {
Base.showWarning(
tr("Problem with rename"), tr("The main file cannot use an extension"), null);
return;
}
// Primary file, rename the entire sketch
final File parent = sketch.getFolder().getParentFile();
File newFolder = new File(parent, split.basename);
try {
sketch.renameTo(newFolder);
} catch (IOException e) {
// This does not pass on e, to prevent showing a backtrace for
// "normal" errors.
Base.showWarning(tr("Error"), e.getMessage(), null);
return;
}
editor.base.rebuildSketchbookMenus();
} else {
// Non-primary file, rename just that file
try {
current.renameTo(newName);
} catch (IOException e) {
// This does not pass on e, to prevent showing a backtrace for
// "normal" errors.
Base.showWarning(tr("Error"), e.getMessage(), null);
return;
}
}
} else { // creating a new file
SketchFile file;
try {
file = sketch.addFile(newName);
editor.addTab(file, "");
} catch (IOException e) {
// This does not pass on e, to prevent showing a backtrace for
// "normal" errors.
Base.showWarning(tr("Error"), e.getMessage(), null);
return;
}
editor.selectTab(editor.findTabIndex(file));
}
// update the tabs
editor.header.rebuild();
}
