forked from platformio/platformio-core
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
|
# Copyright (C) Ivan Kravets <me@ikravets.com>
|
||
|
|
# See LICENSE for details.
|
||
|
|
|
||
|
|
import json
|
||
|
|
from os import listdir, makedirs, remove
|
||
|
|
from os.path import isdir, isfile, join
|
||
|
|
from shutil import rmtree
|
||
|
|
from tempfile import gettempdir
|
||
|
|
|
||
|
|
from platformio.downloader import FileDownloader
|
||
|
|
from platformio.exception import LibAlreadyInstalledError, LibNotInstalledError
|
||
|
|
from platformio.unpacker import FileUnpacker
|
||
|
|
from platformio.util import get_api_result
|
||
|
|
|
||
|
|
|
||
|
|
class LibraryManager(object):
|
||
|
|
|
||
|
|
CONFIG_NAME = "library.json"
|
||
|
|
|
||
|
|
def __init__(self, lib_dir):
|
||
|
|
self.lib_dir = lib_dir
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def download(url, dest_dir):
|
||
|
|
fd = FileDownloader(url, dest_dir)
|
||
|
|
fd.start()
|
||
|
|
return fd.get_filepath()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def unpack(pkgpath, dest_dir):
|
||
|
|
fu = FileUnpacker(pkgpath, dest_dir)
|
||
|
|
return fu.start()
|
||
|
|
|
||
|
|
def get_installed(self):
|
||
|
|
items = []
|
||
|
|
for item in listdir(self.lib_dir):
|
||
|
|
conf_path = join(self.lib_dir, item, self.CONFIG_NAME)
|
||
|
|
if isfile(conf_path):
|
||
|
|
items.append(item)
|
||
|
|
return items
|
||
|
|
|
||
|
|
def get_info(self, name):
|
||
|
|
conf_path = join(self.lib_dir, name, self.CONFIG_NAME)
|
||
|
|
if not isfile(conf_path):
|
||
|
|
raise LibNotInstalledError(name)
|
||
|
|
with open(conf_path, "r") as f:
|
||
|
|
return json.load(f)
|
||
|
|
|
||
|
|
def is_installed(self, name):
|
||
|
|
return isfile(join(self.lib_dir, name, self.CONFIG_NAME))
|
||
|
|
|
||
|
|
def install(self, name, version=None):
|
||
|
|
if self.is_installed(name):
|
||
|
|
raise LibAlreadyInstalledError()
|
||
|
|
|
||
|
|
_lib_dir = join(self.lib_dir, name)
|
||
|
|
if not isdir(_lib_dir):
|
||
|
|
makedirs(_lib_dir)
|
||
|
|
|
||
|
|
dlinfo = get_api_result("/lib/download/" + name, dict(version=version)
|
||
|
|
if version else None)
|
||
|
|
try:
|
||
|
|
dlpath = self.download(dlinfo['url'], gettempdir())
|
||
|
|
self.unpack(dlpath, _lib_dir)
|
||
|
|
finally:
|
||
|
|
remove(dlpath)
|
||
|
|
|
||
|
|
return self.is_installed(name)
|
||
|
|
|
||
|
|
def uninstall(self, name):
|
||
|
|
if self.is_installed(name):
|
||
|
|
rmtree(join(self.lib_dir, name))
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
raise LibNotInstalledError(name)
|