/** Add import statements to the current tab for the specified library */
public void importLibrary(UserLibrary lib) throws IOException {
// make sure the user didn't hide the sketch folder
ensureExistence();
List<String> list = lib.getIncludes();
if (list == null) {
File srcFolder = lib.getSrcFolder();
String[] headers = Base.headerListFromIncludePath(srcFolder);
list = Arrays.asList(headers);
}
if (list.isEmpty()) {
return;
}
// import statements into the main sketch file (code[0])
// if the current code is a .java file, insert into current
// if (current.flavor == PDE) {
SketchFile file = editor.getCurrentTab().getSketchFile();
if (file.isExtension(Sketch.SKETCH_EXTENSIONS)) editor.selectTab(0);
// could also scan the text in the file to see if each import
// statement is already in there, but if the user has the import
// commented out, then this will be a problem.
StringBuilder buffer = new StringBuilder();
for (String aList : list) {
buffer.append("#include <");
buffer.append(aList);
buffer.append(">\n");
}
buffer.append('\n');
buffer.append(editor.getCurrentTab().getText());
editor.getCurrentTab().setText(buffer.toString());
editor.getCurrentTab().setSelection(0, 0); // scroll to start
}
/** 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();
}
}
}
/**
* 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();
}
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));
}
}