OLD | NEW |
1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import re | 5 import re |
6 import xml.dom.minidom as xml | 6 import xml.dom.minidom as xml |
7 | 7 |
8 import file_system | 8 import file_system |
9 from future import Future | 9 from future import Future |
10 | 10 |
11 class SubversionFileSystem(file_system.FileSystem): | 11 class SubversionFileSystem(file_system.FileSystem): |
12 """Class to fetch resources from src.chromium.org. | 12 """Class to fetch resources from src.chromium.org. |
13 """ | 13 """ |
14 def __init__(self, fetcher): | 14 def __init__(self, fetcher): |
15 self._fetcher = fetcher | 15 self._fetcher = fetcher |
16 | 16 |
17 def Read(self, paths): | 17 def Read(self, paths, binary=False): |
18 return Future(delegate=_AsyncFetchFuture(paths, self._fetcher)) | 18 return Future(delegate=_AsyncFetchFuture(paths, self._fetcher, binary)) |
19 | 19 |
20 def Stat(self, path): | 20 def Stat(self, path): |
21 directory = path.rsplit('/', 1)[0] | 21 directory = path.rsplit('/', 1)[0] |
22 dir_html = self._fetcher.Fetch(directory + '/').content | 22 dir_html = self._fetcher.Fetch(directory + '/').content |
23 return self.StatInfo(int(re.search('([0-9]+)', dir_html).group(0))) | 23 return self.StatInfo(int(re.search('([0-9]+)', dir_html).group(0))) |
24 | 24 |
25 class _AsyncFetchFuture(object): | 25 class _AsyncFetchFuture(object): |
26 def __init__(self, paths, fetcher): | 26 def __init__(self, paths, fetcher, binary): |
27 # A list of tuples of the form (path, Future). | 27 # A list of tuples of the form (path, Future). |
28 self._fetches = [] | 28 self._fetches = [] |
29 self._value = {} | 29 self._value = {} |
30 self._error = None | 30 self._error = None |
31 self._fetches = [(path, fetcher.FetchAsync(path)) for path in paths] | 31 self._fetches = [(path, fetcher.FetchAsync(path)) for path in paths] |
| 32 self._binary = binary |
32 | 33 |
33 def _ListDir(self, directory): | 34 def _ListDir(self, directory): |
34 dom = xml.parseString(directory) | 35 dom = xml.parseString(directory) |
35 files = [elem.childNodes[0].data for elem in dom.getElementsByTagName('a')] | 36 files = [elem.childNodes[0].data for elem in dom.getElementsByTagName('a')] |
36 if '..' in files: | 37 if '..' in files: |
37 files.remove('..') | 38 files.remove('..') |
38 return files | 39 return files |
39 | 40 |
40 def Get(self): | 41 def Get(self): |
41 for path, future in self._fetches: | 42 for path, future in self._fetches: |
42 result = future.Get() | 43 result = future.Get() |
43 if result.status_code == 404: | 44 if result.status_code == 404: |
44 self._value[path] = None | 45 self._value[path] = None |
45 elif path.endswith('/'): | 46 elif path.endswith('/'): |
46 self._value[path] = self._ListDir(result.content) | 47 self._value[path] = self._ListDir(result.content) |
| 48 elif not self._binary: |
| 49 self._value[path] = file_system._ProcessFileData(result.content, path) |
47 else: | 50 else: |
48 self._value[path] = file_system._ProcessFileData(result.content, path) | 51 self._value[path] = result.content |
49 if self._error is not None: | 52 if self._error is not None: |
50 raise self._error | 53 raise self._error |
51 return self._value | 54 return self._value |
52 | 55 |
OLD | NEW |