/*
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 distributions

import (
	"k8s.io/klog/v2"
)

type Distribution string

const (
	DistributionDebian9      Distribution = "stretch"
	DistributionDebian10     Distribution = "buster"
	DistributionUbuntu1604   Distribution = "xenial"
	DistributionUbuntu1804   Distribution = "bionic"
	DistributionUbuntu2004   Distribution = "focal"
	DistributionAmazonLinux2 Distribution = "amazonlinux2"
	DistributionRhel7        Distribution = "rhel7"
	DistributionCentos7      Distribution = "centos7"
	DistributionRhel8        Distribution = "rhel8"
	DistributionCentos8      Distribution = "centos8"
	DistributionFlatcar      Distribution = "flatcar"
	DistributionContainerOS  Distribution = "containeros"
)

func (d Distribution) IsDebianFamily() bool {
	switch d {
	case DistributionDebian9, DistributionDebian10:
		return true
	case DistributionUbuntu1604, DistributionUbuntu1804, DistributionUbuntu2004:
		return true
	case DistributionCentos7, DistributionRhel7, DistributionCentos8, DistributionRhel8, DistributionAmazonLinux2:
		return false
	case DistributionFlatcar, DistributionContainerOS:
		return false
	default:
		klog.Fatalf("unknown distribution: %s", d)
		return false
	}
}

func (d Distribution) IsUbuntu() bool {
	switch d {
	case DistributionDebian9, DistributionDebian10:
		return false
	case DistributionUbuntu1604, DistributionUbuntu1804, DistributionUbuntu2004:
		return true
	case DistributionCentos7, DistributionRhel7, DistributionCentos8, DistributionRhel8, DistributionAmazonLinux2:
		return false
	case DistributionFlatcar, DistributionContainerOS:
		return false
	default:
		klog.Fatalf("unknown distribution: %s", d)
		return false
	}
}

func (d Distribution) IsRHELFamily() bool {
	switch d {
	case DistributionCentos7, DistributionRhel7, DistributionCentos8, DistributionRhel8, DistributionAmazonLinux2:
		return true
	case DistributionUbuntu1604, DistributionUbuntu1804, DistributionUbuntu2004, DistributionDebian9, DistributionDebian10:
		return false
	case DistributionFlatcar, DistributionContainerOS:
		return false
	default:
		klog.Fatalf("unknown distribution: %s", d)
		return false
	}
}

func (d Distribution) IsSystemd() bool {
	switch d {
	case DistributionUbuntu1604, DistributionUbuntu1804, DistributionUbuntu2004, DistributionDebian9, DistributionDebian10:
		return true
	case DistributionCentos7, DistributionRhel7, DistributionCentos8, DistributionRhel8, DistributionAmazonLinux2:
		return true
	case DistributionFlatcar:
		return true
	case DistributionContainerOS:
		return true
	default:
		klog.Fatalf("unknown distribution: %s", d)
		return false
	}
}