76 lines
2.3 KiB
C++
76 lines
2.3 KiB
C++
// Copyright (c) Facebook, Inc. and its affiliates.
|
|
|
|
// This source code is licensed under the MIT license found in the
|
|
// LICENSE file in the root directory of this source tree.
|
|
|
|
#include "RAMBundleRegistry.h"
|
|
|
|
#include <folly/Memory.h>
|
|
#include <folly/String.h>
|
|
|
|
namespace facebook {
|
|
namespace react {
|
|
|
|
constexpr uint32_t RAMBundleRegistry::MAIN_BUNDLE_ID;
|
|
|
|
std::unique_ptr<RAMBundleRegistry> RAMBundleRegistry::singleBundleRegistry(
|
|
std::unique_ptr<JSModulesUnbundle> mainBundle) {
|
|
return folly::make_unique<RAMBundleRegistry>(std::move(mainBundle));
|
|
}
|
|
|
|
std::unique_ptr<RAMBundleRegistry> RAMBundleRegistry::multipleBundlesRegistry(
|
|
std::unique_ptr<JSModulesUnbundle> mainBundle,
|
|
std::function<std::unique_ptr<JSModulesUnbundle>(std::string)> factory) {
|
|
return folly::make_unique<RAMBundleRegistry>(
|
|
std::move(mainBundle), std::move(factory));
|
|
}
|
|
|
|
RAMBundleRegistry::RAMBundleRegistry(
|
|
std::unique_ptr<JSModulesUnbundle> mainBundle,
|
|
std::function<std::unique_ptr<JSModulesUnbundle>(std::string)> factory):
|
|
m_factory(std::move(factory)) {
|
|
m_bundles.emplace(MAIN_BUNDLE_ID, std::move(mainBundle));
|
|
}
|
|
|
|
void RAMBundleRegistry::registerBundle(
|
|
uint32_t bundleId, std::string bundlePath) {
|
|
m_bundlePaths.emplace(bundleId, std::move(bundlePath));
|
|
}
|
|
|
|
JSModulesUnbundle::Module RAMBundleRegistry::getModule(
|
|
uint32_t bundleId, uint32_t moduleId) {
|
|
if (m_bundles.find(bundleId) == m_bundles.end()) {
|
|
if (!m_factory) {
|
|
throw std::runtime_error(
|
|
"You need to register factory function in order to "
|
|
"support multiple RAM bundles."
|
|
);
|
|
}
|
|
|
|
auto bundlePath = m_bundlePaths.find(bundleId);
|
|
if (bundlePath == m_bundlePaths.end()) {
|
|
throw std::runtime_error(
|
|
"In order to fetch RAM bundle from the registry, its file "
|
|
"path needs to be registered first."
|
|
);
|
|
}
|
|
m_bundles.emplace(bundleId, m_factory(bundlePath->second));
|
|
}
|
|
|
|
auto module = getBundle(bundleId)->getModule(moduleId);
|
|
if (bundleId == MAIN_BUNDLE_ID) {
|
|
return module;
|
|
}
|
|
return {
|
|
folly::to<std::string>("seg-", bundleId, '_', std::move(module.name)),
|
|
std::move(module.code),
|
|
};
|
|
}
|
|
|
|
JSModulesUnbundle* RAMBundleRegistry::getBundle(uint32_t bundleId) const {
|
|
return m_bundles.at(bundleId).get();
|
|
}
|
|
|
|
} // namespace react
|
|
} // namespace facebook
|