Application: gvSIG desktop » gvSIG scripting

package org.gvsig.scripting.impl;

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileReader;

import java.io.FileWriter;

import java.io.IOException;

import java.io.Writer;

import java.net.MalformedURLException;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import javax.script.ScriptContext;

import javax.script.ScriptEngine;

import javax.script.ScriptEngineManager;

import javax.script.SimpleBindings;

import javax.swing.ImageIcon;

import org.gvsig.scripting.ScriptingBaseScript;

import org.gvsig.scripting.ScriptingDialog;

import org.gvsig.scripting.ScriptingFolder;

import org.gvsig.scripting.ScriptingHelpManager;

import org.gvsig.scripting.ScriptingManager;

import org.gvsig.scripting.ScriptingProject;

import org.gvsig.scripting.ScriptingScript;

import org.gvsig.scripting.ScriptingUnit;

public class DefaultScriptingManager implements ScriptingManager{

        protected String rootUserFolder;

        protected Map icons;

        protected List systemFolders = new ArrayList();

        protected ScriptEngineManager engineManager = null;

        private SimpleBindings bindings = new SimpleBindings();

        private ScriptingHelpManager helpManager;

        public DefaultScriptingManager(){

                this.rootUserFolder = System.getProperty("user.home") + File.separator + "scripts";

                this.helpManager = new DefaultScriptingHelpManager(this);

                File f = new File (System.getProperty("user.dir")+File.separator+".."+File.separator+"org.gvsig.scripting.lib"+File.separator+"org.gvsig.scripting.lib.api"+File.separator+"target"+File.separator+"site"+File.separator+"apidoc");

                if(f.exists()){

                        try {

                                this.helpManager.importHelp("ScriptingFrameworkHelp", f.toURL());

                        } catch (MalformedURLException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        }

                }

                this.bindings.put("ScriptingManager", this);

        }

        protected ScriptEngineManager getEngineManager(){

                if(this.engineManager == null){

                        this.engineManager = new ScriptEngineManager();

                }

                return this.engineManager;

        }

        public ScriptEngine getEngine(String langName){

                ScriptEngine engine = this.getEngineManager().getEngineByName(langName);

                engine.getBindings(ScriptContext.ENGINE_SCOPE).putAll(bindings);

                return engine;

        }

        public ImageIcon getIcon(String name){

                return (ImageIcon) this.icons.get(name);

        }

        public Map<String,String> getSupportedLanguagesByExtension(){

                Map<String,String> l = new HashMap<String,String>();

                l.put(".py", "python");

                l.put(".js", "javascript");

                l.put(".bsh", "beanshell");

                l.put(".gvy", "groovy");

                return l;

        }

        public Map getExtensionsByLanguages(){

                Map l = getSupportedLanguagesByExtension();

                Map result = new HashMap();

                Iterator it = l.entrySet().iterator();

            while (it.hasNext()) {

                Map.Entry pairs = (Map.Entry)it.next();

                result.put(pairs.getValue(), pairs.getKey());

            }

                return result;

        }

        public boolean validateUnitId(ScriptingFolder folder, String id){

                List units = folder.getUnits();

                String fileName;

                String s[];

                String extension;

                for(int i=0; i<units.size();i++){

                        fileName = ((ScriptingUnit)units.get(i)).getId();

                        s = fileName.split("\\.");

                        extension = null;

                        if (s.length>1){

                                extension ="." +  s[s.length-1];

                                fileName = fileName.substring(0,fileName.length()-extension.length());

                        }

                        if(fileName.equals(id)){

                                return false;

                        }

                }

                return true;

        }

