001/**
002 *
003 * Copyright (c) 2014, the Railo Company Ltd. All rights reserved.
004 *
005 * This library is free software; you can redistribute it and/or
006 * modify it under the terms of the GNU Lesser General Public
007 * License as published by the Free Software Foundation; either 
008 * version 2.1 of the License, or (at your option) any later version.
009 * 
010 * This library is distributed in the hope that it will be useful,
011 * but WITHOUT ANY WARRANTY; without even the implied warranty of
012 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
013 * Lesser General Public License for more details.
014 * 
015 * You should have received a copy of the GNU Lesser General Public 
016 * License along with this library.  If not, see <http://www.gnu.org/licenses/>.
017 * 
018 **/
019package lucee.runtime.functions.file;
020
021import java.io.IOException;
022
023import lucee.commons.io.res.Resource;
024import lucee.commons.lang.StringUtil;
025import lucee.runtime.PageContext;
026import lucee.runtime.PageContextImpl;
027import lucee.runtime.exp.PageException;
028import lucee.runtime.op.Caster;
029
030public class FileWrite {
031
032        public static String call(PageContext pc, Object obj, Object data) throws PageException {
033                return call(pc,obj,data,((PageContextImpl)pc).getResourceCharset().name());
034        }
035        
036        public static String call(PageContext pc, Object obj, Object data,String charset) throws PageException {
037                FileStreamWrapper fsw=null;
038                boolean close=false;
039                if(StringUtil.isEmpty(charset,true))
040                        charset=((PageContextImpl)pc).getResourceCharset().name();
041                try {
042                        try {
043                                if(obj instanceof FileStreamWrapper) {
044                                        fsw=(FileStreamWrapper)obj;
045                                }
046                                else {
047                                        close=true;
048                                        Resource res = Caster.toResource(pc,obj,false);
049                                        pc.getConfig().getSecurityManager().checkFileLocation(res);
050                                        fsw=new FileStreamWrapperWrite(res,charset,false,false);
051                                }
052                                fsw.write(data);
053                        }
054                        finally {
055                                if(close && fsw!=null)fsw.close();
056                        }
057                        
058                } 
059                catch (IOException e) {
060                        throw Caster.toPageException(e);
061                }
062                return null;
063        }
064}