mirror of https://github.com/kubernetes/kops.git
146 lines
3.6 KiB
Go
146 lines
3.6 KiB
Go
/*
|
|
Copyright 2019 The Kubernetes Authors.
|
|
|
|
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 awstasks
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"k8s.io/kops/upup/pkg/fi"
|
|
"k8s.io/kops/upup/pkg/fi/cloudup/awsup"
|
|
"k8s.io/kops/upup/pkg/fi/cloudup/cloudformation"
|
|
"k8s.io/kops/upup/pkg/fi/cloudup/terraform"
|
|
)
|
|
|
|
// +kops:fitask
|
|
type VPCCIDRBlock struct {
|
|
Name *string
|
|
Lifecycle *fi.Lifecycle
|
|
|
|
VPC *VPC
|
|
CIDRBlock *string
|
|
|
|
// Shared is set if this is a shared VPC
|
|
Shared *bool
|
|
}
|
|
|
|
func (e *VPCCIDRBlock) Find(c *fi.Context) (*VPCCIDRBlock, error) {
|
|
cloud := c.Cloud.(awsup.AWSCloud)
|
|
|
|
vpcID := e.VPC.ID
|
|
|
|
vpc, err := cloud.DescribeVPC(*vpcID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
actual := &VPCCIDRBlock{
|
|
CIDRBlock: e.CIDRBlock,
|
|
}
|
|
actual.VPC = &VPC{ID: vpc.VpcId}
|
|
|
|
// Prevent spurious changes
|
|
actual.Shared = e.Shared
|
|
actual.Name = e.Name
|
|
actual.Lifecycle = e.Lifecycle
|
|
|
|
return actual, nil
|
|
}
|
|
|
|
func (e *VPCCIDRBlock) Run(c *fi.Context) error {
|
|
return fi.DefaultDeltaRunMethod(e, c)
|
|
}
|
|
|
|
func (s *VPCCIDRBlock) CheckChanges(a, e, changes *VPCCIDRBlock) error {
|
|
if e.VPC == nil {
|
|
return fi.RequiredField("VPC")
|
|
}
|
|
|
|
if e.CIDRBlock == nil {
|
|
return fi.RequiredField("CIDRBlock")
|
|
}
|
|
|
|
if a != nil && changes != nil {
|
|
if changes.VPC != nil {
|
|
return fi.CannotChangeField("VPC")
|
|
}
|
|
|
|
if changes.CIDRBlock != nil {
|
|
return fi.CannotChangeField("CIDRBlock")
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (_ *VPCCIDRBlock) RenderAWS(t *awsup.AWSAPITarget, a, e, changes *VPCCIDRBlock) error {
|
|
shared := fi.BoolValue(e.Shared)
|
|
if shared {
|
|
// Verify the CIDR block was found.
|
|
if a == nil {
|
|
return fmt.Errorf("CIDR block %q not found", fi.StringValue(e.CIDRBlock))
|
|
}
|
|
}
|
|
|
|
if changes.CIDRBlock != nil {
|
|
request := &ec2.AssociateVpcCidrBlockInput{
|
|
VpcId: e.VPC.ID,
|
|
CidrBlock: e.CIDRBlock,
|
|
}
|
|
|
|
_, err := t.Cloud.EC2().AssociateVpcCidrBlock(request)
|
|
if err != nil {
|
|
return fmt.Errorf("error associating AdditionalCIDR to VPC: %v", err)
|
|
}
|
|
}
|
|
|
|
return nil // no tags
|
|
}
|
|
|
|
type terraformVPCCIDRBlock struct {
|
|
VPCID *terraform.Literal `json:"vpc_id" cty:"vpc_id"`
|
|
CIDRBlock *string `json:"cidr_block" cty:"cidr_block"`
|
|
}
|
|
|
|
func (_ *VPCCIDRBlock) RenderTerraform(t *terraform.TerraformTarget, a, e, changes *VPCCIDRBlock) error {
|
|
|
|
// When this has been enabled please fix test TestAdditionalCIDR in integration_test.go to run runTestAWS.
|
|
tf := &terraformVPCCIDRBlock{
|
|
VPCID: e.VPC.TerraformLink(),
|
|
CIDRBlock: e.CIDRBlock,
|
|
}
|
|
|
|
// Terraform 0.12 doesn't support resource names that start with digits. See #7052
|
|
// and https://www.terraform.io/upgrade-guides/0-12.html#pre-upgrade-checklist
|
|
name := fmt.Sprintf("cidr-%v", *e.Name)
|
|
return t.RenderResource("aws_vpc_ipv4_cidr_block_association", name, tf)
|
|
}
|
|
|
|
type cloudformationVPCCIDRBlock struct {
|
|
VPCID *cloudformation.Literal `json:"VpcId"`
|
|
CIDRBlock *string `json:"CidrBlock"`
|
|
}
|
|
|
|
func (_ *VPCCIDRBlock) RenderCloudformation(t *cloudformation.CloudformationTarget, a, e, changes *VPCCIDRBlock) error {
|
|
cf := &cloudformationVPCCIDRBlock{
|
|
VPCID: e.VPC.CloudformationLink(),
|
|
CIDRBlock: e.CIDRBlock,
|
|
}
|
|
|
|
return t.RenderResource("AWS::EC2::VPCCidrBlock", *e.Name, cf)
|
|
}
|