        public static String getStringFromFile(String filename){

                BufferedReader b = null;

                try {

                        b = new BufferedReader(new FileReader(filename));

                } catch (FileNotFoundException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                StringBuffer sb=new StringBuffer();

                String str;

                try {

                        while((str=b.readLine())!=null) {

                                sb.append(str);

                                sb.append('\n');

                        }

                } catch (IOException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                return sb.toString();

        }

        public ScriptingDialog createDialog(ScriptingFolder folder, String id) {

                DefaultScriptingDialog sd = new DefaultScriptingDialog(this);

                sd.setId(id);

                File file = new File(((DefaultScriptingFolder)folder).getPath()+File.separator+id);

                try {

                        file.createNewFile();

                } catch (IOException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                // Eliminamos la extensión, para crear el '.dlg'

                String s[] = id.split("\\.");

                String newId;

                if (s.length>=2){

                        String extension ="." +  s[s.length-1];

                        newId = id.substring(0,id.length()-extension.length());

                }else{

                        newId = id;

                }

                File fileDlg = new File(((DefaultScriptingFolder)folder).getPath()+File.separator+newId+".dlg");

                try {

                        fileDlg.createNewFile();

                } catch (IOException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                // Escribimos en el fichero '.dlg' lo mínimo para que pueda ejecutarse

            Writer output = null; 

            try {

                        output = new BufferedWriter(new FileWriter(fileDlg));

                        output.write("<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?>\n<panel/>");

                        output.close();

                } catch (IOException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                sd.load(folder, id);

                return (ScriptingDialog)sd;

        }

        public ScriptingFolder createFolder(ScriptingFolder folder, String id) {

                DefaultScriptingFolder fd = new DefaultScriptingFolder(this);

                File dir = new File(((DefaultScriptingFolder)folder).getPath()+File.separator+id);

                dir.mkdir();

                fd.load(folder,id);

                return (ScriptingFolder)fd;

        }

        public ScriptingProject createProject(ScriptingFolder folder, String id) {

                // TODO Auto-generated method stub

                return null;

        }

        public ScriptingScript createScript(ScriptingFolder folder, String id) {

                DefaultScriptingScript sc = new DefaultScriptingScript(this);

                sc.setId(id);

                File file = new File(((DefaultScriptingFolder)folder).getPath()+File.separator+id);

                try {

                        file.createNewFile();

                } catch (IOException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                sc.load(folder, id);

                return (ScriptingScript)sc;

        }

        public ScriptingBaseScript getScript(File file) {

                ScriptingBaseScript script = (ScriptingBaseScript) this.getUnit(file);

                if (script == null){

                        throw new RuntimeException(file.getPath());

                }

                return script;

        }

        public ScriptingFolder getFolder(File file) {

                ScriptingFolder folder = (ScriptingFolder) this.getUnit(file);

                if (folder == null){

                        throw new RuntimeException(file.getAbsolutePath());

                }

                return folder;

        }

        public ScriptingUnit getUnit(File file) {

                ScriptingFolder folder = null;

                ScriptingUnit unit = null;

                folder = this.getUserFolder();

                unit = folder.getUnit(file);

                if( unit != null ) {

                        return unit;

                }

                folder = this.getSystemFolder();

                unit = folder.getUnit(file);

                if( unit != null ) {

                        return unit;

                }

                return null;

        }

        public ScriptingFolder getSystemFolder() {

                return new SystemFolder((ScriptingManager)this);

        }

        public ScriptingFolder getUserFolder() {

                return new UserFolder((ScriptingManager)this,new File(this.getRootUserFolder()));

        }

        public String getRootUserFolder() {

                return this.rootUserFolder;

        }

        public void registerSystemFolder(String name, File folder) {

                this.systemFolders.add(new RegisterSystemFolder(name, folder));

        }

        public List getSystemFolders(){

                return this.systemFolders;

        }

        public static class RegisterSystemFolder {

                public String name;

                public File folder;

                public RegisterSystemFolder(String name, File folder) {

                        this.name = name;

                        this.folder = folder;

                }

        }

        public String getExtensionByLanguage(String langName) {

                Map extensions = getSupportedLanguagesByExtension();

                Iterator iterator = extensions.keySet().iterator();

                String extension = null;

                String s = null;

                while (iterator. hasNext()){

                        extension = (String) iterator.next();

                        s = (String) extensions.get(extension);

                        if(s.equals(langName)){

                                return extension;

                        }

                }

                return null;

        }

        public Object get(String key) {

                return this.bindings.get(key);

        }

        public void put(String key, Object value) {

                this.bindings.put(key,value); 

        }

        public ScriptingHelpManager getHelpManager() {

                return this.helpManager;

        }

}