šablona/funkce/src/cz/frantovo/xmlWebGenerator/makra/Skriptování.java
author František Kučera <franta-hg@frantovo.cz>
Sat, 23 Jun 2012 21:10:21 +0200
changeset 101 024ee27ce4fa
parent 95 eea9c4713045
child 103 aa91d1c6d4c1
permissions -rw-r--r--
drobnosti
franta-hg@94
     1
/**
franta-hg@94
     2
 * XML Web generátor – program na generování webových stránek
franta-hg@94
     3
 * Copyright © 2012 František Kučera (frantovo.cz)
franta-hg@94
     4
 *
franta-hg@94
     5
 * This program is free software: you can redistribute it and/or modify
franta-hg@94
     6
 * it under the terms of the GNU General Public License as published by
franta-hg@94
     7
 * the Free Software Foundation, either version 3 of the License, or
franta-hg@94
     8
 * (at your option) any later version.
franta-hg@94
     9
 *
franta-hg@94
    10
 * This program is distributed in the hope that it will be useful,
franta-hg@94
    11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
franta-hg@94
    12
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
franta-hg@94
    13
 * GNU General Public License for more details.
franta-hg@94
    14
 *
franta-hg@94
    15
 * You should have received a copy of the GNU General Public License
franta-hg@94
    16
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
franta-hg@94
    17
 */
franta-hg@94
    18
package cz.frantovo.xmlWebGenerator.makra;
franta-hg@94
    19
franta-hg@94
    20
import static cz.frantovo.xmlWebGenerator.NástrojeCLI.načtiProud;
franta-hg@95
    21
import java.io.ByteArrayInputStream;
franta-hg@94
    22
import java.io.File;
franta-hg@94
    23
import java.io.PrintStream;
franta-hg@94
    24
import java.net.URI;
franta-hg@94
    25
import java.util.Collections;
franta-hg@94
    26
import java.util.HashMap;
franta-hg@94
    27
import java.util.Map;
franta-hg@95
    28
import javax.xml.parsers.DocumentBuilder;
franta-hg@95
    29
import javax.xml.parsers.DocumentBuilderFactory;
franta-hg@95
    30
import org.w3c.dom.Document;
franta-hg@94
    31
franta-hg@94
    32
/**
franta-hg@94
    33
 * Provedeme skript a do stránky vložíme jeho výstup.
franta-hg@94
    34
 *
franta-hg@94
    35
 * @author František Kučera (frantovo.cz)
franta-hg@94
    36
 */
franta-hg@94
    37
