79 lines
2.3 KiB
C++
79 lines
2.3 KiB
C++
|
/*
|
||
|
|
* Copyright (c) Meta Platforms, Inc. and 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/String.h>
|
||
|
|
|
||
|
|
#include <memory>
|
||
|
|
|
||
|
|
namespace facebook {
|
||
|
|
namespace react {
|
||
|
|
|
||
|
|
constexpr uint32_t RAMBundleRegistry::MAIN_BUNDLE_ID;
|
||
|
|
|
||
|
|
std::unique_ptr<RAMBundleRegistry> RAMBundleRegistry::singleBundleRegistry(
|
||
|
|
std::unique_ptr<JSModulesUnbundle> mainBundle) {
|
||
|
|
return std::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 std::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
|