šablona/funkce/src/cz/frantovo/xmlWebGenerator/makra/Skriptování.java
author František Kučera <franta-hg@frantovo.cz>
Fri, 06 Jul 2012 14:52:05 +0200
changeset 113 18bf0044f5ab
parent 108 8d9cab64c335
child 136 d5feb9d8ebc3
permissions -rw-r--r--
#20 Skriptování: uvnitř zadání skriptu lze používat jiná makra (interpretují se před provedením skriptu).
     1 /**
     2  * XML Web generátor – program na generování webových stránek
     3  * Copyright © 2012 František Kučera (frantovo.cz)
     4  *
     5  * This program is free software: you can redistribute it and/or modify
     6  * it under the terms of the GNU General Public License as published by
     7  * the Free Software Foundation, either version 3 of the License, or
     8  * (at your option) any later version.
     9  *
    10  * This program is distributed in the hope that it will be useful,
    11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    13  * GNU General Public License for more details.
    14  *
    15  * You should have received a copy of the GNU General Public License
    16  * along with this program. If not, see <http://www.gnu.org/licenses/>.
    17  */
    18 package cz.frantovo.xmlWebGenerator.makra;
    19 
    20 import static cz.frantovo.xmlWebGenerator.NástrojeCLI.načtiProud;
    21 import static cz.frantovo.xmlWebGenerator.Funkce.spojText;
    22 import static cz.frantovo.xmlWebGenerator.Xmlns.*;
    23 import java.io.ByteArrayInputStream;
    24 import java.io.File;
    25 import java.io.PrintStream;
    26 import java.net.URI;
    27 import java.util.Collections;
    28 import java.util.HashMap;
    29 import java.util.Map;
    30 import javax.xml.parsers.DocumentBuilder;
    31 import javax.xml.parsers.DocumentBuilderFactory;
    32 import javax.xml.transform.Source;
    33 import javax.xml.transform.dom.DOMSource;
    34 import org.w3c.dom.Document;
    35 import org.w3c.dom.Node;
    36 
    37 /**
    38  * Provedeme skript a do stránky vložíme jeho výstup.
    39  *
    40  * @author František Kučera (frantovo.cz)
    41  */
    42 public class Skriptování {
    43 
    44 	private enum FORMÁT {
    45 
    46 		xml,
    47 		xhtml,
    48 		text
    49 	}
    50 	/**
    51 	 * klíč = jazyk – např. bash
    52 	 * hodnota = interpret – např. /bin/bash
    53 	 */
    54 	private static final Map<String, String> interpreti;
    55 
    56 	static {
    57 		Map<String, String> podporovanýJazyk = new HashMap<String, String>();
    58 		podporovanýJazyk.put("bash", "/bin/bash");
    59 		podporovanýJazyk.put("perl", "/usr/bin/perl");
    60 		podporovanýJazyk.put("php", "/usr/bin/php");
    61 		podporovanýJazyk.put("python", "/usr/bin/python");
    62 		interpreti = Collections.unmodifiableMap(podporovanýJazyk);
    63 	}
    64 
    65 	/**
    66 	 * TODO: podporovat i složitější scénáře (např. kompilaci),
    67 	 * než jen vložení do souboru a přidání správného záhlaví.
    68 	 *
    69 	 * @param skriptText skript k vykonání
    70 	 * @param skriptSoubor cesta k souboru se skriptem/programem
    71 	 * @param jazyk programovací jazyk
    72 	 * @param výstupníFormát text (výchozí) | xml (v tom případě kontrolujeme validitu)
    73 	 * @param uriStránky URI aktuálně generované stránky → proměnná prostředí
    74 	 * @param nadpisStránky nadpis stránky → proměnná prostředí
    75 	 * @param perexStránky perex stránky → proměnná prostředí
    76 	 * @return výstup příkazu buď jako textový řetězec nebo jako XML (DOMSource)
    77 	 */
    78 	public static Source interpretuj(String[] skriptText, String skriptSoubor, String jazyk, String výstupníFormát, String uriStránky, String nadpisStránky, String perexStránky) throws Exception {
    79 		String výstupSkriptu = získejVýstupSkriptu(spojText(skriptText), skriptSoubor, jazyk, uriStránky, nadpisStránky, perexStránky);
    80 		return vyrobXml(výstupSkriptu, zjistiFormát(výstupníFormát));
    81 	}
    82 
    83 	private static String získejVýstupSkriptu(String skriptText, String skriptSoubor, String jazyk, String uriStránky, String nadpisStránky, String perexStránky) throws Exception {
    84 
    85 		try {
    86 			if (isNeprázdný(skriptSoubor)) {
    87 				System.err.println("\tInterpretuji skript ze souboru: " + skriptSoubor);
    88 			} else {
    89 				System.err.println("\tInterpretuji skript v jazyce:   " + jazyk);
    90 			}
    91 
    92 			File souborStránky = new File(new URI(uriStránky));
    93 			File f;
    94 
    95 			if (isNeprázdný(skriptText)) {
    96 				/** Skript je zadán uvnitř elementu přímo ve stránce */
    97 				String interpret = interpreti.get(jazyk);
    98 				if (interpret == null) {
    99 					throw new Exception("Neznámý skriptovací jazyk: " + jazyk);
   100 				}
   101 
   102 				f = File.createTempFile("xml-web-generátor-", ".skript");
   103 				f.deleteOnExit();
   104 				f.setExecutable(true);
   105 
   106 				PrintStream ps = new PrintStream(f);
   107 				ps.print("#!");
   108 				ps.println(interpret);
   109 				ps.println();
   110 				ps.print(skriptText);
   111 				ps.close();
   112 			} else if (isNeprázdný(skriptSoubor)) {
   113 				/** Skript/program je uložen v externím souboru */
   114 				if (skriptSoubor.startsWith(File.separator)) {
   115 					/** absolutní cesta */
   116 					f = new File(skriptSoubor);
   117 				} else {
   118 					/** relativní cesta */
   119 					f = new File(souborStránky.getParent(), File.separatorChar + skriptSoubor);
   120 				}
   121 
   122 				if (!f.canExecute()) {
   123 					throw new Exception("Soubor se skriptem není spustitelný → nastavte: chmod +x " + f);
   124 				}
   125 			} else {
   126 				throw new Exception("Musí být vyplněn text skriptu, nebo cesta k souboru.");
   127 			}
   128 
   129 
   130 			String[] prostředí = new String[]{
   131 				"LANG=" + System.getenv("LANG"),
   132 				"USER=" + System.getenv("USER"),
   133 				"XWG_SKRIPTOVANI_JAVA=" + "šablona" + File.separator + "funkce" + File.separator + "src" + File.separator + Skriptování.class.getName().replaceAll("\\.", File.separator) + ".java",
   134 				"XWG_STRANKA_URI=" + uriStránky, // env:URI aktuálně zpracovávané stránky
   135 				"XWG_STRANKA_SOUBOR=" + souborStránky.getAbsolutePath(), // env:absolutní cesta k souboru
   136 				"XWG_STRANKA_NADPIS=" + nadpisStránky, // env:nadpis stránky
   137 				"XWG_STRANKA_PEREX=" + perexStránky // env:perex stránky
   138 			};
   139 
   140 			Runtime r = Runtime.getRuntime();
   141 			Process p = r.exec(new String[]{f.getAbsolutePath()}, prostředí);
   142 
   143 			String výsledek = načtiProud(p.getInputStream());
   144 			String chyby = načtiProud(p.getErrorStream());
   145 
   146 			p.waitFor();
   147 
   148 			if (p.exitValue() == 0) {
   149 				if (chyby.length() > 0) {
   150 					System.err.println("--- Chybový výstup skriptu -----");
   151 					System.err.println(chyby);
   152 					System.err.println("--------------------------------");
   153 					System.err.println("Nicméně skript skončil úspěšně, takže pokračujeme dál.");
   154 				}
   155 
   156 				return výsledek.trim();
   157 			} else {
   158 				System.err.println("--- Standardní výstup skriptu: -----");
   159 				System.err.println(výsledek);
   160 				System.err.println("--- Cyhbový výstup skriptu: ---------");
   161 				System.err.println(chyby);
   162 				System.err.println("--------------------------------------");
   163 				throw new Exception("Návratová hodnota: " + p.exitValue());
   164 			}
   165 		} catch (Exception e) {
   166 			System.err.println("Došlo k chybě při vykonávání skriptu.");
   167 			System.err.println("--------");
   168 			System.err.println(skriptText);
   169 			System.err.println("--------");
   170 			e.printStackTrace(System.err);
   171 			throw e;
   172 		}
   173 	}
   174 
   175 	private static boolean isNeprázdný(String s) {
   176 		return !(s == null || s.trim().isEmpty());
   177 	}
   178 
   179 	private static FORMÁT zjistiFormát(String výstupníFormát) {
   180 		try {
   181 			return FORMÁT.valueOf(výstupníFormát);
   182 		} catch (NullPointerException e) {
   183 			return FORMÁT.text;
   184 		} catch (IllegalArgumentException e) {
   185 			return FORMÁT.text;
   186 		}
   187 	}
   188 
   189 	/**
   190 	 * @param zadání výstup vygenerovaný skriptem
   191 	 * @param xmlFormát formát zadání: true = xml fragment | false = prostý text
   192 	 * @return xml fragment nebo prostý text zabalený do html/body
   193 	 * @throws Exception
   194 	 */
   195 	private static Source vyrobXml(String zadání, FORMÁT formát) throws Exception {
   196 		DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
   197 		DocumentBuilder db = dbf.newDocumentBuilder();
   198 		Document d;
   199 
   200 
   201 		if (formát == FORMÁT.text) {
   202 			d = db.newDocument();
   203 			Node html = d.createElementNS(XHTML, "html");
   204 			Node body = d.createElementNS(XHTML, "body");
   205 			Node text = d.createTextNode(zadání);
   206 			body.appendChild(text);
   207 			html.appendChild(body);
   208 			d.appendChild(html);
   209 		} else {
   210 			if (formát == FORMÁT.xhtml) {
   211 				zadání = "<html xmlns='" + XHTML + "' xmlns:m='" + MAKRO + "'><body>" + zadání + "</body></html>";
   212 			}
   213 			try {
   214 				d = db.parse(new ByteArrayInputStream(zadání.getBytes()));
   215 			} catch (Exception e) {
   216 				System.err.println("Chyba: Skript vrátil neplatné XML.");
   217 				throw e;
   218 			}
   219 		}
   220 
   221 		return new DOMSource(d);
   222 	}
   223 }