/** Handler for the Rename Code menu option. */
public void handleRenameCode() {
SketchFile current = editor.getCurrentTab().getSketchFile();
editor.status.clearState();
// make sure the user didn't hide the sketch folder
ensureExistence();
if (current.isPrimary() && editor.untitled) {
Base.showMessage(
tr("Sketch is Untitled"),
tr("How about saving the sketch first \n" + "before trying to rename it?"));
return;
}
// if read-only, give an error
if (isReadOnly(
BaseNoGui.librariesIndexer.getInstalledLibraries(), BaseNoGui.getExamplesPath())) {
// if the files are read-only, need to first do a "save as".
Base.showMessage(
tr("Sketch is Read-Only"),
tr(
"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 = current.isPrimary() ? "New name for sketch:" : "New name for file:";
String oldName = current.getPrettyName();
editor.status.edit(prompt, oldName);
}
/** Remove a piece of code from the sketch and from the disk. */
public void handleDeleteCode() throws IOException {
SketchFile current = editor.getCurrentTab().getSketchFile();
editor.status.clearState();
// make sure the user didn't hide the sketch folder
ensureExistence();
// if read-only, give an error
if (isReadOnly(
BaseNoGui.librariesIndexer.getInstalledLibraries(), BaseNoGui.getExamplesPath())) {
// if the files are read-only, need to first do a "save as".
Base.showMessage(
tr("Sketch is Read-Only"),
tr(
"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 = {tr("OK"), tr("Cancel")};
String prompt =
current.isPrimary()
? tr("Are you sure you want to delete this sketch?")
: I18n.format(tr("Are you sure you want to delete \"{0}\"?"), current.getPrettyName());
int result =
JOptionPane.showOptionDialog(
editor,
prompt,
tr("Delete"),
JOptionPane.YES_NO_OPTION,
JOptionPane.QUESTION_MESSAGE,
null,
options,
options[0]);
if (result == JOptionPane.YES_OPTION) {
if (current.isPrimary()) {
sketch.delete();
editor.base.handleClose(editor);
} else {
// delete the file
if (!current.delete(sketch.getBuildPath().toPath())) {
Base.showMessage(
tr("Couldn't do it"),
I18n.format(tr("Could not delete \"{0}\"."), current.getFileName()));
return;
}
// just set current tab to the main tab
editor.selectTab(0);
// update the tabs
editor.header.repaint();
}
}
}
/** 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 void init(Editor _editor) {
this.m_editor = _editor;
File toolRoot = null;
try {
toolRoot =
new File(
SequantoAutomationTool.class
.getProtectionDomain()
.getCodeSource()
.getLocation()
.toURI())
.getParentFile()
.getParentFile();
} catch (java.net.URISyntaxException ex) {
toolRoot =
new File(
SequantoAutomationTool.class
.getProtectionDomain()
.getCodeSource()
.getLocation()
.getPath())
.getParentFile()
.getParentFile();
}
m_generatorPy =
new File(new File(toolRoot, "generator"), "generate_automation_defines.py")
.getAbsolutePath();
m_isWindows = System.getProperty("os.name").toLowerCase().contains("win");
if (m_isWindows) {
for (File root : File.listRoots()) {
File[] files =
root.listFiles(
new FileFilter() {
public boolean accept(File f) {
return f.getName().toLowerCase().startsWith("python");
}
});
if (files != null) {
for (File directory : files) {
m_pythonPath = new File(directory, "python.exe");
break;
}
}
}
if (m_pythonPath == null) {
Base.showMessage(
"ERROR",
String.format(
"Could not python interpreter - Generate Automation tool will not work."));
}
}
}
/**
* 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(
BaseNoGui.librariesIndexer.getInstalledLibraries(), BaseNoGui.getExamplesPath())) {
// if the files are read-only, need to first do a "save as".
Base.showMessage(
tr("Sketch is Read-Only"),
tr(
"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
FileDialog fd =
new FileDialog(
editor,
tr("Select an image or other data file to copy to your sketch"),
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(tr("One file added to the sketch."));
PreferencesData.set("last.folder", sourceFile.getAbsolutePath());
}
}
/** Handler for the New Code menu option. */
public void handleNewCode() {
editor.status.clearState();
// make sure the user didn't hide the sketch folder
ensureExistence();
// if read-only, give an error
if (isReadOnly(
BaseNoGui.librariesIndexer.getInstalledLibraries(), BaseNoGui.getExamplesPath())) {
// if the files are read-only, need to first do a "save as".
Base.showMessage(
tr("Sketch is Read-Only"),
tr(
"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(tr("Name for new file:"), "");
}
public void run() {
if (!new File(m_generatorPy).exists()) {
Base.showMessage(
"ERROR", String.format("Could not find generator python script at %s", m_generatorPy));
}
Sketch sketch = m_editor.getSketch();
if (sketch.isModified()) {
try {
sketch.save();
} catch (java.io.IOException ex) {
// Base.showMessage("ERROR", "Could not save sketch before trying to generate." );
}
}
// String sketchName = sketch.getName();
// SketchCode codeObject = sketch.getCurrentCode();
// String code = codeObject.getProgram();
try {
String code = m_editor.getCurrentTab().getText();
// String code = m_editor.getText();
int start = code.indexOf("BEGIN AUTOMATION");
if (start != -1) {
int end = code.indexOf("END AUTOMATION", start);
if (end != -1) {
String automationCode = code.substring(start + "BEGIN AUTOMATION".length(), end);
automationCode = automationCode.replaceAll("\\*\\s+", "");
// SketchData sketchData = new MySketchData(sketch.getMainFilePath());
// File buildFolder = BaseNoGui.getBuildFolder(sketchData);
File buildFolder = getBuildFolder(sketch);
// File codeFolder = sketchData.getCodeFolder();
File codeFolder = new File(sketch.getFolder(), "code");
Files.createDirectories(codeFolder.toPath());
File automationFileName = new File(buildFolder, "automation.automation");
FileWriter writer = new FileWriter(automationFileName);
writer.write("name automation\n");
writer.write("import " + sketch.getMainFilePath().toString() + "\n");
writer.write(automationCode);
writer.close();
try {
ProcessBuilder processBuilder = null;
if (m_isWindows) {
processBuilder =
new ProcessBuilder(
m_pythonPath.getAbsolutePath(),
m_generatorPy,
"-s",
"--arduino",
automationFileName.getAbsolutePath());
} else {
processBuilder =
new ProcessBuilder(
m_generatorPy, "-s", "--arduino", automationFileName.getAbsolutePath());
}
processBuilder.directory(codeFolder);
processBuilder.redirectErrorStream(true);
Process process = processBuilder.start();
inheritIO(process.getInputStream(), System.out);
int result = process.waitFor();
String includeLibLine = "#include \"SequantoAutomation.h\"\n";
if (!code.contains(includeLibLine)) {
code = includeLibLine + code;
}
/*
File generatedFileName = new File(codeFolder, "automation_automation.c" );
String includeLine = String.format("#include \"%s\"\n", generatedFileName.getAbsolutePath());
if ( !code.contains(includeLine) )
{
int i = code.indexOf(includeLibLine);
code = code.substring(0, i + includeLibLine.length()) +
includeLine +
code.substring ( i + includeLibLine.length(), code.length() );
}
*/
String includeLine = String.format("#include \"code/automation_automation.h\"\n");
if (!code.contains(includeLine)) {
int i = code.indexOf(includeLibLine);
code =
code.substring(0, i + includeLibLine.length())
+ includeLine
+ code.substring(i + includeLibLine.length(), code.length());
}
String includeCodeLine = String.format("\n#include \"code/automation_automation.c\"\n");
if (!code.contains(includeCodeLine)) {
code = code + includeCodeLine;
}
if (m_editor.getCurrentTab().getText() != code) {
// System.out.println ( "Setting code to" );
// System.out.println ( code );
// System.out.println ( "Current text was" );
// System.out.println ( m_editor.getText() );
m_editor.getCurrentTab().setText(code);
}
if (result == 0) {
System.out.println("Sequanto automation code generated successfully!");
} else {
System.out.println("ERROR!!!");
}
} catch (Exception ex) {
Base.showMessage("ERROR", String.format("Could not start generator script: %s", ex));
}
} else {
Base.showMessage("ERROR", "Can not find END AUTOMATION.");
}
} else {
Base.showMessage("ERROR", "Can not find BEGIN AUTOMATION.");
}
} catch (java.io.IOException ex) {
Base.showMessage("ERROR", String.format("IO Error: %s.", ex));
}
}
