Exemple #1
0
func putCredential(request *restful.Request, response *restful.Response) {
	ip := request.PathParameter("ip")

	credential := host.Credential{}
	err := request.ReadEntity(&credential)
	if err != nil {
		jsonMap := make(map[string]interface{})
		jsonMap["Error"] = "Read body failure"
		jsonMap["ErrorMessage"] = err.Error()
		jsonMap["ip"] = ip
		errorMessageByteSlice, _ := json.Marshal(jsonMap)
		log.Error(jsonMap)
		response.WriteErrorString(400, string(errorMessageByteSlice))
		return
	}

	if ip != credential.IP {
		jsonMap := make(map[string]interface{})
		jsonMap["Error"] = "Path parameter ip is different from ip in the body"
		jsonMap["path"] = ip
		jsonMap["body"] = credential.IP
		errorMessageByteSlice, _ := json.Marshal(jsonMap)
		log.Error(jsonMap)
		response.WriteErrorString(400, string(errorMessageByteSlice))
		return
	}

	oldCredential, _ := host.GetStorage().LoadCredential(credential.IP)
	if oldCredential == nil {
		jsonMap := make(map[string]interface{})
		jsonMap["Error"] = "The credential to update doesn't exist"
		jsonMap["ip"] = credential.IP
		errorMessageByteSlice, _ := json.Marshal(jsonMap)
		log.Error(jsonMap)
		response.WriteErrorString(404, string(errorMessageByteSlice))
		return
	}

	err = host.GetStorage().SaveCredential(&credential)
	if err != nil {
		jsonMap := make(map[string]interface{})
		jsonMap["Error"] = "Save credential failure"
		jsonMap["ErrorMessage"] = err.Error()
		jsonMap["credential"] = credential
		errorMessageByteSlice, _ := json.Marshal(jsonMap)
		log.Error(jsonMap)
		response.WriteErrorString(422, string(errorMessageByteSlice))
		return
	}
}
Exemple #2
0
func RemoveImageFromAllHost(imageRecordSlcie []ImageRecord) error {
	credentialSlice, err := host.GetStorage().LoadAllCredential()
	if err != nil {
		log.Error(err)
		return err
	}

	amount := len(imageRecordSlcie)

	commandSlice := make([]string, 0)
	// Delete all stopped instance so image could be removed
	commandSlice = append(commandSlice, "sudo docker rm $(sudo docker ps -aqf status=exited | xargs)\n")
	for _, imageRecord := range imageRecordSlcie {
		commandSlice = append(commandSlice, "sudo docker rmi "+imageRecord.Path+"\n")
	}

	hasError := false
	buffer := bytes.Buffer{}
	for _, credential := range credentialSlice {
		if credential.Disabled == false {
			interactiveMap := make(map[string]string)
			interactiveMap["[sudo]"] = credential.SSH.Password + "\n"

			resultSlice, err := sshclient.InteractiveSSH(
				2*time.Second,
				time.Duration(amount)*time.Minute,
				credential.IP,
				credential.SSH.Port,
				credential.SSH.User,
				credential.SSH.Password,
				commandSlice,
				interactiveMap)

			log.Info("Issue command via ssh with result:\n %v", resultSlice)

			if err != nil {
				hasError = true
				errorMessage := fmt.Sprintf("Error message: %v Result Output: %v .", err, resultSlice)
				log.Error(errorMessage)
				buffer.WriteString(errorMessage)
			}
		}
	}

	if hasError {
		return errors.New(buffer.String())
	} else {
		return nil
	}
}
Exemple #3
0
func getAllCredential(request *restful.Request, response *restful.Response) {
	credentialClusterSlice, err := host.GetStorage().LoadAllCredential()
	if err != nil {
		jsonMap := make(map[string]interface{})
		jsonMap["Error"] = "Get all credential failure"
		jsonMap["ErrorMessage"] = err.Error()
		errorMessageByteSlice, _ := json.Marshal(jsonMap)
		log.Error(jsonMap)
		response.WriteErrorString(422, string(errorMessageByteSlice))
		return
	}

	response.WriteJson(credentialClusterSlice, "[]Credential")
}
Exemple #4
0
func postCredential(request *restful.Request, response *restful.Response) {
	credential := host.Credential{}
	err := request.ReadEntity(&credential)
	if err != nil {
		jsonMap := make(map[string]interface{})
		jsonMap["Error"] = "Read body failure"
		jsonMap["ErrorMessage"] = err.Error()
		errorMessageByteSlice, _ := json.Marshal(jsonMap)
		log.Error(jsonMap)
		response.WriteErrorString(400, string(errorMessageByteSlice))
		return
	}

	oldCredential, _ := host.GetStorage().LoadCredential(credential.IP)
	if oldCredential != nil {
		jsonMap := make(map[string]interface{})
		jsonMap["Error"] = "The credential to create already exists"
		jsonMap["ip"] = credential.IP
		errorMessageByteSlice, _ := json.Marshal(jsonMap)
		log.Error(jsonMap)
		response.WriteErrorString(409, string(errorMessageByteSlice))
		return
	}

	err = host.GetStorage().SaveCredential(&credential)
	if err != nil {
		jsonMap := make(map[string]interface{})
		jsonMap["Error"] = "Save credential failure"
		jsonMap["ErrorMessage"] = err.Error()
		jsonMap["credential"] = credential
		errorMessageByteSlice, _ := json.Marshal(jsonMap)
		log.Error(jsonMap)
		response.WriteErrorString(422, string(errorMessageByteSlice))
		return
	}
}
Exemple #5
0
func deleteCredential(request *restful.Request, response *restful.Response) {
	ip := request.PathParameter("ip")

	err := host.GetStorage().DeleteCredential(ip)
	if err != nil {
		jsonMap := make(map[string]interface{})
		jsonMap["Error"] = "Delete credential failure"
		jsonMap["ErrorMessage"] = err.Error()
		jsonMap["ip"] = ip
		errorMessageByteSlice, _ := json.Marshal(jsonMap)
		log.Error(jsonMap)
		response.WriteErrorString(422, string(errorMessageByteSlice))
		return
	}
}