1// Copyright 2016 The etcd Authors
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15package v3rpc
16
17import (
18	"github.com/coreos/etcd/etcdserver"
19	pb "github.com/coreos/etcd/etcdserver/etcdserverpb"
20	"golang.org/x/net/context"
21)
22
23type AuthServer struct {
24	authenticator etcdserver.Authenticator
25}
26
27func NewAuthServer(s *etcdserver.EtcdServer) *AuthServer {
28	return &AuthServer{authenticator: s}
29}
30
31func (as *AuthServer) AuthEnable(ctx context.Context, r *pb.AuthEnableRequest) (*pb.AuthEnableResponse, error) {
32	resp, err := as.authenticator.AuthEnable(ctx, r)
33	if err != nil {
34		return nil, togRPCError(err)
35	}
36	return resp, nil
37}
38
39func (as *AuthServer) AuthDisable(ctx context.Context, r *pb.AuthDisableRequest) (*pb.AuthDisableResponse, error) {
40	resp, err := as.authenticator.AuthDisable(ctx, r)
41	if err != nil {
42		return nil, togRPCError(err)
43	}
44	return resp, nil
45}
46
47func (as *AuthServer) Authenticate(ctx context.Context, r *pb.AuthenticateRequest) (*pb.AuthenticateResponse, error) {
48	resp, err := as.authenticator.Authenticate(ctx, r)
49	if err != nil {
50		return nil, togRPCError(err)
51	}
52	return resp, nil
53}
54
55func (as *AuthServer) RoleAdd(ctx context.Context, r *pb.AuthRoleAddRequest) (*pb.AuthRoleAddResponse, error) {
56	resp, err := as.authenticator.RoleAdd(ctx, r)
57	if err != nil {
58		return nil, togRPCError(err)
59	}
60	return resp, nil
61}
62
63func (as *AuthServer) RoleDelete(ctx context.Context, r *pb.AuthRoleDeleteRequest) (*pb.AuthRoleDeleteResponse, error) {
64	resp, err := as.authenticator.RoleDelete(ctx, r)
65	if err != nil {
66		return nil, togRPCError(err)
67	}
68	return resp, nil
69}
70
71func (as *AuthServer) RoleGet(ctx context.Context, r *pb.AuthRoleGetRequest) (*pb.AuthRoleGetResponse, error) {
72	resp, err := as.authenticator.RoleGet(ctx, r)
73	if err != nil {
74		return nil, togRPCError(err)
75	}
76	return resp, nil
77}
78
79func (as *AuthServer) RoleList(ctx context.Context, r *pb.AuthRoleListRequest) (*pb.AuthRoleListResponse, error) {
80	resp, err := as.authenticator.RoleList(ctx, r)
81	if err != nil {
82		return nil, togRPCError(err)
83	}
84	return resp, nil
85}
86
87func (as *AuthServer) RoleRevokePermission(ctx context.Context, r *pb.AuthRoleRevokePermissionRequest) (*pb.AuthRoleRevokePermissionResponse, error) {
88	resp, err := as.authenticator.RoleRevokePermission(ctx, r)
89	if err != nil {
90		return nil, togRPCError(err)
91	}
92	return resp, nil
93}
94
95func (as *AuthServer) RoleGrantPermission(ctx context.Context, r *pb.AuthRoleGrantPermissionRequest) (*pb.AuthRoleGrantPermissionResponse, error) {
96	resp, err := as.authenticator.RoleGrantPermission(ctx, r)
97	if err != nil {
98		return nil, togRPCError(err)
99	}
100	return resp, nil
101}
102
103func (as *AuthServer) UserAdd(ctx context.Context, r *pb.AuthUserAddRequest) (*pb.AuthUserAddResponse, error) {
104	resp, err := as.authenticator.UserAdd(ctx, r)
105	if err != nil {
106		return nil, togRPCError(err)
107	}
108	return resp, nil
109}
110
111func (as *AuthServer) UserDelete(ctx context.Context, r *pb.AuthUserDeleteRequest) (*pb.AuthUserDeleteResponse, error) {
112	resp, err := as.authenticator.UserDelete(ctx, r)
113	if err != nil {
114		return nil, togRPCError(err)
115	}
116	return resp, nil
117}
118
119func (as *AuthServer) UserGet(ctx context.Context, r *pb.AuthUserGetRequest) (*pb.AuthUserGetResponse, error) {
120	resp, err := as.authenticator.UserGet(ctx, r)
121	if err != nil {
122		return nil, togRPCError(err)
123	}
124	return resp, nil
125}
126
127func (as *AuthServer) UserList(ctx context.Context, r *pb.AuthUserListRequest) (*pb.AuthUserListResponse, error) {
128	resp, err := as.authenticator.UserList(ctx, r)
129	if err != nil {
130		return nil, togRPCError(err)
131	}
132	return resp, nil
133}
134
135func (as *AuthServer) UserGrantRole(ctx context.Context, r *pb.AuthUserGrantRoleRequest) (*pb.AuthUserGrantRoleResponse, error) {
136	resp, err := as.authenticator.UserGrantRole(ctx, r)
137	if err != nil {
138		return nil, togRPCError(err)
139	}
140	return resp, nil
141}
142
143func (as *AuthServer) UserRevokeRole(ctx context.Context, r *pb.AuthUserRevokeRoleRequest) (*pb.AuthUserRevokeRoleResponse, error) {
144	resp, err := as.authenticator.UserRevokeRole(ctx, r)
145	if err != nil {
146		return nil, togRPCError(err)
147	}
148	return resp, nil
149}
150
151func (as *AuthServer) UserChangePassword(ctx context.Context, r *pb.AuthUserChangePasswordRequest) (*pb.AuthUserChangePasswordResponse, error) {
152	resp, err := as.authenticator.UserChangePassword(ctx, r)
153	if err != nil {
154		return nil, togRPCError(err)
155	}
156	return resp, nil
157}
158