/* -*- mode: java; c-basic-offset: 2; indent-tabs-mode: nil -*- */

/*

Part of the Processing project - http://processing.org

Copyright (c) 2004-10 Ben Fry and Casey Reas

Copyright (c) 2001-04 Massachusetts Institute of Technology

This program is free software; you can redistribute it and/or modify

it under the terms of the GNU General Public License as published by

the Free Software Foundation; either version 2 of the License, or

(at your option) any later version.

This program is distributed in the hope that it will be useful,

but WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

GNU General Public License for more details.

You should have received a copy of the GNU General Public License

along with this program; if not, write to the Free Software Foundation,

Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

*/

package processing.app;

import processing.app.debug.AvrdudeUploader;

import processing.app.debug.Compiler;

import processing.app.debug.RunnerException;

import processing.app.debug.Sizer;

import processing.app.debug.Uploader;

import processing.app.preproc.*;

import processing.core.*;

import static processing.app.I18n._;

import java.awt.*;

import java.awt.event.*;

import java.beans.*;

import java.io.*;

import java.util.*;

import java.util.List;

import java.util.zip.*;

import javax.swing.*;

import javax.swing.border.EmptyBorder;

import javax.swing.border.TitledBorder;

/**

* Stores information about files in the current sketch

*/

public class Sketch {

static private File tempBuildFolder;

private Editor editor;

/** main pde file for this sketch. */

private File primaryFile;

/**

* Name of sketch, which is the name of main file

* (without .pde or .java extension)

*/

private String name;

/** true if any of the files have been modified. */

private boolean modified;

/** folder that contains this sketch */

private File folder;

/** data folder location for this sketch (may not exist yet) */

private File dataFolder;

/** code folder location for this sketch (may not exist yet) */

private File codeFolder;

private SketchCode current;

private int currentIndex;

/**

* Number of sketchCode objects (tabs) in the current sketch. Note that this

* will be the same as code.length, because the getCode() method returns

* just the code[] array, rather than a copy of it, or an array that's been

* resized to just the relevant files themselves.

* http://dev.processing.org/bugs/show_bug.cgi?id=940

*/

private int codeCount;

private SketchCode[] code;

/** Class name for the PApplet, as determined by the preprocessor. */

private String appletClassName;

/** Class path determined during build. */

private String classPath;

/**

* This is *not* the "Processing" libraries path, this is the Java libraries

* path, as in java.library.path=BlahBlah, which identifies search paths for

* DLLs or JNILIBs.

*/

private String libraryPath;

/**

* List of library folders.

*/

private ArrayList<File> importedLibraries;

/**

* path is location of the main .pde file, because this is also

* simplest to use when opening the file from the finder/explorer.

*/

public Sketch(Editor editor, String path) throws IOException {

this.editor = editor;

primaryFile = new File(path);

// get the name of the sketch by chopping .pde or .java

// off of the main file name

String mainFilename = primaryFile.getName();

int suffixLength = getDefaultExtension().length() + 1;

name = mainFilename.substring(0, mainFilename.length() - suffixLength);

// lib/build must exist when the application is started

// it is added to the CLASSPATH by default, but if it doesn't

// exist when the application is started, then java will remove

// the entry from the CLASSPATH, causing Runner to fail.

//

/*

tempBuildFolder = new File(TEMP_BUILD_PATH);

if (!tempBuildFolder.exists()) {

tempBuildFolder.mkdirs();

Base.showError("Required folder missing",

"A required folder was missing from \n" +

"from your installation of Processing.\n" +

"It has now been replaced, please restart \n" +

"the application to complete the repair.", null);

}

*/

tempBuildFolder = Base.getBuildFolder();

//Base.addBuildFolderToClassPath();

folder = new File(new File(path).getParent());

//System.out.println("sketch dir is " + folder);

load();

}

/**

* Build the list of files.

* <P>

* Generally this is only done once, rather than

* each time a change is made, because otherwise it gets to be

* a nightmare to keep track of what files went where, because

* not all the data will be saved to disk.

* <P>

* This also gets called when the main sketch file is renamed,

* because the sketch has to be reloaded from a different folder.

* <P>

* Another exception is when an external editor is in use,

* in which case the load happens each time "run" is hit.

*/

protected void load() {

codeFolder = new File(folder, "code");

dataFolder = new File(folder, "data");

// get list of files in the sketch folder

String list[] = folder.list();

// reset these because load() may be called after an

// external editor event. (fix for 0099)

codeCount = 0;

code = new SketchCode[list.length];

String[] extensions = getExtensions();

for (String filename : list) {

// Ignoring the dot prefix files is especially important to avoid files

// with the ._ prefix on Mac OS X. (You'll see this with Mac files on

// non-HFS drives, i.e. a thumb drive formatted FAT32.)

if (filename.startsWith(".")) continue;

// Don't let some wacko name a directory blah.pde or bling.java.

if (new File(folder, filename).isDirectory()) continue;

// figure out the name without any extension

String base = filename;

// now strip off the .pde and .java extensions

for (String extension : extensions) {

if (base.toLowerCase().endsWith("." + extension)) {

base = base.substring(0, base.length() - (extension.length() + 1));

// Don't allow people to use files with invalid names, since on load,

// it would be otherwise possible to sneak in nasty filenames. [0116]

if (Sketch.isSanitaryName(base)) {

code[codeCount++] =

new SketchCode(new File(folder, filename), extension);

}

}

}

}

// Remove any code that wasn't proper

code = (SketchCode[]) PApplet.subset(code, 0, codeCount);

// move the main class to the first tab

// start at 1, if it's at zero, don't bother

for (int i = 1; i < codeCount; i++) {

//if (code[i].file.getName().equals(mainFilename)) {

if (code[i].getFile().equals(primaryFile)) {

SketchCode temp = code[0];

code[0] = code[i];

code[i] = temp;

break;

}

}

// sort the entries at the top

sortCode();

// set the main file to be the current tab

if (editor != null) {

setCurrentCode(0);

}

}

protected void replaceCode(SketchCode newCode) {

for (int i = 0; i < codeCount; i++) {

if (code[i].getFileName().equals(newCode.getFileName())) {

code[i] = newCode;

break;

}

}

}

protected void insertCode(SketchCode newCode) {

// make sure the user didn't hide the sketch folder

ensureExistence();

// add file to the code/codeCount list, resort the list

//if (codeCount == code.length) {

code = (SketchCode[]) PApplet.append(code, newCode);

codeCount++;

//}

//code[codeCount++] = newCode;

}

protected void sortCode() {

// cheap-ass sort of the rest of the files

// it's a dumb, slow sort, but there shouldn't be more than ~5 files

for (int i = 1; i < codeCount; i++) {

int who = i;

for (int j = i + 1; j < codeCount; j++) {

if (code[j].getFileName().compareTo(code[who].getFileName()) < 0) {

who = j; // this guy is earlier in the alphabet

}

}

if (who != i) { // swap with someone if changes made

SketchCode temp = code[who];

code[who] = code[i];

code[i] = temp;

}

}

}

boolean renamingCode;

/**

* Handler for the New Code menu option.

*/

public void handleNewCode() {

// make sure the user didn't hide the sketch folder

ensureExistence();

// if read-only, give an error

if (isReadOnly()) {

// if the files are read-only, need to first do a "save as".

Base.showMessage(_("Sketch is Read-Only"),

_("Some files are marked \"read-only\", so you'll\n" +

"need to re-save the sketch in another location,\n" +

"and try again."));

return;

}

renamingCode = false;

editor.status.edit(_("Name for new file:"), "");

}

/**

* Handler for the Rename Code menu option.

*/

public void handleRenameCode() {

// make sure the user didn't hide the sketch folder

ensureExistence();

if (currentIndex == 0 && editor.untitled) {

Base.showMessage(_("Sketch is Untitled"),

_("How about saving the sketch first \n" +

"before trying to rename it?"));

return;

}

// if read-only, give an error

if (isReadOnly()) {

// if the files are read-only, need to first do a "save as".

Base.showMessage(_("Sketch is Read-Only"),

_("Some files are marked \"read-only\", so you'll\n" +

"need to re-save the sketch in another location,\n" +

"and try again."));

return;

}

// ask for new name of file (internal to window)

// TODO maybe just popup a text area?

renamingCode = true;

String prompt = (currentIndex == 0) ?

"New name for sketch:" : "New name for file:";

String oldName = (current.isExtension("ino")) ?

current.getPrettyName() : current.getFileName();

editor.status.edit(prompt, oldName);

}

/**

* 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();

// Add the extension here, this simplifies some of the logic below.

if (newName.indexOf('.') == -1) {

newName += "." + getDefaultExtension();

}

// if renaming to the same thing as before, just ignore.

// also ignoring case here, because i don't want to write

// a bunch of special stuff for each platform

// (osx is case insensitive but preserving, windows insensitive,

// *nix is sensitive and preserving.. argh)

if (renamingCode) {

if (newName.equalsIgnoreCase(current.getFileName())) {

// exit quietly for the 'rename' case.

// if it's a 'new' then an error will occur down below

return;

}

}

newName = newName.trim();

if (newName.equals("")) return;

int dot = newName.indexOf('.');

if (dot == 0) {

Base.showWarning(_("Problem with rename"),

_("The name cannot start with a period."), null);

return;

}

String newExtension = newName.substring(dot+1).toLowerCase();

if (!validExtension(newExtension)) {

Base.showWarning(_("Problem with rename"),

I18n.format(

_("\".{0}\" is not a valid extension."), newExtension

), null);

return;

}

// Don't let the user create the main tab as a .java file instead of .pde

if (!isDefaultExtension(newExtension)) {

if (renamingCode) { // If creating a new tab, don't show this error

if (current == code[0]) { // If this is the main tab, disallow

Base.showWarning(_("Problem with rename"),

_("The main file can't use an extension.\n" +

"(It may be time for your to graduate to a\n" +

"\"real\" programming environment)"), null);

return;

}

}

}

// dots are allowed for the .pde and .java, but not in the name

// make sure the user didn't name things poo.time.pde

// or something like that (nothing against poo time)

String shortName = newName.substring(0, dot);

String sanitaryName = Sketch.sanitizeName(shortName);

if (!shortName.equals(sanitaryName)) {

newName = sanitaryName + "." + newExtension;

}

// In Arduino, we want to allow files with the same name but different

// extensions, so compare the full names (including extensions). This

// might cause problems: http://dev.processing.org/bugs/show_bug.cgi?id=543

for (SketchCode c : code) {

if (newName.equalsIgnoreCase(c.getFileName())) {

Base.showMessage(_("Nope"),

I18n.format(

_("A file named \"{0}\" already exists in \"{1}\""),

c.getFileName(),

folder.getAbsolutePath()

));

return;

}

}

// In Arduino, don't allow a .cpp file with the same name as the sketch,

// because the sketch is concatenated into a file with that name as part

// of the build process.

if (newName.equals(getName() + ".cpp")) {

Base.showMessage(_("Nope"),

_("You can't have a .cpp file with the same name as the sketch."));

return;

}

if (renamingCode && currentIndex == 0) {

for (int i = 1; i < codeCount; i++) {

if (sanitaryName.equalsIgnoreCase(code[i].getPrettyName()) &&

code[i].getExtension().equalsIgnoreCase("cpp")) {

Base.showMessage(_("Nope"),

I18n.format(

_("You can't rename the sketch to \"{0}\"\n" +

"because the sketch already has a .cpp file with that name."),

sanitaryName

));

return;

}

}

}

File newFile = new File(folder, newName);

// if (newFile.exists()) { // yay! users will try anything

// Base.showMessage("Nope",

// "A file named \"" + newFile + "\" already exists\n" +

// "in \"" + folder.getAbsolutePath() + "\"");

// return;

// }

// File newFileHidden = new File(folder, newName + ".x");

// if (newFileHidden.exists()) {

// // don't let them get away with it if they try to create something

// // with the same name as something hidden

// Base.showMessage("No Way",

// "A hidden tab with the same name already exists.\n" +

// "Use \"Unhide\" to bring it back.");

// return;

// }

if (renamingCode) {

if (currentIndex == 0) {

// get the new folder name/location

String folderName = newName.substring(0, newName.indexOf('.'));

File newFolder = new File(folder.getParentFile(), folderName);

if (newFolder.exists()) {

Base.showWarning(_("Cannot Rename"),

I18n.format(

_("Sorry, a sketch (or folder) named " +

"\"{0}\" already exists."),

newName

), null);

return;

}

// unfortunately this can't be a "save as" because that

// only copies the sketch files and the data folder

// however this *will* first save the sketch, then rename

// first get the contents of the editor text area

if (current.isModified()) {

current.setProgram(editor.getText());

try {

// save this new SketchCode

current.save();

} catch (Exception e) {

Base.showWarning(_("Error"), _("Could not rename the sketch. (0)"), e);

return;

}

}

if (!current.renameTo(newFile, newExtension)) {

Base.showWarning(_("Error"),

I18n.format(

_("Could not rename \"{0}\" to \"{1}\""),

current.getFileName(),

newFile.getName()

), null);

return;

}

// save each of the other tabs because this is gonna be re-opened

try {

for (int i = 1; i < codeCount; i++) {

code[i].save();

}

} catch (Exception e) {

Base.showWarning(_("Error"), _("Could not rename the sketch. (1)"), e);

return;

}

// now rename the sketch folder and re-open

boolean success = folder.renameTo(newFolder);

if (!success) {

Base.showWarning(_("Error"), _("Could not rename the sketch. (2)"), null);

return;

}

// if successful, set base properties for the sketch

File newMainFile = new File(newFolder, newName + ".ino");

String newMainFilePath = newMainFile.getAbsolutePath();

// having saved everything and renamed the folder and the main .pde,

// use the editor to re-open the sketch to re-init state

// (unfortunately this will kill positions for carets etc)

editor.handleOpenUnchecked(newMainFilePath,

currentIndex,

editor.getSelectionStart(),

editor.getSelectionStop(),

editor.getScrollPosition());

// get the changes into the sketchbook menu

// (re-enabled in 0115 to fix bug #332)

editor.base.rebuildSketchbookMenus();

} else { // else if something besides code[0]

if (!current.renameTo(newFile, newExtension)) {

Base.showWarning(_("Error"),

I18n.format(

_("Could not rename \"{0}\" to \"{1}\""),

current.getFileName(),

newFile.getName()

), null);

return;

}

}

} else { // creating a new file

try {

if (!newFile.createNewFile()) {

// Already checking for IOException, so make our own.

throw new IOException(_("createNewFile() returned false"));

}

} catch (IOException e) {

Base.showWarning(_("Error"),

I18n.format(

"Could not create the file \"{0}\" in \"{1}\"",

newFile,

folder.getAbsolutePath()

), e);

return;

}

SketchCode newCode = new SketchCode(newFile, newExtension);

//System.out.println("new code is named " + newCode.getPrettyName() + " " + newCode.getFile());

insertCode(newCode);

}

// sort the entries

sortCode();

// set the new guy as current

setCurrentCode(newName);

// update the tabs

editor.header.rebuild();

}

/**

* Remove a piece of code from the sketch and from the disk.

*/

public void handleDeleteCode() {

// make sure the user didn't hide the sketch folder

ensureExistence();

// if read-only, give an error

if (isReadOnly()) {

// if the files are read-only, need to first do a "save as".

Base.showMessage(_("Sketch is Read-Only"),

_("Some files are marked \"read-only\", so you'll\n" +

"need to re-save the sketch in another location,\n" +

"and try again."));

return;

}

// confirm deletion with user, yes/no

Object[] options = { _("OK"), _("Cancel") };

String prompt = (currentIndex == 0) ?

_("Are you sure you want to delete this sketch?") :

I18n.format(_("Are you sure you want to delete \"{0}\"?"), current.getPrettyName());

int result = JOptionPane.showOptionDialog(editor,

prompt,

_("Delete"),

JOptionPane.YES_NO_OPTION,

JOptionPane.QUESTION_MESSAGE,

null,

options,

options[0]);

if (result == JOptionPane.YES_OPTION) {

if (currentIndex == 0) {

// need to unset all the modified flags, otherwise tries

// to do a save on the handleNew()

// delete the entire sketch

Base.removeDir(folder);

// get the changes into the sketchbook menu

//sketchbook.rebuildMenus();

// make a new sketch, and i think this will rebuild the sketch menu

//editor.handleNewUnchecked();

//editor.handleClose2();

editor.base.handleClose(editor);

} else {

// delete the file

if (!current.deleteFile()) {

Base.showMessage(_("Couldn't do it"),

I18n.format(_("Could not delete \"{0}\"."), current.getFileName()));

return;

}

// remove code from the list

removeCode(current);

// just set current tab to the main tab

setCurrentCode(0);

// update the tabs

editor.header.repaint();

}

}

}

protected void removeCode(SketchCode which) {

// remove it from the internal list of files

// resort internal list of files

for (int i = 0; i < codeCount; i++) {

if (code[i] == which) {

for (int j = i; j < codeCount-1; j++) {

code[j] = code[j+1];

}

codeCount--;

code = (SketchCode[]) PApplet.shorten(code);

return;

}

}

System.err.println(_("removeCode: internal error.. could not find code"));

}

/**

* Move to the previous tab.

*/

public void handlePrevCode() {

int prev = currentIndex - 1;

if (prev < 0) prev = codeCount-1;

setCurrentCode(prev);

}

/**

* Move to the next tab.

*/

public void handleNextCode() {

setCurrentCode((currentIndex + 1) % codeCount);

}

/**

* Sets the modified value for the code in the frontmost tab.

*/

public void setModified(boolean state) {

//System.out.println("setting modified to " + state);

//new Exception().printStackTrace();

current.setModified(state);

calcModified();

}

protected void calcModified() {

modified = false;

for (int i = 0; i < codeCount; i++) {

if (code[i].isModified()) {

modified = true;

break;

}

}

editor.header.repaint();

if (Base.isMacOS()) {

// http://developer.apple.com/qa/qa2001/qa1146.html

Object modifiedParam = modified ? Boolean.TRUE : Boolean.FALSE;

editor.getRootPane().putClientProperty("windowModified", modifiedParam);

}

}

public boolean isModified() {

return modified;

}

/**

* Save all code in the current sketch.

*/

public boolean save() throws IOException {

// make sure the user didn't hide the sketch folder

ensureExistence();

// first get the contents of the editor text area

if (current.isModified()) {

current.setProgram(editor.getText());

}

// don't do anything if not actually modified

//if (!modified) return false;

if (isReadOnly()) {

// if the files are read-only, need to first do a "save as".

Base.showMessage(_("Sketch is read-only"),

_("Some files are marked \"read-only\", so you'll\n" +

"need to re-save this sketch to another location."));

// if the user cancels, give up on the save()

if (!saveAs()) return false;

} else {

// rename .pde files to .ino

File mainFile = new File(getMainFilePath());

File mainFolder = mainFile.getParentFile();

File[] pdeFiles = mainFolder.listFiles(new FilenameFilter() {

public boolean accept(File dir, String name) {

return name.toLowerCase().endsWith(".pde");

}

});

if (pdeFiles != null && pdeFiles.length > 0) {

if (Preferences.get("editor.update_extension") == null) {

Object[] options = { _("OK"), _("Cancel") };

int result = JOptionPane.showOptionDialog(editor,

_("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?"),

_(".pde -> .ino"),

JOptionPane.OK_CANCEL_OPTION,

JOptionPane.QUESTION_MESSAGE,

null,

options,

options[0]);

if (result != JOptionPane.OK_OPTION) return false; // save cancelled

Preferences.setBoolean("editor.update_extension", true);

}

if (Preferences.getBoolean("editor.update_extension")) {

// Do rename of all .pde files to new .ino extension

for (File pdeFile : pdeFiles)

renameCodeToInoExtension(pdeFile);

}

}

}

for (int i = 0; i < codeCount; i++) {

if (code[i].isModified())

code[i].save();

}

calcModified();

return true;

}

protected boolean renameCodeToInoExtension(File pdeFile) {

for (SketchCode c : code) {

if (!c.getFile().equals(pdeFile))

continue;

String pdeName = pdeFile.getPath();

pdeName = pdeName.substring(0, pdeName.length() - 4) + ".ino";

return c.renameTo(new File(pdeName), "ino");

}

return false;

}

/**

* 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 {

String newParentDir = null;

String newName = null;

/*

JFileChooser fc = new JFileChooser();

fc.setDialogTitle("Save sketch folder as...");

if (isReadOnly() || isUntitled()) {

// default to the sketchbook folder

fc.setCurrentDirectory(new File(Preferences.get("sketchbook.path")));

} else {

// default to the parent folder of where this was

fc.setCurrentDirectory(folder.getParentFile());

}

// can't do this, will try to save into itself by default

//fc.setSelectedFile(folder);

int result = fc.showSaveDialog(editor);

if (result == JFileChooser.APPROVE_OPTION) {

File selection = fc.getSelectedFile();

newParentDir = selection.getParent();

newName = selection.getName();

}

*/

// get new name for folder

FileDialog fd = new FileDialog(editor,

_("Save sketch folder as..."),

FileDialog.SAVE);

if (isReadOnly() || isUntitled()) {

// default to the sketchbook folder

fd.setDirectory(Preferences.get("sketchbook.path"));

} else {

// default to the parent folder of where this was

fd.setDirectory(folder.getParent());

}

String oldName = folder.getName();

fd.setFile(oldName);

fd.setVisible(true);

newParentDir = fd.getDirectory();

newName = fd.getFile();

// user canceled selection

if (newName == null) return false;

newName = Sketch.checkName(newName);

File newFolder = new File(newParentDir, newName);

// String newPath = newFolder.getAbsolutePath();

// String oldPath = folder.getAbsolutePath();

// if (newPath.equals(oldPath)) {

// return false; // Can't save a sketch over itself

// }

// make sure there doesn't exist a .cpp file with that name already

// but ignore this situation for the first tab, since it's probably being

// resaved (with the same name) to another location/folder.

for (int i = 1; i < codeCount; i++) {

if (newName.equalsIgnoreCase(code[i].getPrettyName()) &&

code[i].getExtension().equalsIgnoreCase("cpp")) {

Base.showMessage(_("Nope"),

I18n.format(

_("You can't save the sketch as \"{0}\"\n" +

"because the sketch already has a .cpp file with that name."),

newName

));

return false;

}

}

// check if the paths are identical

if (newFolder.equals(folder)) {

// 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 = folder.getCanonicalPath() + File.separator;

if (newPath.indexOf(oldPath) == 0) {

Base.showWarning(_("How very Borges of you"),

_("You cannot save the sketch into a folder\n" +

"inside itself. This would go on forever."), null);

return false;

}

} catch (IOException e) { }

// 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()) {

Base.removeDir(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.

// now make a fresh copy of the folder

newFolder.mkdirs();

// grab the contents of the current tab before saving

// first get the contents of the editor text area

if (current.isModified()) {

current.setProgram(editor.getText());

}

// save the other tabs to their new location

for (int i = 1; i < codeCount; i++) {

File newFile = new File(newFolder, code[i].getFileName());

code[i].saveAs(newFile);

}

// re-copy the data folder (this may take a while.. add progress bar?)

if (dataFolder.exists()) {

File newDataFolder = new File(newFolder, "data");

Base.copyDir(dataFolder, newDataFolder);

}

// re-copy the code folder

if (codeFolder.exists()) {

File newCodeFolder = new File(newFolder, "code");

Base.copyDir(codeFolder, newCodeFolder);

}

// copy custom applet.html file if one exists

// http://dev.processing.org/bugs/show_bug.cgi?id=485

File customHtml = new File(folder, "applet.html");

if (customHtml.exists()) {

File newHtml = new File(newFolder, "applet.html");

Base.copyFile(customHtml, newHtml);

}

// save the main tab with its new name

File newFile = new File(newFolder, newName + ".ino");

code[0].saveAs(newFile);

editor.handleOpenUnchecked(newFile.getPath(),

currentIndex,

editor.getSelectionStart(),

editor.getSelectionStop(),

editor.getScrollPosition());

// Name changed, rebuild the sketch menus

//editor.sketchbook.rebuildMenusAsync();

editor.base.rebuildSketchbookMenus();

// Make sure that it's not an untitled sketch

setUntitled(false);

// let Editor know that the save was successful

return true;

}

/**

* Prompt the user for a new file to the sketch, then call the

* other addFile() function to actually add it.

*/

public void handleAddFile() {

// make sure the user didn't hide the sketch folder

ensureExistence();

// if read-only, give an error

if (isReadOnly()) {

// if the files are read-only, need to first do a "save as".

Base.showMessage(_("Sketch is Read-Only"),

_("Some files are marked \"read-only\", so you'll\n" +

"need to re-save the sketch in another location,\n" +

"and try again."));

return;

}

// get a dialog, select a file to add to the sketch

String prompt =

_("Select an image or other data file to copy to your sketch");

//FileDialog fd = new FileDialog(new Frame(), prompt, FileDialog.LOAD);

FileDialog fd = new FileDialog(editor, prompt, FileDialog.LOAD);

fd.setVisible(true);

String directory = fd.getDirectory();

String filename = fd.getFile();

if (filename == null) return;

// copy the file into the folder. if people would rather

// it move instead of copy, they can do it by hand

File sourceFile = new File(directory, filename);

// now do the work of adding the file

boolean result = addFile(sourceFile);

if (result) {

editor.statusNotice(_("One file added to the sketch."));

}

}