mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-20 22:21:13 +02:00
64b5b32732
This makes use of the external procfs pacakge that is based on the pacakge we are removing here. Signed-off-by: Andrew Rynhard <andrew@andrewrynhard.com>
101 lines
2.7 KiB
Go
101 lines
2.7 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package digitalocean
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net"
|
|
"net/http"
|
|
|
|
"github.com/talos-systems/go-procfs/procfs"
|
|
|
|
"github.com/talos-systems/talos/internal/pkg/runtime"
|
|
"github.com/talos-systems/talos/pkg/download"
|
|
)
|
|
|
|
const (
|
|
// DigitalOceanExternalIPEndpoint displays all external addresses associated with the instance
|
|
DigitalOceanExternalIPEndpoint = "http://169.254.169.254/metadata/v1/interfaces/public/0/ipv4/address"
|
|
// DigitalOceanHostnameEndpoint is the local endpoint for the hostname.
|
|
DigitalOceanHostnameEndpoint = "http://169.254.169.254/metadata/v1/hostname"
|
|
// DigitalOceanUserDataEndpoint is the local endpoint for the config.
|
|
DigitalOceanUserDataEndpoint = "http://169.254.169.254/metadata/v1/user-data"
|
|
)
|
|
|
|
// DigitalOcean is the concrete type that implements the platform.Platform interface.
|
|
type DigitalOcean struct{}
|
|
|
|
// Name implements the platform.Platform interface.
|
|
func (d *DigitalOcean) Name() string {
|
|
return "digital-ocean"
|
|
}
|
|
|
|
// Configuration implements the platform.Platform interface.
|
|
func (d *DigitalOcean) Configuration() ([]byte, error) {
|
|
return download.Download(DigitalOceanUserDataEndpoint)
|
|
}
|
|
|
|
// Mode implements the platform.Platform interface.
|
|
func (d *DigitalOcean) Mode() runtime.Mode {
|
|
return runtime.Cloud
|
|
}
|
|
|
|
// Hostname implements the platform.Platform interface.
|
|
func (d *DigitalOcean) Hostname() (hostname []byte, err error) {
|
|
resp, err := http.Get(DigitalOceanHostnameEndpoint)
|
|
if err != nil {
|
|
return
|
|
}
|
|
// nolint: errcheck
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return hostname, fmt.Errorf("failed to fetch hostname from metadata service: %d", resp.StatusCode)
|
|
}
|
|
|
|
return ioutil.ReadAll(resp.Body)
|
|
}
|
|
|
|
// ExternalIPs implements the runtime.Platform interface.
|
|
func (d *DigitalOcean) ExternalIPs() (addrs []net.IP, err error) {
|
|
var (
|
|
body []byte
|
|
req *http.Request
|
|
resp *http.Response
|
|
)
|
|
|
|
if req, err = http.NewRequest("GET", DigitalOceanExternalIPEndpoint, nil); err != nil {
|
|
return
|
|
}
|
|
|
|
client := &http.Client{}
|
|
if resp, err = client.Do(req); err != nil {
|
|
return
|
|
}
|
|
|
|
// nolint: errcheck
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return addrs, fmt.Errorf("failed to retrieve external addresses for instance")
|
|
}
|
|
|
|
if body, err = ioutil.ReadAll(resp.Body); err != nil {
|
|
return
|
|
}
|
|
|
|
addrs = append(addrs, net.ParseIP(string(body)))
|
|
|
|
return addrs, err
|
|
}
|
|
|
|
// KernelArgs implements the runtime.Platform interface.
|
|
func (d *DigitalOcean) KernelArgs() procfs.Parameters {
|
|
return []*procfs.Parameter{
|
|
procfs.NewParameter("console").Append("ttyS0"),
|
|
}
|
|
}
|