forked from DrFaust92/terraform-provider-airflow
-
Notifications
You must be signed in to change notification settings - Fork 0
/
provider.go
112 lines (101 loc) · 3.13 KB
/
provider.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
package main
import (
"context"
"fmt"
"log"
"net/url"
"strings"
"github.com/apache/airflow-client-go/airflow"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
)
type ProviderConfig struct {
ApiClient *airflow.APIClient
AuthContext context.Context
}
func AirflowProvider() *schema.Provider {
return &schema.Provider{
Schema: map[string]*schema.Schema{
"base_endpoint": {
Type: schema.TypeString,
Required: true,
DefaultFunc: schema.EnvDefaultFunc("AIRFLOW_BASE_ENDPOINT", nil),
ValidateFunc: validation.IsURLWithHTTPorHTTPS,
},
"oauth2_token": {
Type: schema.TypeString,
Optional: true,
Sensitive: true,
Description: "The oauth to use for API authentication",
DefaultFunc: schema.EnvDefaultFunc("AIRFLOW_OAUTH2_TOKEN", nil),
ConflictsWith: []string{"username", "password"},
},
"username": {
Type: schema.TypeString,
DefaultFunc: schema.EnvDefaultFunc("AIRFLOW_API_USERNAME", nil),
Optional: true,
Description: "The username to use for API basic authentication",
RequiredWith: []string{"password"},
ConflictsWith: []string{"oauth2_token"},
},
"password": {
Type: schema.TypeString,
DefaultFunc: schema.EnvDefaultFunc("AIRFLOW_API_PASSWORD", nil),
Optional: true,
Sensitive: true,
Description: "The password to use for API basic authentication",
RequiredWith: []string{"username"},
ConflictsWith: []string{"oauth2_token"},
},
},
ResourcesMap: map[string]*schema.Resource{
"airflow_connection": resourceConnection(),
"airflow_dag": resourceDag(),
"airflow_dag_run": resourceDagRun(),
"airflow_variable": resourceVariable(),
"airflow_pool": resourcePool(),
"airflow_role": resourceRole(),
"airflow_user": resourceUser(),
},
ConfigureFunc: providerConfigure,
}
}
func providerConfigure(d *schema.ResourceData) (interface{}, error) {
endpoint := d.Get("base_endpoint").(string)
u, err := url.Parse(endpoint)
if err != nil {
return nil, fmt.Errorf("invalid base_endpoint: %w", err)
}
authCtx := context.Background()
if v, ok := d.GetOk("oauth2_token"); ok {
authCtx = context.WithValue(authCtx, airflow.ContextAccessToken, v)
}
if username, ok := d.GetOk("username"); ok {
var password interface{}
if password, ok = d.GetOk("password"); !ok {
return nil, fmt.Errorf("found username for basic auth, but password not specified")
}
log.Printf("[DEBUG] Using API Basic Auth")
cred := airflow.BasicAuth{
UserName: username.(string),
Password: password.(string),
}
authCtx = context.WithValue(authCtx, airflow.ContextBasicAuth, cred)
}
path := strings.TrimRight(u.Path, "/")
clientConf := &airflow.Configuration{
Scheme: u.Scheme,
Host: u.Host,
Debug: true,
Servers: airflow.ServerConfigurations{
{
URL: fmt.Sprint(path, "/api/v1"),
Description: "Apache Airflow Stable API.",
},
},
}
return ProviderConfig{
ApiClient: airflow.NewAPIClient(clientConf),
AuthContext: authCtx,
}, nil
}