1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
# Copyright (c) 2020 Egor Tensin <egor@tensin.name>
# This file is part of the "cmake-common" project.
# For details, see https://github.com/egor-tensin/cmake-common
# Distributed under the MIT License.
import abc
from contextlib import contextmanager
import logging
import os.path
import shutil
from project.boost.directory import BoostDir
from project import utils
class Archive:
def __init__(self, version, path):
self.version = version
self.path = path
@property
def dir_name(self):
return self.version.dir_name
def unpack(self, dest_dir):
path = os.path.join(dest_dir, self.dir_name)
if os.path.exists(path):
raise RuntimeError(f"Boost directory already exists: {path}")
logging.info("Unpacking Boost to: %s", path)
shutil.unpack_archive(self.path, dest_dir)
return BoostDir(path)
class ArchiveStorage(abc.ABC):
@abc.abstractmethod
def get_archive(self, version):
pass
@contextmanager
@abc.abstractmethod
def write_archive(self, version, contents):
pass
class PermanentStorage(ArchiveStorage):
def __init__(self, cache_dir):
self._dir = cache_dir
def _archive_path(self, version):
return os.path.join(self._dir, version.archive_name)
def get_archive(self, version):
path = self._archive_path(version)
if os.path.exists(path):
return path
return None
@contextmanager
def write_archive(self, version, contents):
path = self._archive_path(version)
logging.info("Writing Boost archive: %s", path)
if os.path.exists(path):
raise RuntimeError(f"cannot download Boost, file already exists: {path}")
with open(path, mode="w+b") as dest:
dest.write(contents)
yield path
class TemporaryStorage(ArchiveStorage):
def __init__(self, temp_dir):
self._dir = temp_dir
def get_archive(self, version):
return None
@contextmanager
def write_archive(self, version, contents):
tmp = utils.temp_file(
prefix=f"boost_{version}_", suffix=version.archive_ext, dir=self._dir
)
with tmp as archive_path:
with open(archive_path, mode="wb") as fd:
fd.write(contents)
yield archive_path
|
