checkingrequiredfiles.go

// Copyright 2018 Appliscale
//
// Maintainers and contributors are listed in README file inside repository.
//
// 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 checkingrequiredfiles checks if .aws/config, .aws/credentials main.yaml and other configuration files exist.
package checkingrequiredfiles

import (
	"bufio"
	"github.com/Appliscale/perun/cliparser"
	"github.com/Appliscale/perun/configuration"
	"github.com/Appliscale/perun/configurator"
	"github.com/Appliscale/perun/context"
	"github.com/Appliscale/perun/helpers"
	"github.com/Appliscale/perun/logger"
	"github.com/Appliscale/perun/myuser"
	"github.com/go-ini/ini"
	"io"
	"net/http"
	"os"
	"strings"
)

//CheckingRequiredFiles looks for required and default files and if doesn't find will create these.
func CheckingRequiredFiles(ctx *context.Context) {
	myLogger := logger.CreateDefaultLogger()
	homePath, pathError := myuser.GetUserHomeDir()
	if pathError != nil {
		myLogger.Error(pathError.Error())
	}
	mainYAMLexists, mainError := isMainYAMLPresent(&myLogger)
	if mainError != nil {
		myLogger.Error(mainError.Error())
	}

	downloadError := downloadDefaultFiles()
	if downloadError != nil {
		myLogger.Error(downloadError.Error())
	}
	//Checking if perun is running on EC2.
	_, isRunningOnEc2, _ := getRegion()
	if isRunningOnEc2 {
		if !mainYAMLexists {
			profile, region, err := workingOnEC2(&myLogger)
			if err == nil {
				*ctx = createEC2context(profile, homePath, region, ctx, &myLogger)
			} else {
				myLogger.Error(err.Error())
			}
		}
		downloadError := downloadDefaultFiles()
		if downloadError != nil {
			myLogger.Error(downloadError.Error())
		}
		return
	}
	//Checking if Mode is "offline" or "online".
	if isOffline() {
		*ctx = initOffline(mainYAMLexists, homePath, ctx, &myLogger)
		return
	}

	configAWSExists, configError := isAWSConfigPresent(&myLogger)
	if configError != nil {
		myLogger.Error(configError.Error())
	}

	profile := "default"
	region := "us-east-1"

	if !mainYAMLexists {
		profile, *ctx = configIsPresent(profile, homePath, ctx, &myLogger)
	} else {
		profile = ctx.Config.DefaultProfile
		region = ctx.Config.DefaultRegion
	}
	if configAWSExists {
		createCredentials(profile, homePath, ctx, &myLogger)
	} else { //configAWSExists == false
		var answer string
		myLogger.GetInput("Config doesn't exist, create default - "+profile+" *Y* or new *N*?", &answer)
		if strings.ToUpper(answer) == "N" {
			profile, region, *ctx = newConfigFile(profile, region, homePath, ctx, &myLogger)
			addProfileToCredentials(profile, homePath, ctx, &myLogger)
		}
		if strings.ToUpper(answer) == "Y" {
			configurator.CreateAWSConfigFile(&myLogger, profile, region)
			if !isProfileInCredentials(profile, homePath+"/.aws/credentials", &myLogger) {
				configurator.CreateAWSCredentialsFile(ctx, profile)
			}
			if !mainYAMLexists {
				*ctx = createNewMainYaml(profile, homePath, ctx, &myLogger)
			}
		}
		addNewProfileFromCredentialsToConfig(profile, homePath, ctx, &myLogger)
	}
}

// Checking if Mode is "online" - needs config and credentials files or "offline" - needs only main.yaml.
func isOffline() bool {
	args, _ := cliparser.ParseCliArguments(os.Args)
	offline := [3]string{cliparser.CreateParametersMode, cliparser.LintMode, cliparser.ConfigureMode}
	for _, off := range offline {
		if *args.Mode == off {
			return true
		}
	}
	return false
}

// Creating only main.yaml and context if Mode is "offline".
func initOffline(mainYAMLexists bool, homePath string, ctx *context.Context, myLogger logger.LoggerInt) context.Context {
	if !mainYAMLexists {
		var profile string
		var region string
		var answer string
		myLogger.GetInput("You haven't got main.yaml, create default *Y* or new *N*", &answer)
		if strings.ToUpper(answer) == "Y" {
			profile = "default"
			region = "us-east-1"
		} else if strings.ToUpper(answer) == "N" {
			myLogger.GetInput("Profile", &profile)
			myLogger.GetInput("Region", &region)
		}
		con := configurator.CreateMainYaml(myLogger, profile, region)
		configuration.SaveToFile(con, homePath+"/.config/perun/main.yaml", myLogger)
		var err error
		*ctx, err = context.GetContext(cliparser.ParseCliArguments, configuration.GetConfiguration, configuration.ReadInconsistencyConfiguration)
		if err != nil {
			myLogger.Error(err.Error())
		}
	}
	return *ctx
}

