137 lines
4.2 KiB
Python
137 lines
4.2 KiB
Python
# Author: Neal Shrader <neal@digitalocean.com>
|
|
# Author: Ben Howard <bh@digitalocean.com>
|
|
#
|
|
# This file is part of cloud-init. See LICENSE file for license information.
|
|
|
|
# DigitalOcean Droplet API:
|
|
# https://developers.digitalocean.com/documentation/metadata/
|
|
|
|
import logging
|
|
|
|
import cloudinit.sources.helpers.digitalocean as do_helper
|
|
from cloudinit import lifecycle, sources, util
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
BUILTIN_DS_CONFIG = {
|
|
"metadata_url": "http://169.254.169.254/metadata/v1.json",
|
|
}
|
|
|
|
# Wait for a up to a minute, retrying the meta-data server
|
|
# every 2 seconds.
|
|
MD_RETRIES = 30
|
|
MD_TIMEOUT = 2
|
|
MD_WAIT_RETRY = 2
|
|
MD_USE_IPV4LL = True
|
|
|
|
|
|
class DataSourceDigitalOcean(sources.DataSource):
|
|
|
|
dsname = "DigitalOcean"
|
|
|
|
def __init__(self, sys_cfg, distro, paths):
|
|
sources.DataSource.__init__(self, sys_cfg, distro, paths)
|
|
self.distro = distro
|
|
self.metadata = dict()
|
|
self.ds_cfg = util.mergemanydict(
|
|
[
|
|
util.get_cfg_by_path(
|
|
sys_cfg, ["datasource", "DigitalOcean"], {}
|
|
),
|
|
BUILTIN_DS_CONFIG,
|
|
]
|
|
)
|
|
self._deprecate()
|
|
self.metadata_address = self.ds_cfg["metadata_url"]
|
|
self.retries = self.ds_cfg.get("retries", MD_RETRIES)
|
|
self.timeout = self.ds_cfg.get("timeout", MD_TIMEOUT)
|
|
self.use_ip4LL = self.ds_cfg.get("use_ip4LL", MD_USE_IPV4LL)
|
|
self.wait_retry = self.ds_cfg.get("wait_retry", MD_WAIT_RETRY)
|
|
self._network_config = None
|
|
self.metadata_full = None
|
|
|
|
def _unpickle(self, ci_pkl_version: int) -> None:
|
|
super()._unpickle(ci_pkl_version)
|
|
self._deprecate()
|
|
|
|
def _deprecate(self):
|
|
lifecycle.deprecate(
|
|
deprecated="DataSourceDigitalOcean",
|
|
deprecated_version="23.2",
|
|
extra_message="Deprecated in favour of DataSourceConfigDrive.",
|
|
)
|
|
|
|
def _get_sysinfo(self):
|
|
return do_helper.read_sysinfo()
|
|
|
|
def _get_data(self):
|
|
(is_do, droplet_id) = self._get_sysinfo()
|
|
|
|
# only proceed if we know we are on DigitalOcean
|
|
if not is_do:
|
|
return False
|
|
|
|
LOG.info("Running on DigitalOcean. droplet_id=%s", droplet_id)
|
|
|
|
ipv4LL_nic = None
|
|
if self.use_ip4LL:
|
|
ipv4LL_nic = do_helper.assign_ipv4_link_local(self.distro)
|
|
|
|
md = do_helper.read_metadata(
|
|
self.metadata_address,
|
|
timeout=self.timeout,
|
|
sec_between=self.wait_retry,
|
|
retries=self.retries,
|
|
)
|
|
|
|
self.metadata_full = md
|
|
self.metadata["instance-id"] = md.get("droplet_id", droplet_id)
|
|
self.metadata["local-hostname"] = md.get("hostname", droplet_id)
|
|
self.metadata["interfaces"] = md.get("interfaces")
|
|
self.metadata["public-keys"] = md.get("public_keys")
|
|
self.metadata["availability_zone"] = md.get("region", "default")
|
|
self.vendordata_raw = md.get("vendor_data", None)
|
|
self.userdata_raw = md.get("user_data", None)
|
|
|
|
if ipv4LL_nic:
|
|
do_helper.del_ipv4_link_local(self.distro, ipv4LL_nic)
|
|
|
|
return True
|
|
|
|
def check_instance_id(self, sys_cfg):
|
|
return sources.instance_id_matches_system_uuid(
|
|
self.get_instance_id(), "system-serial-number"
|
|
)
|
|
|
|
@property
|
|
def network_config(self):
|
|
"""Configure the networking. This needs to be done each boot, since
|
|
the IP information may have changed due to snapshot and/or
|
|
migration.
|
|
"""
|
|
|
|
if self._network_config:
|
|
return self._network_config
|
|
|
|
interfaces = self.metadata.get("interfaces")
|
|
LOG.debug(interfaces)
|
|
if not interfaces:
|
|
raise RuntimeError("Unable to get meta-data from server....")
|
|
|
|
nameservers = self.metadata_full["dns"]["nameservers"]
|
|
self._network_config = do_helper.convert_network_configuration(
|
|
interfaces, nameservers
|
|
)
|
|
return self._network_config
|
|
|
|
|
|
# Used to match classes to dependencies
|
|
datasources = [
|
|
(DataSourceDigitalOcean, (sources.DEP_FILESYSTEM,)),
|
|
]
|
|
|
|
|
|
# Return a list of data sources that match this set of dependencies
|
|
def get_datasource_list(depends):
|
|
return sources.list_from_depends(depends, datasources)
|