/* ======================================================================== * PlantUML : a free UML diagram generator * ======================================================================== * * (C) Copyright 2009-2024, Arnaud Roques * * Project Info: https://plantuml.com * * If you like this project or if you find it useful, you can support us at: * * https://plantuml.com/patreon (only 1$ per month!) * https://plantuml.com/paypal * * This file is part of PlantUML. * * PlantUML is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * PlantUML distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public * License for more details. * * You should have received a copy of the GNU General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, * USA. * * * Original Author: Arnaud Roques * */ package net.sourceforge.plantuml.tim.stdlib; import java.util.List; import java.util.Map; import java.util.Set; import net.sourceforge.plantuml.OptionFlags; import net.sourceforge.plantuml.security.SecurityUtils; import net.sourceforge.plantuml.tim.EaterException; import net.sourceforge.plantuml.tim.EaterExceptionLocated; import net.sourceforge.plantuml.tim.TContext; import net.sourceforge.plantuml.tim.TFunctionSignature; import net.sourceforge.plantuml.tim.TMemory; import net.sourceforge.plantuml.tim.expression.TValue; import net.sourceforge.plantuml.utils.LineLocation; public class Getenv extends SimpleReturnFunction { public TFunctionSignature getSignature() { return new TFunctionSignature("%getenv", 1); } public boolean canCover(int nbArg, Set namedArgument) { return nbArg == 1; } public TValue executeReturnFunction(TContext context, TMemory memory, LineLocation location, List values, Map named) throws EaterException, EaterExceptionLocated { // ::comment when __CORE__ if (OptionFlags.ALLOW_INCLUDE == false) // ::done return TValue.fromString(""); // ::comment when __CORE__ final String name = values.get(0).toString(); final String value = getenv(name); if (value == null) return TValue.fromString(""); return TValue.fromString(value); // ::done } // ::comment when __CORE__ private String getenv(String name) { // Check, if the script requests secret information. // A plantuml server should have an own SecurityManager to // avoid access to properties and environment variables, but we should // also stop here in other deployments. if (SecurityUtils.isSecurityEnv(name)) return null; final String env = System.getProperty(name); if (env != null) return env; return System.getenv(name); } // ::done }