mirror of
https://github.com/Kichiyaki/terraform-provider-woodpecker.git
synced 2024-06-29 07:38:05 +00:00
123 lines
3.5 KiB
Go
123 lines
3.5 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/hashicorp/terraform-plugin-framework/datasource"
|
|
"github.com/hashicorp/terraform-plugin-framework/datasource/schema"
|
|
"github.com/hashicorp/terraform-plugin-framework/types"
|
|
"github.com/woodpecker-ci/woodpecker/woodpecker-go/woodpecker"
|
|
)
|
|
|
|
type userDataSource struct {
|
|
client woodpecker.Client
|
|
}
|
|
|
|
var _ datasource.DataSource = (*userDataSource)(nil)
|
|
|
|
func newUserDataSource() datasource.DataSource {
|
|
return &userDataSource{}
|
|
}
|
|
|
|
type userDataSourceModel struct {
|
|
ID types.Int64 `tfsdk:"id"`
|
|
Login types.String `tfsdk:"login"`
|
|
Email types.String `tfsdk:"email"`
|
|
Avatar types.String `tfsdk:"avatar"`
|
|
Active types.Bool `tfsdk:"active"`
|
|
Admin types.Bool `tfsdk:"admin"`
|
|
}
|
|
|
|
func (m *userDataSourceModel) setValues(user *woodpecker.User) {
|
|
m.ID = types.Int64Value(user.ID)
|
|
m.Login = types.StringValue(user.Login)
|
|
m.Email = types.StringValue(user.Email)
|
|
m.Avatar = types.StringValue(user.Avatar)
|
|
m.Active = types.BoolValue(user.Active)
|
|
m.Admin = types.BoolValue(user.Admin)
|
|
}
|
|
|
|
func (d *userDataSource) Metadata(_ context.Context, req datasource.MetadataRequest, resp *datasource.MetadataResponse) {
|
|
resp.TypeName = req.ProviderTypeName + "_user"
|
|
}
|
|
|
|
func (d *userDataSource) Schema(_ context.Context, _ datasource.SchemaRequest, resp *datasource.SchemaResponse) {
|
|
resp.Schema = schema.Schema{
|
|
// This description is used by the documentation generator and the language server.
|
|
MarkdownDescription: "Use this data source to retrieve information about a user.",
|
|
|
|
Attributes: map[string]schema.Attribute{
|
|
"id": schema.Int64Attribute{
|
|
Computed: true,
|
|
Description: "the user's id",
|
|
},
|
|
"login": schema.StringAttribute{
|
|
Required: true,
|
|
Description: "The user's login. Use an empty string \"\" to retrieve information about the currently authenticated user.",
|
|
},
|
|
"email": schema.StringAttribute{
|
|
Computed: true,
|
|
Description: "the user's email",
|
|
},
|
|
"avatar": schema.StringAttribute{
|
|
Computed: true,
|
|
Description: "the user's avatar URL",
|
|
},
|
|
"active": schema.BoolAttribute{
|
|
Computed: true,
|
|
Description: "Whether user is active in the system",
|
|
},
|
|
"admin": schema.BoolAttribute{
|
|
Computed: true,
|
|
Description: "Whether user is an admin",
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (d *userDataSource) Configure(ctx context.Context, req datasource.ConfigureRequest, resp *datasource.ConfigureResponse) {
|
|
// Prevent panic if the provider has not been configured.
|
|
if req.ProviderData == nil {
|
|
return
|
|
}
|
|
|
|
client, ok := req.ProviderData.(woodpecker.Client)
|
|
if !ok {
|
|
resp.Diagnostics.AddError(
|
|
"Unexpected Data Source Configure Type",
|
|
fmt.Sprintf("Expected woodpecker.Client, got: %T. Please report this issue to the provider developers.", req.ProviderData),
|
|
)
|
|
return
|
|
}
|
|
|
|
d.client = client
|
|
}
|
|
|
|
func (d *userDataSource) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
|
|
var data userDataSourceModel
|
|
|
|
// Read Terraform configuration data into the model
|
|
resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
|
|
if resp.Diagnostics.HasError() {
|
|
return
|
|
}
|
|
|
|
var user *woodpecker.User
|
|
var err error
|
|
if login := data.Login.ValueString(); login != "" {
|
|
user, err = d.client.User(login)
|
|
} else {
|
|
user, err = d.client.Self()
|
|
}
|
|
if err != nil {
|
|
resp.Diagnostics.AddError("Couldn't read user data", err.Error())
|
|
return
|
|
}
|
|
|
|
data.setValues(user)
|
|
|
|
// Save data into Terraform state
|
|
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
|
|
}
|