1# Copyright (C) 2010 Google Inc. All rights reserved. 2# 3# Redistribution and use in source and binary forms, with or without 4# modification, are permitted provided that the following conditions 5# are met: 6# 7# 1. Redistributions of source code must retain the above copyright 8# notice, this list of conditions and the following disclaimer. 9# 2. Redistributions in binary form must reproduce the above copyright 10# notice, this list of conditions and the following disclaimer in the 11# documentation and/or other materials provided with the distribution. 12# 13# THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 14# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 15# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 16# DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY 17# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 18# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 19# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 20# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 21# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 22# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 23 24from __future__ import with_statement 25 26from webkitpy.common.system.filesystem import FileSystem 27 28 29class FileSetFileHandle(object): 30 """Points to a file that resides in a file set""" 31 def __init__(self, fileset, filename, filesystem=None): 32 self._filename = filename 33 self._fileset = fileset 34 self._contents = None 35 self._filesystem = filesystem or FileSystem() 36 37 def __str__(self): 38 return "%s:%s" % (self._fileset, self._filename) 39 40 def close(self): 41 pass 42 43 def contents(self): 44 if self._contents is None: 45 self._contents = self._fileset.read(self._filename) 46 return self._contents 47 48 def save_to(self, path, filename=None): 49 if filename is None: 50 self._fileset.extract(self._filename, path) 51 return 52 with self._filesystem.mkdtemp() as temp_dir: 53 self._fileset.extract(self._filename, temp_dir) 54 55 src = self._filesystem.join(temp_dir, self._filename) 56 dest = self._filesystem.join(path, filename) 57 self._filesystem.copyfile(src, dest) 58 59 def delete(self): 60 self._fileset.delete(self._filename) 61 62 def name(self): 63 return self._filename 64 65 def splitext(self): 66 return self._filesystem.splitext(self.name()) 67