-
Notifications
You must be signed in to change notification settings - Fork 9
/
resource.go
87 lines (71 loc) · 2.68 KB
/
resource.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
package sqlUser
import (
"context"
"github.com/PGSSoft/terraform-provider-mssql/internal/core/resource"
common2 "github.com/PGSSoft/terraform-provider-mssql/internal/services/common"
"strconv"
"strings"
"github.com/PGSSoft/terraform-provider-mssql/internal/sql"
"github.com/PGSSoft/terraform-provider-mssql/internal/utils"
"github.com/hashicorp/terraform-plugin-framework/tfsdk"
)
type res struct{}
func (r *res) GetName() string {
return "sql_user"
}
func (r *res) GetSchema(context.Context) tfsdk.Schema {
return tfsdk.Schema{
Description: "Manages database-level user, based on SQL login.",
Attributes: map[string]tfsdk.Attribute{
"id": common2.ToResourceId(attributes["id"]),
"name": common2.ToRequired(attributes["name"]),
"database_id": common2.DatabaseIdResourceAttribute,
"login_id": common2.ToRequired(attributes["login_id"]),
},
}
}
func (r *res) Create(ctx context.Context, req resource.CreateRequest[resourceData], resp *resource.CreateResponse[resourceData]) {
var (
db sql.Database
user sql.User
)
req.
Then(func() { db = common2.GetResourceDb(ctx, req.Conn, req.Plan.DatabaseId.ValueString()) }).
Then(func() { user = sql.CreateUser(ctx, db, req.Plan.toSettings()) }).
Then(func() { resp.State = req.Plan.withIds(db.GetId(ctx), user.GetId(ctx)) })
}
func (r *res) Read(ctx context.Context, req resource.ReadRequest[resourceData], resp *resource.ReadResponse[resourceData]) {
var user sql.User
req.
Then(func() { user = getUser(ctx, req.Conn, req.State) }).
Then(func() {
state := req.State.withIds(user.GetDatabaseId(ctx), user.GetId(ctx))
resp.SetState(state.withSettings(user.GetSettings(ctx)))
})
}
func (r *res) Update(ctx context.Context, req resource.UpdateRequest[resourceData], resp *resource.UpdateResponse[resourceData]) {
var user sql.User
req.
Then(func() { user = getUser(ctx, req.Conn, req.Plan) }).
Then(func() { user.UpdateSettings(ctx, req.Plan.toSettings()) }).
Then(func() { resp.State = req.Plan.withSettings(user.GetSettings(ctx)) })
}
func (r *res) Delete(ctx context.Context, req resource.DeleteRequest[resourceData], resp *resource.DeleteResponse[resourceData]) {
var user sql.User
req.
Then(func() { user = getUser(ctx, req.Conn, req.State) }).
Then(func() { user.Drop(ctx) })
}
func getUser(ctx context.Context, conn sql.Connection, data resourceData) sql.User {
idSegments := strings.Split(data.Id.ValueString(), "/")
id, err := strconv.Atoi(idSegments[1])
if err != nil {
utils.AddError(ctx, "Error converting user ID", err)
return nil
}
db := common2.GetResourceDb(ctx, conn, idSegments[0])
if utils.HasError(ctx) {
return nil
}
return sql.GetUser(ctx, db, sql.UserId(id))
}