initial commit

This commit is contained in:
ston1th 2020-06-02 00:38:26 +02:00
commit 9c7f7894b9
23 changed files with 1656 additions and 0 deletions

122
pvecloud/cloud.go Normal file
View file

@ -0,0 +1,122 @@
/*
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package pvecloud
import (
"fmt"
"io"
"os"
"strconv"
pve "git.giftfish.de/ston1th/pve-go"
cloudprovider "k8s.io/cloud-provider"
)
const (
providerName = pve.PVEScheme
scheme = providerName + "://"
)
type cloud struct {
client *pve.Client
instances cloudprovider.Instances
//zones cloudprovider.Zones
//routes cloudprovider.Routes
//network string
}
func newCloud(_ io.Reader) (cloudprovider.Interface, error) {
api := os.Getenv("PVE_API")
if api == "" {
return nil, fmt.Errorf("missing required env PVE_API")
}
user := os.Getenv("PVE_USER")
if user == "" {
return nil, fmt.Errorf("missing required env PVE_USER")
}
pass := os.Getenv("PVE_PASSWORD")
if pass == "" {
return nil, fmt.Errorf("missing required env PVE_PASSWORD")
}
//node := os.Getenv("NODE_NAME")
insecure, _ := strconv.ParseBool(os.Getenv("PVE_INSECURE"))
client := pve.NewClient(
pve.WithEndpoint(api),
pve.WithCredentials(user, pass),
pve.WithInsecureClient(insecure),
//pvego.WithDebugWriter(os.Stderr),
)
return &cloud{
client: client,
instances: newInstances(client),
//zones: newZones(client, nodeName),
//routes: nil,
//network: network,
}, nil
}
func (c *cloud) Initialize(clientBuilder cloudprovider.ControllerClientBuilder, stop <-chan struct{}) {
}
func (c *cloud) Instances() (cloudprovider.Instances, bool) {
return c.instances, true
}
func (c *cloud) Zones() (cloudprovider.Zones, bool) {
//return c.zones, true
return nil, false
}
func (c *cloud) LoadBalancer() (cloudprovider.LoadBalancer, bool) {
return nil, false
}
func (c *cloud) Clusters() (cloudprovider.Clusters, bool) {
return nil, false
}
func (c *cloud) Routes() (cloudprovider.Routes, bool) {
return nil, false
/*
if len(c.network) > 0 {
r, err := newRoutes(c.client, c.network)
if err != nil {
return nil, false
}
return r, true
}
return nil, false // If no network is configured, disable the routes part
*/
}
func (c *cloud) ProviderName() string {
return providerName
}
func (c *cloud) ScrubDNS(nameservers, searches []string) (nsOut, srchOut []string) {
return nil, nil
}
func (c *cloud) HasClusterID() bool {
// TODO set to true
return false
}
func init() {
cloudprovider.RegisterCloudProvider(providerName, func(config io.Reader) (cloudprovider.Interface, error) {
return newCloud(config)
})
}

143
pvecloud/instances.go Normal file
View file

@ -0,0 +1,143 @@
/*
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package pvecloud
import (
"context"
"net"
"strings"
cloudprovider "k8s.io/cloud-provider"
pve "git.giftfish.de/ston1th/pve-go"
v1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/types"
)
// InstanceType is not implemented
const InstanceType = "pveServer"
type instances struct {
client *pve.Client
}
func newInstances(client *pve.Client) *instances {
return &instances{client}
}
func (i *instances) NodeAddressesByProviderID(ctx context.Context, providerID string) ([]v1.NodeAddress, error) {
server, err := getServerByProviderID(ctx, i.client, providerID)
if err != nil {
return nil, err
}
return i.nodeAddresses(ctx, server), nil
}
func (i *instances) NodeAddresses(ctx context.Context, nodeName types.NodeName) ([]v1.NodeAddress, error) {
server, err := getServerByName(ctx, i.client, string(nodeName))
if err != nil {
return nil, err
}
return i.nodeAddresses(ctx, server), nil
}
func (i *instances) InstanceID(ctx context.Context, nodeName types.NodeName) (string, error) {
server, err := getServerByName(ctx, i.client, string(nodeName))
if err != nil {
return "", err
}
return strings.TrimPrefix(server.InstanceID(), scheme), nil
}
func (i *instances) InstanceType(ctx context.Context, nodeName types.NodeName) (string, error) {
/*
server, err := getServerByName(ctx, i.client, string(nodeName))
if err != nil {
return "", err
}
return server.ServerType.Name, nil
*/
return InstanceType, nil
}
func (i *instances) InstanceTypeByProviderID(ctx context.Context, providerID string) (string, error) {
/*
id, err := providerIDToServerID(providerID)
if err != nil {
return "", err
}
server, err := getServerByID(ctx, i.client, id)
if err != nil {
return "", err
}
return server.ServerType.Name, nil
*/
return InstanceType, nil
}
func (i *instances) AddSSHKeyToAllInstances(ctx context.Context, user string, keyData []byte) error {
return cloudprovider.NotImplemented
}
func (i *instances) CurrentNodeName(ctx context.Context, hostname string) (types.NodeName, error) {
return types.NodeName(hostname), nil
}
func (i *instances) InstanceExistsByProviderID(ctx context.Context, providerID string) (exists bool, err error) {
server, err := i.client.Server.GetByURL(ctx, providerID)
if err != nil {
return
}
exists = server != nil
return
}
func (i *instances) InstanceShutdownByProviderID(ctx context.Context, providerID string) (isOff bool, err error) {
server, err := i.client.Server.GetByURL(ctx, providerID)
if err != nil {
return
}
isOff = server != nil && server.Status == pve.ServerStatusStopped
return
}
/*
// TODO
func (i *instances) InstanceMetadataByProviderID(ctx context.Context, providerID string) (*cloudprovider.InstanceMetadata, error) {
addrs, err := i.NodeAddressesByProviderID(ctx, providerID)
if err != nil {
return nil, err
}
return &cloudprovider.InstanceMetadata{
ProviderID: providerID,
NodeAddresses: addrs,
}, nil
}
*/
func (i *instances) nodeAddresses(ctx context.Context, server *pve.Server) []v1.NodeAddress {
var internalIP string
ipc := server.IPConfig.Get(0)
if ipc != nil {
ip, _, err := net.ParseCIDR(ipc.IPv4CIDR)
if err == nil {
internalIP = ip.String()
}
}
addrs := []v1.NodeAddress{
{Type: v1.NodeHostName, Address: server.Name},
{Type: v1.NodeInternalIP, Address: internalIP},
}
return addrs
}

36
pvecloud/util.go Normal file
View file

@ -0,0 +1,36 @@
/*
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package pvecloud
import (
"context"
pve "git.giftfish.de/ston1th/pve-go"
cloudprovider "k8s.io/cloud-provider"
)
func getServerByName(ctx context.Context, c *pve.Client, name string) (server *pve.Server, err error) {
server, err = c.Node.FindServerByName(ctx, name)
if err == pve.ErrServerNotFound {
return nil, cloudprovider.InstanceNotFound
}
return
}
func getServerByProviderID(ctx context.Context, c *pve.Client, providerID string) (server *pve.Server, err error) {
server, err = c.Node.FindServerByURL(ctx, providerID)
if err == pve.ErrServerNotFound {
return nil, cloudprovider.InstanceNotFound
}
return
}