// Looking for main.yaml.
func isMainYAMLPresent(myLogger *logger.Logger) (bool, error) {
	homePath, pathError := myuser.GetUserHomeDir()
	if pathError != nil {
		myLogger.Error(pathError.Error())
		return false, pathError
	}
	_, mainError := os.Open(homePath + "/.config/perun/main.yaml")
	if mainError != nil {
		_, mainError = os.Open(homePath + "/etc/perun/main.yaml")
		if mainError != nil {
			return false, pathError
		}
		return true, pathError
	}
	return true, pathError
}

// Looking for .aws/config.
func isAWSConfigPresent(myLogger *logger.Logger) (bool, error) {
	homePath, pathError := myuser.GetUserHomeDir()
	if pathError != nil {
		myLogger.Error(pathError.Error())
		return false, pathError
	}
	_, credentialsError := os.Open(homePath + "/.aws/config")
	if credentialsError != nil {
		return false, credentialsError
	}
	return true, nil

}

// Looking for [profiles] in credentials or config and return all.
func getProfilesFromFile(path string) []string {
	credentials, credentialsError := os.Open(path)
	if credentialsError != nil {
		return []string{}
	}
	defer credentials.Close()
	profiles := make([]string, 0)
	scanner := bufio.NewScanner(credentials)
	for scanner.Scan() {
		if strings.Contains(scanner.Text(), "[") {
			profile := strings.TrimPrefix(scanner.Text(), "[")
			profile = strings.TrimSuffix(profile, "]")
			if strings.Contains(profile, "profile ") {
				profile = strings.TrimPrefix(profile, "profile ")
			}
			if strings.Contains(profile, "-long-term") {
				profile = strings.TrimSuffix(profile, "-long-term")
			}
			profiles = append(profiles, profile)
		}
	}
	return profiles
}

// Looking for user's profile in credentials or config.
func isProfileInCredentials(profile string, path string, mylogger logger.LoggerInt) bool {
	credentials, _ := os.Open(path)
	defer credentials.Close()
	scanner := bufio.NewScanner(credentials)
	for scanner.Scan() {
		if strings.Contains(scanner.Text(), "["+profile+"]") || strings.Contains(scanner.Text(), "["+profile+"-long-term]") {
			return true
		}
	}
	return false

}

// Looking for region for profile.
func findRegionForProfile(profile string, path string, mylogger logger.LoggerInt) string {
	configuration, loadError := ini.Load(path)
	if loadError != nil {
		mylogger.Error(loadError.Error())
	}
	section, sectionError := configuration.GetSection(profile)
	if sectionError != nil {
		section, sectionError = configuration.GetSection("profile " + profile)
		if sectionError != nil {
			mylogger.Error(sectionError.Error())
			return ""
		}
	}
	region := section.Key("region").Value()
	return region

}

// Getting profiles from credentials and config, if credentials has new profile, add to config.
func findNewProfileInCredentials(credentials []string, config []string) []string {
	profiles := make([]string, 0)
	for i, cred := range credentials {
		if strings.Contains(cred, "-long-term") {
			cred = strings.TrimSuffix(cred, "-long-term")
			credentials[i] = cred
		}
	}
	for _, cred := range credentials {
		isProfileHere := helpers.SliceContains(config, cred)
		if !isProfileHere {
			profiles = append(profiles, cred)
			return profiles
		}
	}
	return []string{}
}

// Downloading other files.
func downloadDefaultFiles() error {
	urls := make(map[string]string)
	urls["blocked.json"] = "https://s3.amazonaws.com/perun-default-file/blocked.json"
	urls["unblocked.json"] = "https://s3.amazonaws.com/perun-default-file/unblocked.json"
	urls["style.yaml"] = "https://s3.amazonaws.com/perun-default-file/style.yaml"
	urls["specification_inconsistency.yaml"] = "https://s3.amazonaws.com/perun-default-file/specification_inconsistency.yaml"

	for file, url := range urls {
		homePath, _ := myuser.GetUserHomeDir()
		homePath += "/.config/perun/"

		if strings.Contains(file, "blocked") {
			homePath += "stack-policies/"
		}

		_, err := os.Stat(homePath)
		if os.IsNotExist(err) {
			os.MkdirAll(homePath, 0755)
		}

		_, openError := os.Open(homePath + file) //checking if file exists
		if openError != nil {
			out, creatingFileError := os.Create(homePath + file)

			if creatingFileError != nil {
				return creatingFileError
			}
			defer out.Close()

			resp, httpGetError := http.Get(url)
			if httpGetError != nil {
				return httpGetError
			}
			defer resp.Body.Close()

			_, copyError := io.Copy(out, resp.Body)
			if copyError != nil {
				return copyError
			}
		}
	}
	return nil
}