add azure and s3 support

Author: pasha-codefresh

codefresh/context/storage.go

@@ -41,7 +41,7 @@ func FlattenStorageContextConfig(spec cfClient.ContextSpec) []interface{} {
 
 }
 
-func storageSchema() *schema.Schema {
+func storageSchema(authSchema *schema.Schema) *schema.Schema {
 	return &schema.Schema{
 		Type:     schema.TypeList,
 		Optional: true,
@@ -55,23 +55,7 @@ func storageSchema() *schema.Schema {
 					MaxItems: 1,
 					Elem: &schema.Resource{
 						Schema: map[string]*schema.Schema{
-							"auth": {
-								Type:     schema.TypeList,
-								Required: true,
-								MaxItems: 1,
-								Elem: &schema.Resource{
-									Schema: map[string]*schema.Schema{
-										"type": {
-											Type:     schema.TypeString,
-											Required: true,
-										},
-										"json_config": {
-											Type:     schema.TypeMap,
-											Required: true,
-										},
-									},
-								},
-							},
+							"auth": authSchema,
 						},
 					},
 				},
@@ -81,9 +65,68 @@ func storageSchema() *schema.Schema {
 }
 
 func GcsSchema() *schema.Schema {
-	return storageSchema()
+	sch := &schema.Schema{
+		Type:     schema.TypeList,
+		Required: true,
+		MaxItems: 1,
+		Elem: &schema.Resource{
+			Schema: map[string]*schema.Schema{
+				"type": {
+					Type:     schema.TypeString,
+					Required: true,
+				},
+				"json_config": {
+					Type:     schema.TypeMap,
+					Required: true,
+				},
+			},
+		},
+	}
+	return storageSchema(sch)
 }
 
 func S3Schema() *schema.Schema {
-	return storageSchema()
+	sch := &schema.Schema{
+		Type:     schema.TypeList,
+		Required: true,
+		MaxItems: 1,
+		Elem: &schema.Resource{
+			Schema: map[string]*schema.Schema{
+				"type": {
+					Type:     schema.TypeString,
+					Required: true,
+				},
+				"json_config": {
+					Type:     schema.TypeMap,
+					Required: true,
+				},
+			},
+		},
+	}
+	return storageSchema(sch)
+}
+
+func AzureStorage() *schema.Schema {
+	sch := &schema.Schema{
+		Type:     schema.TypeList,
+		Required: true,
+		MaxItems: 1,
+		Elem: &schema.Resource{
+			Schema: map[string]*schema.Schema{
+				"type": {
+					Type:     schema.TypeString,
+					Required: true,
+				},
+				"account_name": {
+					Type:     schema.TypeString,
+					Required: true,
+				},
+				"account_key": {
+					Type:     schema.TypeString,
+					Required: true,
+				},
+			},
+		},
+	}
+	return storageSchema(sch)
 }

codefresh/resource_context.go

@@ -16,6 +16,7 @@ const (
 	contextSecretYaml    = "secret-yaml"
 	contextGoogleStorage = "storage.gc"
 	contextS3Storage     = "storage.s3"
+	contextAzureStorage  = "storage.azuref"
 )
 
 var supportedContextType = []string{
@@ -140,6 +141,7 @@ func resourceContext() *schema.Resource {
 						},
 						normalizeFieldName(contextGoogleStorage): storageContext.GcsSchema(),
 						normalizeFieldName(contextS3Storage):     storageContext.S3Schema(),
+						normalizeFieldName(contextAzureStorage):  storageContext.AzureStorage(),
 					},
 				},
 			},
@@ -239,7 +241,7 @@ func flattenContextSpec(spec cfClient.ContextSpec) []interface{} {
 		m[normalizeFieldName(currentContextType)] = flattenContextConfig(spec)
 	case contextYaml, contextSecretYaml:
 		m[normalizeFieldName(currentContextType)] = flattenContextYaml(spec)
-	case contextGoogleStorage, contextS3Storage:
+	case contextGoogleStorage, contextS3Storage, contextAzureStorage:
 		m[normalizeFieldName(currentContextType)] = storageContext.FlattenStorageContextConfig(spec)
 	default:
 		log.Printf("[DEBUG] Invalid context type = %v", currentContextType)
@@ -293,6 +295,9 @@ func mapResourceToContext(d *schema.ResourceData) *cfClient.Context {
 	} else if data, ok := d.GetOk("spec.0." + normalizeFieldName(contextS3Storage) + ".0.data"); ok {
 		normalizedContextType = contextS3Storage
 		normalizedContextData = storageContext.ConvertStorageContext(data.([]interface{}))
+	} else if data, ok := d.GetOk("spec.0." + normalizeFieldName(contextAzureStorage) + ".0.data"); ok {
+		normalizedContextType = contextAzureStorage
+		normalizedContextData = storageContext.ConvertStorageContext(data.([]interface{}))
 	}
 
 	return &cfClient.Context{