public class Skriptování {
franta-hg@94
    38
franta-hg@94
    39
	/**
franta-hg@94
    40
	 * klíč = jazyk – např. bash
franta-hg@94
    41
	 * hodnota = interpret – např. /bin/bash
franta-hg@94
    42
	 */
franta-hg@94
    43
	private static final Map<String, String> interpreti;
franta-hg@94
    44
franta-hg@94
    45
	static {
franta-hg@94
    46
		Map<String, String> podporovanýJazyk = new HashMap<String, String>();
franta-hg@94
    47
		podporovanýJazyk.put("bash", "/bin/bash");
franta-hg@94
    48
		podporovanýJazyk.put("perl", "/usr/bin/perl");
franta-hg@94
    49
		podporovanýJazyk.put("php", "/usr/bin/php");
franta-hg@94
    50
		podporovanýJazyk.put("python", "/usr/bin/python");
franta-hg@94
    51
		interpreti = Collections.unmodifiableMap(podporovanýJazyk);
franta-hg@94
    52
	}
franta-hg@94
    53
franta-hg@94
    54
	/**
franta-hg@94
    55
	 * TODO: podporovat i složitější scénáře (např. kompilaci),
franta-hg@94
    56
	 * než jen vložení do souboru a přidání správného záhlaví.
franta-hg@94
    57
	 *
franta-hg@94
    58
	 * @param skript program k vykonání
franta-hg@94
    59
	 * @param jazyk programovací jazyk
franta-hg@95
    60
	 * @param výstupníFormát text (výchozí) | xml (v tom případě kontrolujeme validitu)
franta-hg@94
    61
	 * @param uriStránky URI aktuálně generované stránky → proměnná prostředí
franta-hg@94
    62
	 * @param nadpisStránky nadpis stránky → proměnná prostředí
franta-hg@94
    63
	 * @param perexStránky perex stránky → proměnná prostředí
franta-hg@94
    64
	 * @return výstup příkazu
franta-hg@94
    65
	 */
franta-hg@95
    66
	public static String interpretuj(String skript, String jazyk, String výstupníFormát, String uriStránky, String nadpisStránky, String perexStránky) {
franta-hg@94
    67
		try {
franta-hg@94
    68
			System.err.println("\tInterpretuji skript v jazyce: " + jazyk);
franta-hg@94
    69
			String interpret = interpreti.get(jazyk);
franta-hg@94
    70
			if (interpret == null) {
franta-hg@94
    71
				System.err.println("Neznámý skriptovací jazyk: " + jazyk);
franta-hg@94
    72
				return null;
franta-hg@94
    73
			} else {
franta-hg@94
    74
				File f = File.createTempFile("xml-web-generátor-", ".skript");
franta-hg@94
    75
				f.deleteOnExit();
franta-hg@94
    76
franta-hg@94
    77
				PrintStream ps = new PrintStream(f);
franta-hg@94
    78
				ps.print("#!");
franta-hg@94
    79
				ps.println(interpret);
franta-hg@94
    80
				ps.println();
franta-hg@94
    81
				ps.print(skript);
franta-hg@94
    82
				ps.close();
franta-hg@94
    83
franta-hg@94
    84
franta-hg@94
    85
				String[] prostředí = new String[]{
franta-hg@94
    86
					"LANG=" + System.getenv("LANG"),
franta-hg@94
    87
					"USER=" + System.getenv("USER"),
franta-hg@95
    88
					"XWG_SKRIPTOVANI_JAVA=" + "šablona" + File.separator + "funkce" + File.separator + "src" + File.separator + Skriptování.class.getName().replaceAll("\\.", File.separator) + ".java",
franta-hg@95
    89
					"XWG_STRANKA_URI=" + uriStránky, // env:URI aktuálně zpracovávané stránky
franta-hg@95
    90
					"XWG_STRANKA_SOUBOR=" + (new File(new URI(uriStránky)).getAbsolutePath()), // env:absolutní cesta k souboru
franta-hg@95
    91
					"XWG_STRANKA_NADPIS=" + nadpisStránky, // env:nadpis stránky
franta-hg@95
    92
					"XWG_STRANKA_PEREX=" + perexStránky // env:perex stránky
franta-hg@94
    93
				};
franta-hg@94
    94
franta-hg@95
    95
				f.setExecutable(true);
franta-hg@94
    96
				Runtime r = Runtime.getRuntime();
franta-hg@94
    97
				Process p = r.exec(new String[]{f.getAbsolutePath()}, prostředí);
franta-hg@94
    98
franta-hg@94
    99
				String výsledek = načtiProud(p.getInputStream());
franta-hg@94
   100
				String chyby = načtiProud(p.getErrorStream());
franta-hg@94
   101
franta-hg@94
   102
				p.waitFor();
franta-hg@94
   103
franta-hg@94
   104
				if (p.exitValue() == 0) {
franta-hg@94
   105
					if (chyby.length() > 0) {
franta-hg@95
   106
						System.err.println("--- Chybový výstup skriptu -----");
franta-hg@94
   107
						System.err.println(chyby);
franta-hg@94
   108
						System.err.println("--------------------------------");
franta-hg@95
   109
						System.err.println("Nicméně skript skončil úspěšně, takže pokračujeme dál.");
franta-hg@94
   110
					}
franta-hg@94
   111
franta-hg@95
   112
					return připravVýstup(výsledek, výstupníFormát);
franta-hg@94
   113
				} else {
franta-hg@94
   114
					System.err.println("--- Standardní výstup skriptu: -----");
franta-hg@94
   115
					System.err.println(výsledek);
franta-hg@94
   116
					System.err.println("--- Cyhbový výstup skriptu: ---------");
franta-hg@94
   117
					System.err.println(chyby);
franta-hg@94
   118
					System.err.println("--------------------------------------");
franta-hg@94
   119
					throw new RuntimeException("Návratová hodnota: " + p.exitValue());
franta-hg@94
   120
				}
franta-hg@94
   121
			}
franta-hg@94
   122
		} catch (Exception e) {
franta-hg@94
   123
			System.err.println("Došlo k chybě při vykonávání skriptu v jazyce: " + jazyk);
franta-hg@94
   124
			System.err.println("--------");
franta-hg@94
   125
			System.err.println(skript);
franta-hg@94
   126
			System.err.println("--------");
franta-hg@94
   127
			e.printStackTrace(System.err);
franta-hg@94
   128
			return null;
franta-hg@94
   129
		}
franta-hg@94
   130
	}
franta-hg@95
   131
franta-hg@95
   132
	private static String připravVýstup(String výsledek, String formát) {
franta-hg@95
   133
		if ("xml".equals(formát)) {
franta-hg@95
   134
			if (zkontrolujXml(výsledek)) {
franta-hg@95
   135
				return výsledek.trim();
franta-hg@95
   136
			} else {
franta-hg@95
   137
				System.err.println("Chyba v XML generovaném skriptem:");
franta-hg@95
   138
				System.err.println(výsledek);
franta-hg@95
   139
				return null;
franta-hg@95
   140
			}
franta-hg@95
   141
		} else {
franta-hg@95
   142
			return výsledek.trim();
franta-hg@95
   143
		}
franta-hg@95
   144
	}
franta-hg@95
   145
franta-hg@95
   146
	/**
franta-hg@95
   147
	 * @param xml fragment XML vygenerovaný skriptem
franta-hg@95
   148
	 * @return true v případě, že výstup je validním fragmentem XML
franta-hg@95
   149
	 */
franta-hg@95
   150
	private static boolean zkontrolujXml(String xml) {
franta-hg@95
   151
		try {
franta-hg@95
   152
			DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
franta-hg@95
   153
			DocumentBuilder db = dbf.newDocumentBuilder();
franta-hg@95
   154
			xml = "<xml>" + xml + "</xml>";
franta-hg@95
   155
			Document d = db.parse(new ByteArrayInputStream(xml.getBytes()));
franta-hg@95
   156
			return true;
franta-hg@95
   157
		} catch (Exception e) {
franta-hg@95
   158
			e.printStackTrace(System.err);
franta-hg@95
   159
			return false;
franta-hg@95
   160
		}
franta-hg@95
   161
	}
franta-hg@94
   162
}