feat: add permission
This commit is contained in:
王性驊
parent
2aa8bd061d
commit
31ab87aadc
|
|
@ -0,0 +1,14 @@
|
||||||
|
[request_definition]
|
||||||
|
r = sub, obj, act
|
||||||
|
|
||||||
|
[policy_definition]
|
||||||
|
p = sub, obj, act
|
||||||
|
|
||||||
|
[role_definition]
|
||||||
|
g = _, _
|
||||||
|
|
||||||
|
[policy_effect]
|
||||||
|
e = some(where (p.eft == allow))
|
||||||
|
|
||||||
|
[matchers]
|
||||||
|
m = g(r.sub, p.sub) && keyMatch2(r.obj, p.obj) && regexMatch(r.act, p.act)
|
||||||
5
go.mod
5
go.mod
|
|
@ -10,7 +10,9 @@ require (
|
||||||
github.com/aws/aws-sdk-go-v2 v1.39.2
|
github.com/aws/aws-sdk-go-v2 v1.39.2
|
||||||
github.com/aws/aws-sdk-go-v2/credentials v1.18.16
|
github.com/aws/aws-sdk-go-v2/credentials v1.18.16
|
||||||
github.com/aws/aws-sdk-go-v2/service/ses v1.34.5
|
github.com/aws/aws-sdk-go-v2/service/ses v1.34.5
|
||||||
|
github.com/casbin/casbin/v2 v2.127.0
|
||||||
github.com/go-playground/validator/v10 v10.27.0
|
github.com/go-playground/validator/v10 v10.27.0
|
||||||
|
github.com/golang-jwt/jwt/v5 v5.3.0
|
||||||
github.com/matcornic/hermes/v2 v2.1.0
|
github.com/matcornic/hermes/v2 v2.1.0
|
||||||
github.com/minchao/go-mitake v1.0.0
|
github.com/minchao/go-mitake v1.0.0
|
||||||
github.com/panjf2000/ants/v2 v2.11.3
|
github.com/panjf2000/ants/v2 v2.11.3
|
||||||
|
|
@ -40,6 +42,8 @@ require (
|
||||||
github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.7.9 // indirect
|
github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.7.9 // indirect
|
||||||
github.com/aws/smithy-go v1.23.0 // indirect
|
github.com/aws/smithy-go v1.23.0 // indirect
|
||||||
github.com/beorn7/perks v1.0.1 // indirect
|
github.com/beorn7/perks v1.0.1 // indirect
|
||||||
|
github.com/bmatcuk/doublestar/v4 v4.6.1 // indirect
|
||||||
|
github.com/casbin/govaluate v1.3.0 // indirect
|
||||||
github.com/cenkalti/backoff/v4 v4.3.0 // indirect
|
github.com/cenkalti/backoff/v4 v4.3.0 // indirect
|
||||||
github.com/cespare/xxhash/v2 v2.3.0 // indirect
|
github.com/cespare/xxhash/v2 v2.3.0 // indirect
|
||||||
github.com/containerd/errdefs v1.0.0 // indirect
|
github.com/containerd/errdefs v1.0.0 // indirect
|
||||||
|
|
@ -104,6 +108,7 @@ require (
|
||||||
github.com/redis/go-redis/v9 v9.15.0 // indirect
|
github.com/redis/go-redis/v9 v9.15.0 // indirect
|
||||||
github.com/rivo/uniseg v0.2.0 // indirect
|
github.com/rivo/uniseg v0.2.0 // indirect
|
||||||
github.com/russross/blackfriday/v2 v2.0.1 // indirect
|
github.com/russross/blackfriday/v2 v2.0.1 // indirect
|
||||||
|
github.com/shirou/gopsutil/v3 v3.24.5 // indirect
|
||||||
github.com/shirou/gopsutil/v4 v4.25.6 // indirect
|
github.com/shirou/gopsutil/v4 v4.25.6 // indirect
|
||||||
github.com/shurcooL/sanitized_anchor_name v1.0.0 // indirect
|
github.com/shurcooL/sanitized_anchor_name v1.0.0 // indirect
|
||||||
github.com/sirupsen/logrus v1.9.3 // indirect
|
github.com/sirupsen/logrus v1.9.3 // indirect
|
||||||
|
|
|
||||||
19
go.sum
19
go.sum
|
|
@ -34,10 +34,16 @@ github.com/aws/smithy-go v1.23.0 h1:8n6I3gXzWJB2DxBDnfxgBaSX6oe0d/t10qGz7OKqMCE=
|
||||||
github.com/aws/smithy-go v1.23.0/go.mod h1:t1ufH5HMublsJYulve2RKmHDC15xu1f26kHCp/HgceI=
|
github.com/aws/smithy-go v1.23.0/go.mod h1:t1ufH5HMublsJYulve2RKmHDC15xu1f26kHCp/HgceI=
|
||||||
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
||||||
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
||||||
|
github.com/bmatcuk/doublestar/v4 v4.6.1 h1:FH9SifrbvJhnlQpztAx++wlkk70QBf0iBWDwNy7PA4I=
|
||||||
|
github.com/bmatcuk/doublestar/v4 v4.6.1/go.mod h1:xBQ8jztBU6kakFMg+8WGxn0c6z1fTSPVIjEY1Wr7jzc=
|
||||||
github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs=
|
github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs=
|
||||||
github.com/bsm/ginkgo/v2 v2.12.0/go.mod h1:SwYbGRRDovPVboqFv0tPTcG1sN61LM1Z4ARdbAV9g4c=
|
github.com/bsm/ginkgo/v2 v2.12.0/go.mod h1:SwYbGRRDovPVboqFv0tPTcG1sN61LM1Z4ARdbAV9g4c=
|
||||||
github.com/bsm/gomega v1.27.10 h1:yeMWxP2pV2fG3FgAODIY8EiRE3dy0aeFYt4l7wh6yKA=
|
github.com/bsm/gomega v1.27.10 h1:yeMWxP2pV2fG3FgAODIY8EiRE3dy0aeFYt4l7wh6yKA=
|
||||||
github.com/bsm/gomega v1.27.10/go.mod h1:JyEr/xRbxbtgWNi8tIEVPUYZ5Dzef52k01W3YH0H+O0=
|
github.com/bsm/gomega v1.27.10/go.mod h1:JyEr/xRbxbtgWNi8tIEVPUYZ5Dzef52k01W3YH0H+O0=
|
||||||
|
github.com/casbin/casbin/v2 v2.127.0 h1:UGK3uO/8cOslnNqFUJ4xzm/bh+N+o45U7cSolaFk38c=
|
||||||
|
github.com/casbin/casbin/v2 v2.127.0/go.mod h1:n4uZK8+tCMvcD6EVQZI90zKAok8iHAvEypcMJVKhGF0=
|
||||||
|
github.com/casbin/govaluate v1.3.0 h1:VA0eSY0M2lA86dYd5kPPuNZMUD9QkWnOCnavGrw9myc=
|
||||||
|
github.com/casbin/govaluate v1.3.0/go.mod h1:G/UnbIjZk/0uMNaLwZZmFQrR72tYRZWQkO70si/iR7A=
|
||||||
github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8=
|
github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8=
|
||||||
github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
|
github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
|
||||||
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
|
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
|
||||||
|
|
@ -95,11 +101,17 @@ github.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=
|
||||||
github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=
|
github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=
|
||||||
github.com/golang-jwt/jwt/v4 v4.5.2 h1:YtQM7lnr8iZ+j5q71MGKkNw9Mn7AjHM68uc9g5fXeUI=
|
github.com/golang-jwt/jwt/v4 v4.5.2 h1:YtQM7lnr8iZ+j5q71MGKkNw9Mn7AjHM68uc9g5fXeUI=
|
||||||
github.com/golang-jwt/jwt/v4 v4.5.2/go.mod h1:m21LjoU+eqJr34lmDMbreY2eSTRJ1cv77w39/MY0Ch0=
|
github.com/golang-jwt/jwt/v4 v4.5.2/go.mod h1:m21LjoU+eqJr34lmDMbreY2eSTRJ1cv77w39/MY0Ch0=
|
||||||
|
github.com/golang-jwt/jwt/v5 v5.3.0 h1:pv4AsKCKKZuqlgs5sUmn4x8UlGa0kEVt/puTpKx9vvo=
|
||||||
|
github.com/golang-jwt/jwt/v5 v5.3.0/go.mod h1:fxCRLWMO43lRc8nhHWY6LGqRcf+1gQWArsqaEUEa5bE=
|
||||||
|
github.com/golang/mock v1.4.4 h1:l75CXGRSwbaYNpl/Z2X1XIIAMSCquvXgpVZDhwEIJsc=
|
||||||
|
github.com/golang/mock v1.4.4/go.mod h1:l3mdAwkq5BuhzHwde/uurv3sEJeZMXNpwsxVWU71h+4=
|
||||||
github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
|
github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
|
||||||
github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
|
github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
|
||||||
github.com/golang/snappy v1.0.0 h1:Oy607GVXHs7RtbggtPBnr2RmDArIsAefDwvrdWvRhGs=
|
github.com/golang/snappy v1.0.0 h1:Oy607GVXHs7RtbggtPBnr2RmDArIsAefDwvrdWvRhGs=
|
||||||
github.com/golang/snappy v1.0.0/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
github.com/golang/snappy v1.0.0/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||||
github.com/google/go-cmp v0.5.6/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
github.com/google/go-cmp v0.5.6/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
|
github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
|
||||||
|
github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
|
||||||
github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
|
github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
|
||||||
github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
|
github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
|
||||||
github.com/google/uuid v1.0.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
github.com/google/uuid v1.0.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
|
|
@ -208,8 +220,12 @@ github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR
|
||||||
github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o=
|
github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o=
|
||||||
github.com/russross/blackfriday/v2 v2.0.1 h1:lPqVAte+HuHNfhJ/0LC98ESWRz8afy9tM/0RK8m9o+Q=
|
github.com/russross/blackfriday/v2 v2.0.1 h1:lPqVAte+HuHNfhJ/0LC98ESWRz8afy9tM/0RK8m9o+Q=
|
||||||
github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
||||||
|
github.com/shirou/gopsutil/v3 v3.24.5 h1:i0t8kL+kQTvpAYToeuiVk3TgDeKOFioZO3Ztz/iZ9pI=
|
||||||
|
github.com/shirou/gopsutil/v3 v3.24.5/go.mod h1:bsoOS1aStSs9ErQ1WWfxllSeS1K5D+U30r2NfcubMVk=
|
||||||
github.com/shirou/gopsutil/v4 v4.25.6 h1:kLysI2JsKorfaFPcYmcJqbzROzsBWEOAtw6A7dIfqXs=
|
github.com/shirou/gopsutil/v4 v4.25.6 h1:kLysI2JsKorfaFPcYmcJqbzROzsBWEOAtw6A7dIfqXs=
|
||||||
github.com/shirou/gopsutil/v4 v4.25.6/go.mod h1:PfybzyydfZcN+JMMjkF6Zb8Mq1A/VcogFFg7hj50W9c=
|
github.com/shirou/gopsutil/v4 v4.25.6/go.mod h1:PfybzyydfZcN+JMMjkF6Zb8Mq1A/VcogFFg7hj50W9c=
|
||||||
|
github.com/shoenig/go-m1cpu v0.1.6/go.mod h1:1JJMcUBvfNwpq05QDQVAnx3gUHr9IYF7GNg9SUEw2VQ=
|
||||||
|
github.com/shoenig/test v0.6.4/go.mod h1:byHiCGXqrVaflBLAMq/srcZIHynQPQgeyvkvXnjqq0k=
|
||||||
github.com/shopspring/decimal v1.4.0 h1:bxl37RwXBklmTi0C79JfXCEBD1cqqHt0bbgBAGFp81k=
|
github.com/shopspring/decimal v1.4.0 h1:bxl37RwXBklmTi0C79JfXCEBD1cqqHt0bbgBAGFp81k=
|
||||||
github.com/shopspring/decimal v1.4.0/go.mod h1:gawqmDU56v4yIKSwfBSFip1HdCCXN8/+DMd9qYNcwME=
|
github.com/shopspring/decimal v1.4.0/go.mod h1:gawqmDU56v4yIKSwfBSFip1HdCCXN8/+DMd9qYNcwME=
|
||||||
github.com/shurcooL/sanitized_anchor_name v1.0.0 h1:PdmoCO6wvbs+7yrJyMORt4/BmY5IYyJwS/kOiWx8mHo=
|
github.com/shurcooL/sanitized_anchor_name v1.0.0 h1:PdmoCO6wvbs+7yrJyMORt4/BmY5IYyJwS/kOiWx8mHo=
|
||||||
|
|
@ -310,6 +326,7 @@ golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
|
||||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
||||||
golang.org/x/net v0.0.0-20180218175443-cbe0f9307d01/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
golang.org/x/net v0.0.0-20180218175443-cbe0f9307d01/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
golang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
golang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
|
|
@ -340,6 +357,7 @@ golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBc
|
||||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||||
golang.org/x/sys v0.36.0 h1:KVRy2GtZBrk1cBYA7MKu5bEZFxQk4NIDV6RLVcC8o0k=
|
golang.org/x/sys v0.36.0 h1:KVRy2GtZBrk1cBYA7MKu5bEZFxQk4NIDV6RLVcC8o0k=
|
||||||
golang.org/x/sys v0.36.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
golang.org/x/sys v0.36.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||||
|
|
@ -356,6 +374,7 @@ golang.org/x/text v0.29.0/go.mod h1:7MhJOA9CD2qZyOKYazxdYMF85OwPdEr9jTtBpO7ydH4=
|
||||||
golang.org/x/time v0.10.0 h1:3usCWA8tQn0L8+hFJQNgzpWbd89begxN66o1Ojdn5L4=
|
golang.org/x/time v0.10.0 h1:3usCWA8tQn0L8+hFJQNgzpWbd89begxN66o1Ojdn5L4=
|
||||||
golang.org/x/time v0.10.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
|
golang.org/x/time v0.10.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
|
||||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/tools v0.0.0-20190425150028-36563e24a262/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
|
||||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=
|
golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=
|
||||||
golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=
|
golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,194 @@
|
||||||
|
package svc
|
||||||
|
|
||||||
|
//func NewPermissionUC(c *config.Config, rds *redis.Redis) usecase.PermissionUseCase {
|
||||||
|
// // 準備Mongo Config (重用現有配置)
|
||||||
|
// conf := &mgo.Conf{
|
||||||
|
// Schema: c.Mongo.Schema,
|
||||||
|
// Host: c.Mongo.Host,
|
||||||
|
// Database: c.Mongo.Database,
|
||||||
|
// MaxStaleness: c.Mongo.MaxStaleness,
|
||||||
|
// MaxPoolSize: c.Mongo.MaxPoolSize,
|
||||||
|
// MinPoolSize: c.Mongo.MinPoolSize,
|
||||||
|
// MaxConnIdleTime: c.Mongo.MaxConnIdleTime,
|
||||||
|
// Compressors: c.Mongo.Compressors,
|
||||||
|
// EnableStandardReadWriteSplitMode: c.Mongo.EnableStandardReadWriteSplitMode,
|
||||||
|
// ConnectTimeoutMs: c.Mongo.ConnectTimeoutMs,
|
||||||
|
// }
|
||||||
|
// if c.Mongo.User != "" {
|
||||||
|
// conf.User = c.Mongo.User
|
||||||
|
// conf.Password = c.Mongo.Password
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// // 快取選項
|
||||||
|
// cacheOpts := []cache.Option{
|
||||||
|
// cache.WithExpiry(c.CacheExpireTime),
|
||||||
|
// cache.WithNotFoundExpiry(c.CacheWithNotFoundExpiry),
|
||||||
|
// }
|
||||||
|
// dbOpts := []mon.Option{
|
||||||
|
// mgo.SetCustomDecimalType(),
|
||||||
|
// mgo.InitMongoOptions(*conf),
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// // 初始化 Casbin Adapter
|
||||||
|
// casbinAdapter := repository.NewCasbinAdapter(repository.CasbinAdapterParam{
|
||||||
|
// Conf: conf,
|
||||||
|
// CacheConf: c.Cache,
|
||||||
|
// CacheOpts: cacheOpts,
|
||||||
|
// DBOpts: dbOpts,
|
||||||
|
// })
|
||||||
|
//
|
||||||
|
// // 初始化 Casbin Enforcer
|
||||||
|
// modelPath := "pkg/permission/config/rbac_model.conf"
|
||||||
|
// enforcer, err := casbin.NewEnforcer(modelPath, casbinAdapter)
|
||||||
|
// if err != nil {
|
||||||
|
// panic("Failed to create casbin enforcer: " + err.Error())
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// // 啟用自動保存
|
||||||
|
// enforcer.EnableAutoSave(true)
|
||||||
|
//
|
||||||
|
// // 載入策略
|
||||||
|
// err = enforcer.LoadPolicy()
|
||||||
|
// if err != nil {
|
||||||
|
// panic("Failed to load casbin policy: " + err.Error())
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// // 初始化其他 Repository
|
||||||
|
// permissionRepo := repository.NewPermissionRepository(repository.PermissionRepositoryParam{
|
||||||
|
// Conf: conf,
|
||||||
|
// CacheConf: c.Cache,
|
||||||
|
// CacheOpts: cacheOpts,
|
||||||
|
// DBOpts: dbOpts,
|
||||||
|
// })
|
||||||
|
//
|
||||||
|
// roleRepo := repository.NewRoleRepository(repository.RoleRepositoryParam{
|
||||||
|
// Conf: conf,
|
||||||
|
// CacheConf: c.Cache,
|
||||||
|
// CacheOpts: cacheOpts,
|
||||||
|
// DBOpts: dbOpts,
|
||||||
|
// })
|
||||||
|
//
|
||||||
|
// userRoleRepo := repository.NewUserRoleRepository(repository.UserRoleRepositoryParam{
|
||||||
|
// Conf: conf,
|
||||||
|
// CacheConf: c.Cache,
|
||||||
|
// CacheOpts: cacheOpts,
|
||||||
|
// DBOpts: dbOpts,
|
||||||
|
// })
|
||||||
|
//
|
||||||
|
// // 創建索引
|
||||||
|
// _, _ = permissionRepo.Index20241226001UP(context.Background())
|
||||||
|
// _, _ = roleRepo.Index20241226001UP(context.Background())
|
||||||
|
// _, _ = userRoleRepo.Index20241226001UP(context.Background())
|
||||||
|
//
|
||||||
|
// return uc.MustPermissionUseCase(uc.PermissionUseCaseParam{
|
||||||
|
// Enforcer: enforcer,
|
||||||
|
// PermissionRepo: permissionRepo,
|
||||||
|
// RoleRepo: roleRepo,
|
||||||
|
// UserRoleRepo: userRoleRepo,
|
||||||
|
// })
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
//func NewAuthUC(c *config.Config, rds *redis.Redis) usecase.AuthUseCase {
|
||||||
|
// // 準備Mongo Config
|
||||||
|
// conf := &mgo.Conf{
|
||||||
|
// Schema: c.Mongo.Schema,
|
||||||
|
// Host: c.Mongo.Host,
|
||||||
|
// Database: c.Mongo.Database,
|
||||||
|
// MaxStaleness: c.Mongo.MaxStaleness,
|
||||||
|
// MaxPoolSize: c.Mongo.MaxPoolSize,
|
||||||
|
// MinPoolSize: c.Mongo.MinPoolSize,
|
||||||
|
// MaxConnIdleTime: c.Mongo.MaxConnIdleTime,
|
||||||
|
// Compressors: c.Mongo.Compressors,
|
||||||
|
// EnableStandardReadWriteSplitMode: c.Mongo.EnableStandardReadWriteSplitMode,
|
||||||
|
// ConnectTimeoutMs: c.Mongo.ConnectTimeoutMs,
|
||||||
|
// }
|
||||||
|
// if c.Mongo.User != "" {
|
||||||
|
// conf.User = c.Mongo.User
|
||||||
|
// conf.Password = c.Mongo.Password
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// // 快取選項
|
||||||
|
// cacheOpts := []cache.Option{
|
||||||
|
// cache.WithExpiry(c.CacheExpireTime),
|
||||||
|
// cache.WithNotFoundExpiry(c.CacheWithNotFoundExpiry),
|
||||||
|
// }
|
||||||
|
// dbOpts := []mon.Option{
|
||||||
|
// mgo.SetCustomDecimalType(),
|
||||||
|
// mgo.InitMongoOptions(*conf),
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// // 初始化 Repository
|
||||||
|
// clientRepo := repository.NewClientRepository(repository.ClientRepositoryParam{
|
||||||
|
// Conf: conf,
|
||||||
|
// CacheConf: c.Cache,
|
||||||
|
// CacheOpts: cacheOpts,
|
||||||
|
// DBOpts: dbOpts,
|
||||||
|
// })
|
||||||
|
//
|
||||||
|
// tokenRepo := repository.NewTokenRepository(repository.TokenRepositoryParam{
|
||||||
|
// Redis: rds,
|
||||||
|
// })
|
||||||
|
//
|
||||||
|
// // JWT 配置
|
||||||
|
// jwtConfig := permissionConfig.JWTConfig{
|
||||||
|
// Secret: c.JWTAuth.AccessSecret, // 使用現有的JWT配置
|
||||||
|
// AccessExpires: c.JWTAuth.AccessExpire,
|
||||||
|
// RefreshExpires: c.JWTAuth.AccessExpire * 7, // refresh token 較長
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// return uc.MustAuthUseCase(uc.AuthUseCaseParam{
|
||||||
|
// ClientRepo: clientRepo,
|
||||||
|
// TokenRepo: tokenRepo,
|
||||||
|
// JWTConfig: jwtConfig,
|
||||||
|
// })
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
//func NewRoleUC(c *config.Config) usecase.RoleUseCase {
|
||||||
|
// // 準備Mongo Config
|
||||||
|
// conf := &mgo.Conf{
|
||||||
|
// Schema: c.Mongo.Schema,
|
||||||
|
// Host: c.Mongo.Host,
|
||||||
|
// Database: c.Mongo.Database,
|
||||||
|
// MaxStaleness: c.Mongo.MaxStaleness,
|
||||||
|
// MaxPoolSize: c.Mongo.MaxPoolSize,
|
||||||
|
// MinPoolSize: c.Mongo.MinPoolSize,
|
||||||
|
// MaxConnIdleTime: c.Mongo.MaxConnIdleTime,
|
||||||
|
// Compressors: c.Mongo.Compressors,
|
||||||
|
// EnableStandardReadWriteSplitMode: c.Mongo.EnableStandardReadWriteSplitMode,
|
||||||
|
// ConnectTimeoutMs: c.Mongo.ConnectTimeoutMs,
|
||||||
|
// }
|
||||||
|
// if c.Mongo.User != "" {
|
||||||
|
// conf.User = c.Mongo.User
|
||||||
|
// conf.Password = c.Mongo.Password
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// // 快取選項
|
||||||
|
// cacheOpts := []cache.Option{
|
||||||
|
// cache.WithExpiry(c.CacheExpireTime),
|
||||||
|
// cache.WithNotFoundExpiry(c.CacheWithNotFoundExpiry),
|
||||||
|
// }
|
||||||
|
// dbOpts := []mon.Option{
|
||||||
|
// mgo.SetCustomDecimalType(),
|
||||||
|
// mgo.InitMongoOptions(*conf),
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// // 初始化 Repository
|
||||||
|
// roleRepo := repository.NewRoleRepository(repository.RoleRepositoryParam{
|
||||||
|
// Conf: conf,
|
||||||
|
// CacheConf: c.Cache,
|
||||||
|
// CacheOpts: cacheOpts,
|
||||||
|
// DBOpts: dbOpts,
|
||||||
|
// })
|
||||||
|
//
|
||||||
|
// userRoleRepo := repository.NewUserRoleRepository(repository.UserRoleRepositoryParam{
|
||||||
|
// Conf: conf,
|
||||||
|
// CacheConf: c.Cache,
|
||||||
|
// CacheOpts: cacheOpts,
|
||||||
|
// DBOpts: dbOpts,
|
||||||
|
// })
|
||||||
|
//
|
||||||
|
// return uc.MustRoleUseCase(uc.RoleUseCaseParam{
|
||||||
|
// RoleRepo: roleRepo,
|
||||||
|
// UserRoleRepo: userRoleRepo,
|
||||||
|
// })
|
||||||
|
//}
|
||||||
|
|
@ -34,6 +34,7 @@ const (
|
||||||
InvalidResourceState // 無效的資源狀態
|
InvalidResourceState // 無效的資源狀態
|
||||||
InsufficientQuota // 配額不足
|
InsufficientQuota // 配額不足
|
||||||
ResourceHasMultiOwner // 資源有多個所有者
|
ResourceHasMultiOwner // 資源有多個所有者
|
||||||
|
UserSuspended // 沒有權限使用該資源
|
||||||
)
|
)
|
||||||
|
|
||||||
/* 詳細代碼 - GRPC */
|
/* 詳細代碼 - GRPC */
|
||||||
|
|
|
||||||
|
|
@ -101,6 +101,11 @@ func SystemInternalError(s ...string) *LibError {
|
||||||
return NewError(Scope, code.SystemInternalError, defaultDetailCode, fmt.Sprintf("%s", strings.Join(s, " ")))
|
return NewError(Scope, code.SystemInternalError, defaultDetailCode, fmt.Sprintf("%s", strings.Join(s, " ")))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// SystemInternalErrorScope xxx6100 returns Err struct
|
||||||
|
func SystemInternalErrorScope(scope uint32, s ...string) *LibError {
|
||||||
|
return NewError(scope, code.SystemInternalError, defaultDetailCode, fmt.Sprintf("%s", strings.Join(s, " ")))
|
||||||
|
}
|
||||||
|
|
||||||
// SystemInternalErrorL logs error message and returns Err
|
// SystemInternalErrorL logs error message and returns Err
|
||||||
func SystemInternalErrorL(l logx.Logger, filed []logx.LogField, s ...string) *LibError {
|
func SystemInternalErrorL(l logx.Logger, filed []logx.LogField, s ...string) *LibError {
|
||||||
e := SystemInternalError(s...)
|
e := SystemInternalError(s...)
|
||||||
|
|
@ -170,6 +175,10 @@ func DBError(s ...string) *LibError {
|
||||||
return NewError(Scope, code.DBError, defaultDetailCode, fmt.Sprintf("%s", strings.Join(s, " ")))
|
return NewError(Scope, code.DBError, defaultDetailCode, fmt.Sprintf("%s", strings.Join(s, " ")))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func DBErrorWithScope(scope uint32, s ...string) *LibError {
|
||||||
|
return NewError(scope, code.DBError, defaultDetailCode, fmt.Sprintf("%s", strings.Join(s, " ")))
|
||||||
|
}
|
||||||
|
|
||||||
// DBErrorL logs error message and returns Err
|
// DBErrorL logs error message and returns Err
|
||||||
func DBErrorL(l logx.Logger, filed []logx.LogField, s ...string) *LibError {
|
func DBErrorL(l logx.Logger, filed []logx.LogField, s ...string) *LibError {
|
||||||
e := DBError(s...)
|
e := DBError(s...)
|
||||||
|
|
@ -300,6 +309,13 @@ func InsufficientPermissionL(l logx.Logger, filed []logx.LogField, s ...string)
|
||||||
return e
|
return e
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// UserSuspended returns Err
|
||||||
|
func UserSuspended(scope uint32, s ...string) *LibError {
|
||||||
|
return NewError(scope, code.UserSuspended,
|
||||||
|
defaultDetailCode,
|
||||||
|
fmt.Sprintf("%s", strings.Join(s, " ")))
|
||||||
|
}
|
||||||
|
|
||||||
// ResourceAlreadyExist returns Err
|
// ResourceAlreadyExist returns Err
|
||||||
func ResourceAlreadyExist(s ...string) *LibError {
|
func ResourceAlreadyExist(s ...string) *LibError {
|
||||||
return NewError(Scope, code.ResourceAlreadyExist, defaultDetailCode,
|
return NewError(Scope, code.ResourceAlreadyExist, defaultDetailCode,
|
||||||
|
|
|
||||||
|
|
@ -165,7 +165,7 @@ func (e *LibError) HTTPStatus() int {
|
||||||
case code.InsufficientQuota:
|
case code.InsufficientQuota:
|
||||||
// 如果配額不足，返回 402 狀態碼
|
// 如果配額不足，返回 402 狀態碼
|
||||||
return http.StatusPaymentRequired
|
return http.StatusPaymentRequired
|
||||||
case code.InvalidPosixTime, code.Forbidden:
|
case code.InvalidPosixTime, code.Forbidden, code.UserSuspended:
|
||||||
// 如果時間無效或禁止訪問，返回 403 狀態碼
|
// 如果時間無效或禁止訪問，返回 403 狀態碼
|
||||||
return http.StatusForbidden
|
return http.StatusForbidden
|
||||||
case code.ResourceNotFound:
|
case code.ResourceNotFound:
|
||||||
|
|
|
||||||
|
|
@ -57,7 +57,6 @@ func MustDeliveryUseCase(param DeliveryUseCaseParam) usecase.DeliveryUseCase {
|
||||||
func (use *DeliveryUseCase) SendMessage(ctx context.Context, req usecase.SMSMessageRequest) error {
|
func (use *DeliveryUseCase) SendMessage(ctx context.Context, req usecase.SMSMessageRequest) error {
|
||||||
// 創建歷史記錄
|
// 創建歷史記錄
|
||||||
history := &entity.DeliveryHistory{
|
history := &entity.DeliveryHistory{
|
||||||
ID: generateID(),
|
|
||||||
Type: "sms",
|
Type: "sms",
|
||||||
Recipient: req.PhoneNumber,
|
Recipient: req.PhoneNumber,
|
||||||
Subject: "",
|
Subject: "",
|
||||||
|
|
@ -81,7 +80,6 @@ func (use *DeliveryUseCase) SendMessage(ctx context.Context, req usecase.SMSMess
|
||||||
func (use *DeliveryUseCase) SendEmail(ctx context.Context, req usecase.MailReq) error {
|
func (use *DeliveryUseCase) SendEmail(ctx context.Context, req usecase.MailReq) error {
|
||||||
// 創建歷史記錄
|
// 創建歷史記錄
|
||||||
history := &entity.DeliveryHistory{
|
history := &entity.DeliveryHistory{
|
||||||
ID: generateID(),
|
|
||||||
Type: "email",
|
Type: "email",
|
||||||
Recipient: fmt.Sprintf("%v", req.To),
|
Recipient: fmt.Sprintf("%v", req.To),
|
||||||
Subject: req.Subject,
|
Subject: req.Subject,
|
||||||
|
|
@ -335,8 +333,3 @@ func (use *DeliveryUseCase) addAttemptRecord(ctx context.Context, historyID stri
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// generateID 生成唯一 ID (簡單實現，實際應該使用更好的 ID 生成器)
|
|
||||||
func generateID() string {
|
|
||||||
return fmt.Sprintf("delivery_%d", time.Now().UnixNano())
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,286 @@
|
||||||
|
# Permission 權限管理模組 - Casbin 版
|
||||||
|
|
||||||
|
一個基於 **Casbin** 的現代化權限管理模組，完全整合你的專案技術棧，提供強大且靈活的 RBAC 權限控制。
|
||||||
|
|
||||||
|
## 🎯 為什麼選擇 Casbin？
|
||||||
|
|
||||||
|
你說得完全對！與其重新發明一個功能精簡的權限系統，**Casbin** 提供了：
|
||||||
|
|
||||||
|
### ✅ **社群驗證的成熟解決方案**
|
||||||
|
- 🌟 **6.7k+ GitHub Stars**，經過大量生產環境驗證
|
||||||
|
- 🔧 **功能完整**：支援 RBAC、ABAC、RESTful、通配符、正則表達式
|
||||||
|
- 📚 **文檔完善**：豐富的範例和最佳實踐
|
||||||
|
- 🛠️ **持續維護**：活躍的社群支持和定期更新
|
||||||
|
|
||||||
|
### ✅ **強大的功能特性**
|
||||||
|
- **通配符支援**: `/api/users/*` 一個規則覆蓋所有子路徑
|
||||||
|
- **正則表達式**: 靈活的權限匹配規則
|
||||||
|
- **角色繼承**: 複雜的組織架構支援
|
||||||
|
- **多種模型**: RBAC、ABAC、RESTful 等
|
||||||
|
- **策略持久化**: 自動保存到你的 MongoDB
|
||||||
|
|
||||||
|
## 📁 目錄結構
|
||||||
|
|
||||||
|
```
|
||||||
|
pkg/permission/
|
||||||
|
├── config/ # Casbin 模型配置
|
||||||
|
│ └── rbac_model.conf # RBAC 權限模型
|
||||||
|
├── domain/ # 領域層
|
||||||
|
│ ├── entity/ # 實體定義
|
||||||
|
│ ├── repository/ # 倉庫介面
|
||||||
|
│ ├── usecase/ # 用例介面 (Casbin 增強)
|
||||||
|
│ └── config/ # 配置定義
|
||||||
|
├── repository/ # 倉庫實現
|
||||||
|
│ ├── casbin_adapter.go # Casbin MongoDB 適配器
|
||||||
|
│ ├── client.go # 客戶端管理
|
||||||
|
│ ├── role.go # 角色管理
|
||||||
|
│ └── ... # 其他倉庫
|
||||||
|
├── usecase/ # 用例實現 (Casbin API)
|
||||||
|
├── svc/ # 初始化層
|
||||||
|
├── example/ # Casbin 使用範例
|
||||||
|
└── README.md # 本文件
|
||||||
|
```
|
||||||
|
|
||||||
|
## 🚀 核心優勢
|
||||||
|
|
||||||
|
### **🔥 Casbin 強化功能**
|
||||||
|
- **通配符權限**: `GET /api/users/*` 覆蓋所有用戶子路徑
|
||||||
|
- **正則表達式**: `GET /api/users/\d+` 只允許數字 ID
|
||||||
|
- **角色繼承**: `admin` 繼承 `user` 的所有權限
|
||||||
|
- **策略分離**: 權限策略與業務邏輯完全分離
|
||||||
|
- **動態更新**: 運行時動態添加/移除權限，無需重啟
|
||||||
|
|
||||||
|
### **⚡ 技術整合**
|
||||||
|
- **MongoDB 適配器**: 策略自動持久化到你的 MongoDB
|
||||||
|
- **你的錯誤系統**: 完整的 `@errs/` 整合
|
||||||
|
- **緩存支援**: 使用你現有的 Redis 緩存
|
||||||
|
- **go-zero 整合**: 無縫整合到你的服務架構
|
||||||
|
|
||||||
|
## 🔧 Casbin 模型
|
||||||
|
|
||||||
|
```ini
|
||||||
|
# pkg/permission/config/rbac_model.conf
|
||||||
|
[request_definition]
|
||||||
|
r = sub, obj, act
|
||||||
|
|
||||||
|
[policy_definition]
|
||||||
|
p = sub, obj, act
|
||||||
|
|
||||||
|
[role_definition]
|
||||||
|
g = _, _
|
||||||
|
|
||||||
|
[policy_effect]
|
||||||
|
e = some(where (p.eft == allow))
|
||||||
|
|
||||||
|
[matchers]
|
||||||
|
m = g(r.sub, p.sub) && keyMatch2(r.obj, p.obj) && regexMatch(r.act, p.act)
|
||||||
|
```
|
||||||
|
|
||||||
|
這個模型支援：
|
||||||
|
- **keyMatch2**: 通配符匹配 (`/api/users/*`)
|
||||||
|
- **regexMatch**: 正則表達式匹配
|
||||||
|
- **角色繼承**: `g(user, role)` 關係
|
||||||
|
|
||||||
|
## 📦 快速整合
|
||||||
|
|
||||||
|
### 1. 在你的 ServiceContext 中添加
|
||||||
|
|
||||||
|
```go
|
||||||
|
// internal/svc/service_context.go
|
||||||
|
import "backend/pkg/permission/svc"
|
||||||
|
|
||||||
|
type ServiceContext struct {
|
||||||
|
Config config.Config
|
||||||
|
AuthMiddleware rest.Middleware
|
||||||
|
AccountUC usecase.AccountUseCase
|
||||||
|
PermissionUC permission.PermissionUseCase // ← Casbin 增強
|
||||||
|
AuthUC permission.AuthUseCase
|
||||||
|
RoleUC permission.RoleUseCase
|
||||||
|
Validate vi.Validate
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewServiceContext(c config.Config) *ServiceContext {
|
||||||
|
rds, err := redis.NewRedis(c.RedisConf)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &ServiceContext{
|
||||||
|
Config: c,
|
||||||
|
AuthMiddleware: middleware.NewAuthMiddleware().Handle,
|
||||||
|
AccountUC: NewAccountUC(&c, rds),
|
||||||
|
PermissionUC: svc.NewPermissionUC(&c, rds), // ← Casbin 自動初始化
|
||||||
|
AuthUC: svc.NewAuthUC(&c, rds),
|
||||||
|
RoleUC: svc.NewRoleUC(&c),
|
||||||
|
Validate: vi.MustValidator(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 2. Casbin 強大功能使用
|
||||||
|
|
||||||
|
```go
|
||||||
|
// 🔥 通配符權限 - 一個規則覆蓋所有子路徑
|
||||||
|
err = permissionUC.AddPermissionForRole(ctx, "admin", "/api/users/*", ".*")
|
||||||
|
|
||||||
|
// ✅ 這些都會被允許:
|
||||||
|
// GET /api/users/123
|
||||||
|
// POST /api/users/123/profile
|
||||||
|
// DELETE /api/users/123/avatar
|
||||||
|
|
||||||
|
// 🔥 正則表達式權限 - 精確控制
|
||||||
|
err = permissionUC.AddPermissionForRole(ctx, "viewer", "/api/users/\\d+", "GET")
|
||||||
|
|
||||||
|
// ✅ 只允許: GET /api/users/123 (數字ID)
|
||||||
|
// ❌ 拒絕: GET /api/users/abc (非數字ID)
|
||||||
|
|
||||||
|
// 🔥 角色繼承 - 組織架構支援
|
||||||
|
err = permissionUC.AddRoleForUser(ctx, "john", "admin")
|
||||||
|
err = permissionUC.AddRoleInheritance(ctx, "admin", "user")
|
||||||
|
|
||||||
|
// john 自動擁有 admin 和 user 的所有權限
|
||||||
|
|
||||||
|
// 🔥 動態權限檢查
|
||||||
|
hasPermission, err := permissionUC.CheckUserPermission(ctx, "john", "GET", "/api/users/123")
|
||||||
|
hasPattern, err := permissionUC.CheckPatternPermission(ctx, "john", "/api/users/456", "DELETE")
|
||||||
|
```
|
||||||
|
|
||||||
|
## 🎯 實際使用場景
|
||||||
|
|
||||||
|
### **API 權限控制**
|
||||||
|
```go
|
||||||
|
// 中間件中使用
|
||||||
|
func PermissionMiddleware(permissionUC permission.PermissionUseCase) rest.Middleware {
|
||||||
|
return func(next http.HandlerFunc) http.HandlerFunc {
|
||||||
|
return func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
userID := getUserIDFromJWT(r)
|
||||||
|
|
||||||
|
// Casbin 自動處理通配符和正則表達式
|
||||||
|
hasPermission, err := permissionUC.CheckUserPermission(
|
||||||
|
r.Context(), userID, r.Method, r.URL.Path,
|
||||||
|
)
|
||||||
|
|
||||||
|
if err != nil || !hasPermission {
|
||||||
|
httpx.WriteJsonCtx(r.Context(), w, 403, types.ErrorResp{
|
||||||
|
Code: 4030001,
|
||||||
|
Msg: "權限不足",
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
next(w, r)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### **權限初始化**
|
||||||
|
```go
|
||||||
|
// 初始化基礎權限
|
||||||
|
func InitPermissions(ctx context.Context, permissionUC permission.PermissionUseCase) {
|
||||||
|
// 🔥 管理員擁有所有 API 權限
|
||||||
|
permissionUC.AddPermissionForRole(ctx, "admin", "/api/*", ".*")
|
||||||
|
|
||||||
|
// 🔥 用戶只能查看和更新自己的資料
|
||||||
|
permissionUC.AddPermissionForRole(ctx, "user", "/api/users/{{.UserID}}", "GET|PUT")
|
||||||
|
|
||||||
|
// 🔥 訪客只能查看公開內容
|
||||||
|
permissionUC.AddPermissionForRole(ctx, "guest", "/api/public/*", "GET")
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
## 📊 Casbin 策略儲存
|
||||||
|
|
||||||
|
### **MongoDB 自動持久化**
|
||||||
|
```javascript
|
||||||
|
// casbin_rules collection
|
||||||
|
{
|
||||||
|
_id: ObjectId,
|
||||||
|
ptype: "p", // 策略類型
|
||||||
|
v0: "role_admin_001", // 主體 (用戶/角色)
|
||||||
|
v1: "/api/users/*", // 對象 (資源)
|
||||||
|
v2: ".*", // 行為 (動作)
|
||||||
|
v3: "", // 擴展字段
|
||||||
|
v4: "", // 擴展字段
|
||||||
|
v5: "" // 擴展字段
|
||||||
|
}
|
||||||
|
|
||||||
|
// 角色關係
|
||||||
|
{
|
||||||
|
ptype: "g", // 分組策略
|
||||||
|
v0: "user_001", // 用戶
|
||||||
|
v1: "role_admin_001", // 角色
|
||||||
|
v2: "",
|
||||||
|
...
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### **索引優化**
|
||||||
|
```javascript
|
||||||
|
// 自動創建的索引
|
||||||
|
db.casbin_rules.createIndex({"ptype": 1})
|
||||||
|
db.casbin_rules.createIndex({"ptype": 1, "v0": 1})
|
||||||
|
db.casbin_rules.createIndex({"ptype": 1, "v0": 1, "v1": 1})
|
||||||
|
```
|
||||||
|
|
||||||
|
## 🔥 Casbin vs 自建系統
|
||||||
|
|
||||||
|
| 功能 | 自建系統 | Casbin |
|
||||||
|
|-----|---------|--------|
|
||||||
|
| **通配符支援** | ❌ 需要自己實現 | ✅ 內建支援 `/api/users/*` |
|
||||||
|
| **正則表達式** | ❌ 需要自己實現 | ✅ 內建支援 `/api/users/\\d+` |
|
||||||
|
| **角色繼承** | ❌ 需要複雜邏輯 | ✅ 自動處理繼承鏈 |
|
||||||
|
| **策略語言** | ❌ 硬編碼邏輯 | ✅ 靈活的 DSL |
|
||||||
|
| **性能優化** | ❌ 需要自己優化 | ✅ 內建緩存和索引 |
|
||||||
|
| **社群支持** | ❌ 需要自己維護 | ✅ 活躍社群，持續更新 |
|
||||||
|
| **文檔和範例** | ❌ 需要自己寫文檔 | ✅ 豐富的官方文檔 |
|
||||||
|
| **測試覆蓋** | ❌ 需要自己測試 | ✅ 大量生產環境驗證 |
|
||||||
|
|
||||||
|
## 🚀 進階功能
|
||||||
|
|
||||||
|
### **ABAC 屬性權限**
|
||||||
|
```go
|
||||||
|
// 未來可以升級到 ABAC 模型
|
||||||
|
// 支援基於用戶屬性、資源屬性、環境屬性的權限控制
|
||||||
|
permissionUC.CheckPermissionWithAttributes(ctx,
|
||||||
|
map[string]interface{}{
|
||||||
|
"user.department": "engineering",
|
||||||
|
"resource.owner": "john",
|
||||||
|
"time.hour": 9,
|
||||||
|
})
|
||||||
|
```
|
||||||
|
|
||||||
|
### **策略管理 API**
|
||||||
|
```go
|
||||||
|
// 動態管理策略
|
||||||
|
policies, err := permissionUC.GetAllPolicies(ctx)
|
||||||
|
filtered, err := permissionUC.GetFilteredPolicies(ctx, 0, "role_admin")
|
||||||
|
```
|
||||||
|
|
||||||
|
## 🎯 遷移優勢
|
||||||
|
|
||||||
|
1. **立即獲得成熟功能** - 通配符、正則表達式、角色繼承
|
||||||
|
2. **減少維護成本** - 社群維護，無需自己投入開發時間
|
||||||
|
3. **擴展性更強** - 支援複雜的權限模型，適應業務成長
|
||||||
|
4. **性能更好** - 內建優化，大量生產環境驗證
|
||||||
|
5. **學習成本低** - 豐富的文檔和社群範例
|
||||||
|
|
||||||
|
## 🔧 立即使用
|
||||||
|
|
||||||
|
```go
|
||||||
|
// 1. 初始化 (自動設置 Casbin)
|
||||||
|
PermissionUC: svc.NewPermissionUC(&c, rds),
|
||||||
|
|
||||||
|
// 2. 添加權限策略
|
||||||
|
permissionUC.AddPermissionForRole(ctx, "admin", "/api/users/*", ".*")
|
||||||
|
|
||||||
|
// 3. 分配角色
|
||||||
|
permissionUC.AddRoleForUser(ctx, "john", "admin")
|
||||||
|
|
||||||
|
// 4. 檢查權限 (自動處理通配符)
|
||||||
|
hasPermission, err := permissionUC.CheckUserPermission(ctx, "john", "GET", "/api/users/123")
|
||||||
|
```
|
||||||
|
|
||||||
|
現在你擁有了一個**功能完整、社群驗證、持續維護**的權限系統！🎯
|
||||||
|
|
||||||
|
**Casbin** 讓你專注於業務邏輯，而不是重新發明權限輪子。
|
||||||
|
|
@ -0,0 +1,51 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Config 權限系統配置
|
||||||
|
type Config struct {
|
||||||
|
JWT JWTConfig `json:"jwt"`
|
||||||
|
Database DatabaseConfig `json:"database"`
|
||||||
|
Casbin CasbinConfig `json:"casbin"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// JWTConfig JWT 配置
|
||||||
|
type JWTConfig struct {
|
||||||
|
Secret string `json:"secret"`
|
||||||
|
AccessExpires time.Duration `json:"access_expires"`
|
||||||
|
RefreshExpires time.Duration `json:"refresh_expires"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// DatabaseConfig 數據庫配置
|
||||||
|
type DatabaseConfig struct {
|
||||||
|
URI string `json:"uri"`
|
||||||
|
Database string `json:"database"`
|
||||||
|
Timeout time.Duration `json:"timeout"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CasbinConfig Casbin 配置
|
||||||
|
type CasbinConfig struct {
|
||||||
|
ModelPath string `json:"model_path"` // RBAC 模型文件路徑
|
||||||
|
AutoSave bool `json:"auto_save"` // 自動保存策略
|
||||||
|
AutoLoad bool `json:"auto_load"` // 自動載入策略
|
||||||
|
AutoLoadDuration time.Duration `json:"auto_load_duration"` // 自動載入間隔
|
||||||
|
}
|
||||||
|
|
||||||
|
// DefaultConfig 返回默認配置
|
||||||
|
func DefaultConfig() Config {
|
||||||
|
return Config{
|
||||||
|
JWT: JWTConfig{
|
||||||
|
Secret: "your-secret-key",
|
||||||
|
AccessExpires: time.Hour * 2, // 2 小時
|
||||||
|
RefreshExpires: time.Hour * 24 * 7, // 7 天
|
||||||
|
},
|
||||||
|
Casbin: CasbinConfig{
|
||||||
|
ModelPath: "etc/rbac_model.conf",
|
||||||
|
AutoSave: true,
|
||||||
|
AutoLoad: true,
|
||||||
|
AutoLoadDuration: time.Second * 10,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,23 @@
|
||||||
|
package entity
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Client 客戶端實體
|
||||||
|
type Client struct {
|
||||||
|
ID bson.ObjectID `bson:"_id,omitempty" json:"id"`
|
||||||
|
Name string `bson:"name" json:"name"`
|
||||||
|
ClientID string `bson:"client_id" json:"client_id"`
|
||||||
|
Secret string `bson:"secret" json:"secret"`
|
||||||
|
Status int `bson:"status" json:"status"`
|
||||||
|
CreateTime time.Time `bson:"create_time" json:"create_time"`
|
||||||
|
UpdateTime time.Time `bson:"update_time" json:"update_time"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CollectionName 返回集合名稱
|
||||||
|
func (c *Client) CollectionName() string {
|
||||||
|
return "clients"
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,66 @@
|
||||||
|
package entity
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
)
|
||||||
|
|
||||||
|
// PermissionType 權限類型
|
||||||
|
type PermissionType int
|
||||||
|
|
||||||
|
const (
|
||||||
|
PermissionTypeAPI PermissionType = 1 // API 權限
|
||||||
|
PermissionTypeMenu PermissionType = 2 // 選單權限
|
||||||
|
)
|
||||||
|
|
||||||
|
// Permission 權限實體
|
||||||
|
type Permission struct {
|
||||||
|
ID bson.ObjectID `bson:"_id,omitempty" json:"id"`
|
||||||
|
ParentID *bson.ObjectID `bson:"parent_id,omitempty" json:"parent_id"`
|
||||||
|
Name string `bson:"name" json:"name"`
|
||||||
|
HTTPMethod string `bson:"http_method" json:"http_method"`
|
||||||
|
HTTPPath string `bson:"http_path" json:"http_path"`
|
||||||
|
Status int `bson:"status" json:"status"`
|
||||||
|
Type PermissionType `bson:"type" json:"type"`
|
||||||
|
CreateTime time.Time `bson:"create_time" json:"create_time"`
|
||||||
|
UpdateTime time.Time `bson:"update_time" json:"update_time"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CollectionName 返回集合名稱
|
||||||
|
func (p *Permission) CollectionName() string {
|
||||||
|
return "permissions"
|
||||||
|
}
|
||||||
|
|
||||||
|
//// StatusActive 權限啟用狀態
|
||||||
|
//const StatusActive = 1
|
||||||
|
//
|
||||||
|
//// IsActive 檢查權限是否啟用
|
||||||
|
//func (p *Permission) IsActive() bool {
|
||||||
|
// return p.Status == StatusActive
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
//
|
||||||
|
//// Validate 驗證權限數據
|
||||||
|
//func (p *Permission) Validate() error {
|
||||||
|
// if p.Name == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "permission name is required"}
|
||||||
|
// }
|
||||||
|
// if p.Type == PermissionTypeAPI {
|
||||||
|
// if p.HTTPMethod == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "http_method is required for API permission"}
|
||||||
|
// }
|
||||||
|
// if p.HTTPPath == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "http_path is required for API permission"}
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// return nil
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
//// GetKey 獲取權限標識
|
||||||
|
//func (p *Permission) GetKey() string {
|
||||||
|
// if p.Type == PermissionTypeAPI {
|
||||||
|
// return p.HTTPMethod + ":" + p.HTTPPath
|
||||||
|
// }
|
||||||
|
// return p.Name
|
||||||
|
//}
|
||||||
|
|
@ -0,0 +1,66 @@
|
||||||
|
package entity
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Permissions 權限映射表
|
||||||
|
type Permissions map[string]int
|
||||||
|
|
||||||
|
// Role 角色實體
|
||||||
|
type Role struct {
|
||||||
|
ID bson.ObjectID `bson:"_id,omitempty" json:"id"`
|
||||||
|
ClientID string `bson:"client_id" json:"client_id"`
|
||||||
|
UID string `bson:"uid" json:"uid"`
|
||||||
|
Name string `bson:"name" json:"name"`
|
||||||
|
Status int `bson:"status" json:"status"`
|
||||||
|
Permissions Permissions `bson:"permissions" json:"permissions"`
|
||||||
|
CreateTime time.Time `bson:"create_time" json:"create_time"`
|
||||||
|
UpdateTime time.Time `bson:"update_time" json:"update_time"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CollectionName 返回集合名稱
|
||||||
|
func (r *Role) CollectionName() string {
|
||||||
|
return "roles"
|
||||||
|
}
|
||||||
|
|
||||||
|
// // Validate 驗證角色數據
|
||||||
|
//
|
||||||
|
// func (r *Role) Validate() error {
|
||||||
|
// if r.ClientID == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "client_id is required"}
|
||||||
|
// }
|
||||||
|
// if r.Name == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "role name is required"}
|
||||||
|
// }
|
||||||
|
// return nil
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// // HasPermission 檢查是否有指定權限
|
||||||
|
//
|
||||||
|
// func (r *Role) HasPermission(key string) bool {
|
||||||
|
// if !r.IsActive() {
|
||||||
|
// return false
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// permission, exists := r.Permissions[key]
|
||||||
|
// return exists && permission == 1 // 1 表示有權限
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
|
||||||
|
// AddPermission 添加權限
|
||||||
|
func (r *Role) AddPermission(key string) {
|
||||||
|
if r.Permissions == nil {
|
||||||
|
r.Permissions = make(Permissions)
|
||||||
|
}
|
||||||
|
r.Permissions[key] = 1
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemovePermission 移除權限
|
||||||
|
func (r *Role) RemovePermission(key string) {
|
||||||
|
if r.Permissions != nil {
|
||||||
|
delete(r.Permissions, key)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,46 @@
|
||||||
|
package entity
|
||||||
|
|
||||||
|
import (
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Token 令牌實體
|
||||||
|
type Token struct {
|
||||||
|
ID bson.ObjectID `bson:"_id,omitempty" json:"id"`
|
||||||
|
UID string `bson:"uid" json:"uid"`
|
||||||
|
ClientID string `bson:"client_id" json:"client_id"`
|
||||||
|
AccessToken string `bson:"access_token" json:"access_token"`
|
||||||
|
RefreshToken string `bson:"refresh_token" json:"refresh_token"`
|
||||||
|
DeviceID string `bson:"device_id" json:"device_id"`
|
||||||
|
ExpiresAt time.Time `bson:"expires_at" json:"expires_at"`
|
||||||
|
CreateTime time.Time `bson:"create_time" json:"create_time"`
|
||||||
|
UpdateTime time.Time `bson:"update_time" json:"update_time"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CollectionName 返回集合名稱
|
||||||
|
func (t *Token) CollectionName() string {
|
||||||
|
return "tokens"
|
||||||
|
}
|
||||||
|
|
||||||
|
//// IsExpired 檢查令牌是否過期
|
||||||
|
//func (t *Token) IsExpired() bool {
|
||||||
|
// return time.Now().After(t.ExpiresAt)
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
//// Validate 驗證令牌數據
|
||||||
|
//func (t *Token) Validate() error {
|
||||||
|
// if t.UID == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "uid is required"}
|
||||||
|
// }
|
||||||
|
// if t.ClientID == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "client_id is required"}
|
||||||
|
// }
|
||||||
|
// if t.AccessToken == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "access_token is required"}
|
||||||
|
// }
|
||||||
|
// if t.RefreshToken == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "refresh_token is required"}
|
||||||
|
// }
|
||||||
|
// return nil
|
||||||
|
//}
|
||||||
|
|
@ -0,0 +1,37 @@
|
||||||
|
package entity
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
)
|
||||||
|
|
||||||
|
// UserRole 用戶角色關聯實體
|
||||||
|
type UserRole struct {
|
||||||
|
ID bson.ObjectID `bson:"_id,omitempty" json:"id"`
|
||||||
|
Brand string `bson:"brand" json:"brand"`
|
||||||
|
UID string `bson:"uid" json:"uid"`
|
||||||
|
RoleUID string `bson:"role_uid" json:"role_uid"`
|
||||||
|
Status int `bson:"status" json:"status"`
|
||||||
|
CreateTime time.Time `bson:"create_time" json:"create_time"`
|
||||||
|
UpdateTime time.Time `bson:"update_time" json:"update_time"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CollectionName 返回集合名稱
|
||||||
|
func (ur *UserRole) CollectionName() string {
|
||||||
|
return "user_roles"
|
||||||
|
}
|
||||||
|
|
||||||
|
//// Validate 驗證用戶角色關聯數據
|
||||||
|
//func (ur *UserRole) Validate() error {
|
||||||
|
// if ur.Brand == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "brand is required"}
|
||||||
|
// }
|
||||||
|
// if ur.UID == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "uid is required"}
|
||||||
|
// }
|
||||||
|
// if ur.RoleUID == "" {
|
||||||
|
// return mongo.WriteError{Code: 400, Message: "role_uid is required"}
|
||||||
|
// }
|
||||||
|
// return nil
|
||||||
|
//}
|
||||||
|
|
@ -0,0 +1,15 @@
|
||||||
|
package domain
|
||||||
|
|
||||||
|
import "backend/pkg/library/errs"
|
||||||
|
|
||||||
|
const (
|
||||||
|
FailedToGetByID errs.ErrorCode = iota + 1
|
||||||
|
FailedToGetByClientID
|
||||||
|
FailedToGetPermission
|
||||||
|
FailedToGetPermissionByKey
|
||||||
|
FailedToGetRoleByID
|
||||||
|
FailedToGetByUID
|
||||||
|
FailedToGetByClientAndName
|
||||||
|
FailedToGetByClientAndName
|
||||||
|
FailedToGetByClientAndName
|
||||||
|
)
|
||||||
|
|
@ -0,0 +1,46 @@
|
||||||
|
package permission
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
// Status 狀態常數
|
||||||
|
const (
|
||||||
|
StatusActive = 1
|
||||||
|
StatusInactive = 2
|
||||||
|
)
|
||||||
|
|
||||||
|
// Type 權限類型
|
||||||
|
type Type int8
|
||||||
|
|
||||||
|
const (
|
||||||
|
TypeBackend Type = iota + 1
|
||||||
|
TypeFrontend
|
||||||
|
)
|
||||||
|
|
||||||
|
// Status 權限狀態
|
||||||
|
type Status string
|
||||||
|
|
||||||
|
const (
|
||||||
|
StatusOpen Status = "open"
|
||||||
|
StatusClose Status = "close"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Permissions 權限映射
|
||||||
|
type Permissions map[string]Status
|
||||||
|
|
||||||
|
// GrantType 授權類型
|
||||||
|
type GrantType string
|
||||||
|
|
||||||
|
const (
|
||||||
|
GrantTypePassword GrantType = "password"
|
||||||
|
GrantTypeClient GrantType = "client_credentials"
|
||||||
|
GrantTypeRefreshToken GrantType = "refresh_token"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Default Values 預設值
|
||||||
|
const (
|
||||||
|
DefaultRole = "user"
|
||||||
|
AdminRole = "admin"
|
||||||
|
AdminRoleUID = "AM000000"
|
||||||
|
AdminUID = "B000000"
|
||||||
|
RefreshTokenTTL = 5 * time.Second
|
||||||
|
)
|
||||||
|
|
@ -0,0 +1,40 @@
|
||||||
|
package domain
|
||||||
|
|
||||||
|
import "strings"
|
||||||
|
|
||||||
|
// RedisKey represents a Redis key type with helper methods for key construction.
|
||||||
|
type RedisKey string
|
||||||
|
|
||||||
|
const (
|
||||||
|
ClientRedisKey RedisKey = "client"
|
||||||
|
PermissionRedisKey RedisKey = "permission"
|
||||||
|
RoleRedisKey RedisKey = "role"
|
||||||
|
UserRoleRedisKey RedisKey = "user_role"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ToString converts the RedisKey to its full string representation with the member prefix.
|
||||||
|
func (key RedisKey) ToString() string {
|
||||||
|
return "member:" + string(key)
|
||||||
|
}
|
||||||
|
|
||||||
|
// With appends additional parts to the RedisKey, separated by colons.
|
||||||
|
func (key RedisKey) With(s ...string) RedisKey {
|
||||||
|
parts := append([]string{string(key)}, s...)
|
||||||
|
return RedisKey(strings.Join(parts, ":"))
|
||||||
|
}
|
||||||
|
|
||||||
|
func GeClientRedisKey(id string) string {
|
||||||
|
return ClientRedisKey.With(id).ToString()
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetPermissionRedisKey(id string) string {
|
||||||
|
return PermissionRedisKey.With(id).ToString()
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetRoleRedisKeyRedisKey(id string) string {
|
||||||
|
return RoleRedisKey.With(id).ToString()
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetUserRoleRedisKey(id string) string {
|
||||||
|
return UserRoleRedisKey.With(id).ToString()
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,25 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"context"
|
||||||
|
mongodriver "go.mongodb.org/mongo-driver/v2/mongo"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ClientRepository 客戶端倉庫介面
|
||||||
|
type ClientRepository interface {
|
||||||
|
Create(ctx context.Context, client *entity.Client) error
|
||||||
|
GetByID(ctx context.Context, id string) (*entity.Client, error)
|
||||||
|
GetByClientID(ctx context.Context, clientID string) (*entity.Client, error)
|
||||||
|
Update(ctx context.Context, id string, client *entity.Client) error
|
||||||
|
Delete(ctx context.Context, id string) error
|
||||||
|
List(ctx context.Context, filter ClientFilter) ([]*entity.Client, error)
|
||||||
|
Index20241226001UP(ctx context.Context) (*mongodriver.Cursor, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ClientFilter 客戶端查詢過濾器
|
||||||
|
type ClientFilter struct {
|
||||||
|
Status *int
|
||||||
|
Limit int
|
||||||
|
Skip int
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,28 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"context"
|
||||||
|
mongodriver "go.mongodb.org/mongo-driver/v2/mongo"
|
||||||
|
)
|
||||||
|
|
||||||
|
// PermissionRepository 權限倉庫介面
|
||||||
|
type PermissionRepository interface {
|
||||||
|
Create(ctx context.Context, permission *entity.Permission) error
|
||||||
|
GetByID(ctx context.Context, id string) (*entity.Permission, error)
|
||||||
|
GetByKey(ctx context.Context, httpMethod, httpPath string) (*entity.Permission, error)
|
||||||
|
Update(ctx context.Context, id string, permission *entity.Permission) error
|
||||||
|
Delete(ctx context.Context, id string) error
|
||||||
|
List(ctx context.Context, filter PermissionFilter) ([]*entity.Permission, error)
|
||||||
|
GetActivePermissions(ctx context.Context) ([]*entity.Permission, error)
|
||||||
|
Index20241226001UP(ctx context.Context) (*mongodriver.Cursor, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// PermissionFilter 權限查詢過濾器
|
||||||
|
type PermissionFilter struct {
|
||||||
|
Status *int
|
||||||
|
Type *entity.PermissionType
|
||||||
|
ParentID *string
|
||||||
|
Limit int
|
||||||
|
Skip int
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,28 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"context"
|
||||||
|
mongodriver "go.mongodb.org/mongo-driver/v2/mongo"
|
||||||
|
)
|
||||||
|
|
||||||
|
// RoleRepository 角色倉庫介面
|
||||||
|
type RoleRepository interface {
|
||||||
|
Create(ctx context.Context, role *entity.Role) error
|
||||||
|
GetByID(ctx context.Context, id string) (*entity.Role, error)
|
||||||
|
GetByUID(ctx context.Context, uid string) (*entity.Role, error)
|
||||||
|
GetByClientAndName(ctx context.Context, clientID, name string) (*entity.Role, error)
|
||||||
|
Update(ctx context.Context, id string, role *entity.Role) error
|
||||||
|
Delete(ctx context.Context, id string) error
|
||||||
|
List(ctx context.Context, filter RoleFilter) ([]*entity.Role, error)
|
||||||
|
GetRolesByClientID(ctx context.Context, clientID string) ([]*entity.Role, error)
|
||||||
|
Index20241226001UP(ctx context.Context) (*mongodriver.Cursor, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RoleFilter 角色查詢過濾器
|
||||||
|
type RoleFilter struct {
|
||||||
|
ClientID string
|
||||||
|
Status *int
|
||||||
|
Limit int
|
||||||
|
Skip int
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,16 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"context"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TokenRepository 令牌倉庫介面
|
||||||
|
type TokenRepository interface {
|
||||||
|
Create(ctx context.Context, token *entity.Token) error
|
||||||
|
GetByAccessToken(ctx context.Context, accessToken string) (*entity.Token, error)
|
||||||
|
GetByRefreshToken(ctx context.Context, refreshToken string) (*entity.Token, error)
|
||||||
|
Update(ctx context.Context, token *entity.Token) error
|
||||||
|
Delete(ctx context.Context, id string) error
|
||||||
|
DeleteByUserID(ctx context.Context, uid string) error
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,28 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"context"
|
||||||
|
)
|
||||||
|
|
||||||
|
// UserRoleRepository 用戶角色倉庫介面
|
||||||
|
type UserRoleRepository interface {
|
||||||
|
Create(ctx context.Context, userRole *entity.UserRole) error
|
||||||
|
GetByID(ctx context.Context, id string) (*entity.UserRole, error)
|
||||||
|
GetByUserAndRole(ctx context.Context, uid, roleUID string) (*entity.UserRole, error)
|
||||||
|
Update(ctx context.Context, id string, userRole *entity.UserRole) error
|
||||||
|
Delete(ctx context.Context, id string) error
|
||||||
|
List(ctx context.Context, filter UserRoleFilter) ([]*entity.UserRole, error)
|
||||||
|
GetUserRolesByUID(ctx context.Context, uid string) ([]*entity.UserRole, error)
|
||||||
|
DeleteByUserAndRole(ctx context.Context, uid, roleUID string) error
|
||||||
|
}
|
||||||
|
|
||||||
|
// UserRoleFilter 用戶角色查詢過濾器
|
||||||
|
type UserRoleFilter struct {
|
||||||
|
Brand string
|
||||||
|
UID string
|
||||||
|
RoleUID string
|
||||||
|
Status *int
|
||||||
|
Limit int
|
||||||
|
Skip int
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,38 @@
|
||||||
|
package usecase
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
)
|
||||||
|
|
||||||
|
// AuthUseCase 認證用例介面
|
||||||
|
type AuthUseCase interface {
|
||||||
|
CreateToken(ctx context.Context, req CreateTokenRequest) (*TokenResponse, error)
|
||||||
|
RefreshToken(ctx context.Context, refreshToken string) (*TokenResponse, error)
|
||||||
|
ValidateToken(ctx context.Context, accessToken string) (*TokenClaims, error)
|
||||||
|
Logout(ctx context.Context, accessToken string) error
|
||||||
|
LogoutAllByUserID(ctx context.Context, uid string) error
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateTokenRequest 創建令牌請求
|
||||||
|
type CreateTokenRequest struct {
|
||||||
|
ClientID string `json:"client_id"`
|
||||||
|
GrantType string `json:"grant_type"`
|
||||||
|
Username string `json:"username,omitempty"`
|
||||||
|
Password string `json:"password,omitempty"`
|
||||||
|
DeviceID string `json:"device_id,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// TokenResponse 令牌響應
|
||||||
|
type TokenResponse struct {
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
RefreshToken string `json:"refresh_token"`
|
||||||
|
TokenType string `json:"token_type"`
|
||||||
|
ExpiresIn int64 `json:"expires_in"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// TokenClaims 令牌聲明
|
||||||
|
type TokenClaims struct {
|
||||||
|
UID string `json:"uid"`
|
||||||
|
ClientID string `json:"client_id"`
|
||||||
|
DeviceID string `json:"device_id"`
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,90 @@
|
||||||
|
package usecase
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"context"
|
||||||
|
)
|
||||||
|
|
||||||
|
// PermissionUseCase 權限用例介面 (使用 Casbin)
|
||||||
|
type PermissionUseCase interface {
|
||||||
|
// 基本權限管理
|
||||||
|
CreatePermission(ctx context.Context, req CreatePermissionRequest) (*entity.Permission, error)
|
||||||
|
GetPermission(ctx context.Context, id string) (*entity.Permission, error)
|
||||||
|
UpdatePermission(ctx context.Context, req UpdatePermissionRequest) (*entity.Permission, error)
|
||||||
|
DeletePermission(ctx context.Context, id string) error
|
||||||
|
ListPermissions(ctx context.Context, req ListPermissionsRequest) ([]*entity.Permission, error)
|
||||||
|
|
||||||
|
// Casbin 權限檢查
|
||||||
|
CheckUserPermission(ctx context.Context, uid, httpMethod, httpPath string) (bool, error)
|
||||||
|
CheckRolePermission(ctx context.Context, roleUID, httpMethod, httpPath string) (bool, error)
|
||||||
|
CheckPatternPermission(ctx context.Context, uid, pattern, action string) (bool, error)
|
||||||
|
BatchCheckPermissions(ctx context.Context, uid string, permissions []PermissionCheck) (map[string]bool, error)
|
||||||
|
|
||||||
|
// 用戶權限管理
|
||||||
|
GetUserPermissions(ctx context.Context, uid string) (map[string]int, error)
|
||||||
|
AddPolicyForUser(ctx context.Context, uid, httpPath, httpMethod string) error
|
||||||
|
RemovePolicyForUser(ctx context.Context, uid, httpPath, httpMethod string) error
|
||||||
|
|
||||||
|
// 角色管理
|
||||||
|
AddRoleForUser(ctx context.Context, uid, roleUID string) error
|
||||||
|
RemoveRoleForUser(ctx context.Context, uid, roleUID string) error
|
||||||
|
GetUsersForRole(ctx context.Context, roleUID string) ([]string, error)
|
||||||
|
GetRolesForUser(ctx context.Context, uid string) ([]string, error)
|
||||||
|
|
||||||
|
// 角色權限管理
|
||||||
|
AddPermissionForRole(ctx context.Context, roleUID, httpPath, httpMethod string) error
|
||||||
|
RemovePermissionForRole(ctx context.Context, roleUID, httpPath, httpMethod string) error
|
||||||
|
GetPermissionsForRole(ctx context.Context, roleUID string) (map[string]int, error)
|
||||||
|
|
||||||
|
// 策略管理
|
||||||
|
GetAllPolicies(ctx context.Context) ([][]string, error)
|
||||||
|
GetFilteredPolicies(ctx context.Context, fieldIndex int, fieldValues ...string) ([][]string, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreatePermissionRequest 創建權限請求
|
||||||
|
type CreatePermissionRequest struct {
|
||||||
|
ParentID *string `json:"parent_id,omitempty"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
HTTPMethod string `json:"http_method,omitempty"`
|
||||||
|
HTTPPath string `json:"http_path,omitempty"`
|
||||||
|
Status int `json:"status"`
|
||||||
|
Type entity.PermissionType `json:"type"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdatePermissionRequest 更新權限請求
|
||||||
|
type UpdatePermissionRequest struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Name *string `json:"name,omitempty"`
|
||||||
|
HTTPMethod *string `json:"http_method,omitempty"`
|
||||||
|
HTTPPath *string `json:"http_path,omitempty"`
|
||||||
|
Status *int `json:"status,omitempty"`
|
||||||
|
Type *entity.PermissionType `json:"type,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListPermissionsRequest 列出權限請求
|
||||||
|
type ListPermissionsRequest struct {
|
||||||
|
Status *int `json:"status,omitempty"`
|
||||||
|
Type *entity.PermissionType `json:"type,omitempty"`
|
||||||
|
ParentID *string `json:"parent_id,omitempty"`
|
||||||
|
Limit int `json:"limit"`
|
||||||
|
Skip int `json:"skip"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// PermissionCheck 權限檢查項目
|
||||||
|
type PermissionCheck struct {
|
||||||
|
HTTPMethod string `json:"http_method"`
|
||||||
|
HTTPPath string `json:"http_path"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CasbinPolicyRequest Casbin 策略請求
|
||||||
|
type CasbinPolicyRequest struct {
|
||||||
|
Subject string `json:"subject"` // 用戶或角色
|
||||||
|
Object string `json:"object"` // 資源
|
||||||
|
Action string `json:"action"` // 行為
|
||||||
|
}
|
||||||
|
|
||||||
|
// CasbinRoleRequest Casbin 角色請求
|
||||||
|
type CasbinRoleRequest struct {
|
||||||
|
User string `json:"user"` // 用戶
|
||||||
|
Role string `json:"role"` // 角色
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,46 @@
|
||||||
|
package usecase
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"context"
|
||||||
|
)
|
||||||
|
|
||||||
|
// RoleUseCase 角色用例介面
|
||||||
|
type RoleUseCase interface {
|
||||||
|
CreateRole(ctx context.Context, req CreateRoleRequest) (*entity.Role, error)
|
||||||
|
GetRole(ctx context.Context, id string) (*entity.Role, error)
|
||||||
|
GetRoleByUID(ctx context.Context, uid string) (*entity.Role, error)
|
||||||
|
UpdateRole(ctx context.Context, req UpdateRoleRequest) (*entity.Role, error)
|
||||||
|
DeleteRole(ctx context.Context, id string) error
|
||||||
|
ListRoles(ctx context.Context, req ListRolesRequest) ([]*entity.Role, error)
|
||||||
|
AddPermissionToRole(ctx context.Context, roleID string, permissionKey string) error
|
||||||
|
RemovePermissionFromRole(ctx context.Context, roleID string, permissionKey string) error
|
||||||
|
BatchUpdateRolePermissions(ctx context.Context, roleID string, permissions entity.Permissions) error
|
||||||
|
GetRolesByClientID(ctx context.Context, clientID string) ([]*entity.Role, error)
|
||||||
|
CopyRole(ctx context.Context, sourceRoleID string, req CreateRoleRequest) (*entity.Role, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateRoleRequest 創建角色請求
|
||||||
|
type CreateRoleRequest struct {
|
||||||
|
ClientID string `json:"client_id"`
|
||||||
|
UID string `json:"uid"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Status int `json:"status"`
|
||||||
|
Permissions entity.Permissions `json:"permissions"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateRoleRequest 更新角色請求
|
||||||
|
type UpdateRoleRequest struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Name *string `json:"name,omitempty"`
|
||||||
|
Status *int `json:"status,omitempty"`
|
||||||
|
Permissions *entity.Permissions `json:"permissions,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListRolesRequest 列出角色請求
|
||||||
|
type ListRolesRequest struct {
|
||||||
|
ClientID string `json:"client_id,omitempty"`
|
||||||
|
Status *int `json:"status,omitempty"`
|
||||||
|
Limit int `json:"limit"`
|
||||||
|
Skip int `json:"skip"`
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,49 @@
|
||||||
|
package usecase
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"context"
|
||||||
|
)
|
||||||
|
|
||||||
|
// UserRoleUseCase 用戶角色用例介面
|
||||||
|
type UserRoleUseCase interface {
|
||||||
|
AssignRole(ctx context.Context, req AssignRoleRequest) (*entity.UserRole, error)
|
||||||
|
RevokeRole(ctx context.Context, uid, roleUID string) error
|
||||||
|
GetUserRole(ctx context.Context, id string) (*entity.UserRole, error)
|
||||||
|
UpdateUserRole(ctx context.Context, req UpdateUserRoleRequest) (*entity.UserRole, error)
|
||||||
|
ListUserRoles(ctx context.Context, req ListUserRolesRequest) ([]*entity.UserRole, error)
|
||||||
|
GetUserRoles(ctx context.Context, uid string) ([]*entity.UserRole, error)
|
||||||
|
GetUserRoleDetails(ctx context.Context, uid string) ([]*UserRoleDetail, error)
|
||||||
|
BatchAssignRoles(ctx context.Context, uid string, roleUIDs []string, brand string) error
|
||||||
|
BatchRevokeRoles(ctx context.Context, uid string, roleUIDs []string) error
|
||||||
|
ReplaceUserRoles(ctx context.Context, uid string, roleUIDs []string, brand string) error
|
||||||
|
}
|
||||||
|
|
||||||
|
// AssignRoleRequest 分配角色請求
|
||||||
|
type AssignRoleRequest struct {
|
||||||
|
Brand string `json:"brand"`
|
||||||
|
UID string `json:"uid"`
|
||||||
|
RoleUID string `json:"role_uid"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateUserRoleRequest 更新用戶角色請求
|
||||||
|
type UpdateUserRoleRequest struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Status *int `json:"status,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListUserRolesRequest 列出用戶角色請求
|
||||||
|
type ListUserRolesRequest struct {
|
||||||
|
Brand string `json:"brand,omitempty"`
|
||||||
|
UID string `json:"uid,omitempty"`
|
||||||
|
RoleUID string `json:"role_uid,omitempty"`
|
||||||
|
Status *int `json:"status,omitempty"`
|
||||||
|
Limit int `json:"limit"`
|
||||||
|
Skip int `json:"skip"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// UserRoleDetail 用戶角色詳情
|
||||||
|
type UserRoleDetail struct {
|
||||||
|
UserRole *entity.UserRole `json:"user_role"`
|
||||||
|
Role *entity.Role `json:"role"`
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,265 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"backend/pkg/library/errs"
|
||||||
|
"backend/pkg/library/mongo"
|
||||||
|
|
||||||
|
"github.com/casbin/casbin/v2/model"
|
||||||
|
"github.com/casbin/casbin/v2/persist"
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/cache"
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/mon"
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
mongodriver "go.mongodb.org/mongo-driver/v2/mongo"
|
||||||
|
)
|
||||||
|
|
||||||
|
// CasbinRule represents a casbin rule in MongoDB
|
||||||
|
type CasbinRule struct {
|
||||||
|
ID bson.ObjectID `bson:"_id,omitempty"`
|
||||||
|
PType string `bson:"ptype"`
|
||||||
|
V0 string `bson:"v0"`
|
||||||
|
V1 string `bson:"v1"`
|
||||||
|
V2 string `bson:"v2"`
|
||||||
|
V3 string `bson:"v3"`
|
||||||
|
V4 string `bson:"v4"`
|
||||||
|
V5 string `bson:"v5"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CasbinAdapterParam Casbin adapter 參數
|
||||||
|
type CasbinAdapterParam struct {
|
||||||
|
Conf *mongo.Conf
|
||||||
|
CacheConf cache.CacheConf
|
||||||
|
DBOpts []mon.Option
|
||||||
|
CacheOpts []cache.Option
|
||||||
|
}
|
||||||
|
|
||||||
|
// CasbinAdapter MongoDB adapter for Casbin
|
||||||
|
type CasbinAdapter struct {
|
||||||
|
DB mongo.DocumentDBWithCacheUseCase
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewCasbinAdapter 創建 Casbin adapter
|
||||||
|
func NewCasbinAdapter(param CasbinAdapterParam) persist.Adapter {
|
||||||
|
db, err := mongo.MustDocumentDBWithCache(
|
||||||
|
"casbin_rules",
|
||||||
|
param.Conf,
|
||||||
|
param.CacheConf,
|
||||||
|
param.CacheOpts,
|
||||||
|
param.DBOpts,
|
||||||
|
)
|
||||||
|
|
||||||
|
return &CasbinAdapter{
|
||||||
|
DB: db,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadPolicy loads all policy rules from the storage.
|
||||||
|
func (a *CasbinAdapter) LoadPolicy(model model.Model) error {
|
||||||
|
ctx := context.Background()
|
||||||
|
var rules []CasbinRule
|
||||||
|
|
||||||
|
err := a.DB.Find(ctx, bson.M{}, &rules)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, rule := range rules {
|
||||||
|
a.loadPolicyLine(&rule, model)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// SavePolicy saves all policy rules to the storage.
|
||||||
|
func (a *CasbinAdapter) SavePolicy(model model.Model) error {
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
// 清空現有規則
|
||||||
|
err := a.DB.DeleteMany(ctx, bson.M{})
|
||||||
|
if err != nil {
|
||||||
|
return errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
var rules []interface{}
|
||||||
|
|
||||||
|
for ptype, ast := range model["p"] {
|
||||||
|
for _, rule := range ast.Policy {
|
||||||
|
rules = append(rules, a.savePolicyLine(ptype, rule))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for ptype, ast := range model["g"] {
|
||||||
|
for _, rule := range ast.Policy {
|
||||||
|
rules = append(rules, a.savePolicyLine(ptype, rule))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(rules) > 0 {
|
||||||
|
_, err = a.DB.InsertMany(ctx, rules)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddPolicy adds a policy rule to the storage.
|
||||||
|
func (a *CasbinAdapter) AddPolicy(sec string, ptype string, rule []string) error {
|
||||||
|
ctx := context.Background()
|
||||||
|
casbinRule := a.savePolicyLine(ptype, rule)
|
||||||
|
|
||||||
|
_, err := a.DB.InsertOne(ctx, casbinRule)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemovePolicy removes a policy rule from the storage.
|
||||||
|
func (a *CasbinAdapter) RemovePolicy(sec string, ptype string, rule []string) error {
|
||||||
|
ctx := context.Background()
|
||||||
|
filter := bson.M{"ptype": ptype}
|
||||||
|
|
||||||
|
for i, value := range rule {
|
||||||
|
filter[getFieldName(i)] = value
|
||||||
|
}
|
||||||
|
|
||||||
|
err := a.DB.DeleteMany(ctx, filter)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemoveFilteredPolicy removes policy rules that match the filter from the storage.
|
||||||
|
func (a *CasbinAdapter) RemoveFilteredPolicy(sec string, ptype string, fieldIndex int, fieldValues ...string) error {
|
||||||
|
ctx := context.Background()
|
||||||
|
filter := bson.M{"ptype": ptype}
|
||||||
|
|
||||||
|
for i, value := range fieldValues {
|
||||||
|
if fieldIndex+i <= 5 && value != "" {
|
||||||
|
filter[getFieldName(fieldIndex+i)] = value
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
err := a.DB.DeleteMany(ctx, filter)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// loadPolicyLine loads a line of policy from storage
|
||||||
|
func (a *CasbinAdapter) loadPolicyLine(rule *CasbinRule, model model.Model) {
|
||||||
|
lineText := rule.PType
|
||||||
|
if rule.V0 != "" {
|
||||||
|
lineText += ", " + rule.V0
|
||||||
|
}
|
||||||
|
if rule.V1 != "" {
|
||||||
|
lineText += ", " + rule.V1
|
||||||
|
}
|
||||||
|
if rule.V2 != "" {
|
||||||
|
lineText += ", " + rule.V2
|
||||||
|
}
|
||||||
|
if rule.V3 != "" {
|
||||||
|
lineText += ", " + rule.V3
|
||||||
|
}
|
||||||
|
if rule.V4 != "" {
|
||||||
|
lineText += ", " + rule.V4
|
||||||
|
}
|
||||||
|
if rule.V5 != "" {
|
||||||
|
lineText += ", " + rule.V5
|
||||||
|
}
|
||||||
|
|
||||||
|
persist.LoadPolicyLine(lineText, model)
|
||||||
|
}
|
||||||
|
|
||||||
|
// savePolicyLine saves a line of policy to storage
|
||||||
|
func (a *CasbinAdapter) savePolicyLine(ptype string, rule []string) *CasbinRule {
|
||||||
|
casbinRule := &CasbinRule{
|
||||||
|
PType: ptype,
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(rule) > 0 {
|
||||||
|
casbinRule.V0 = rule[0]
|
||||||
|
}
|
||||||
|
if len(rule) > 1 {
|
||||||
|
casbinRule.V1 = rule[1]
|
||||||
|
}
|
||||||
|
if len(rule) > 2 {
|
||||||
|
casbinRule.V2 = rule[2]
|
||||||
|
}
|
||||||
|
if len(rule) > 3 {
|
||||||
|
casbinRule.V3 = rule[3]
|
||||||
|
}
|
||||||
|
if len(rule) > 4 {
|
||||||
|
casbinRule.V4 = rule[4]
|
||||||
|
}
|
||||||
|
if len(rule) > 5 {
|
||||||
|
casbinRule.V5 = rule[5]
|
||||||
|
}
|
||||||
|
|
||||||
|
return casbinRule
|
||||||
|
}
|
||||||
|
|
||||||
|
// getFieldName returns the field name for the given index
|
||||||
|
func getFieldName(index int) string {
|
||||||
|
switch index {
|
||||||
|
case 0:
|
||||||
|
return "v0"
|
||||||
|
case 1:
|
||||||
|
return "v1"
|
||||||
|
case 2:
|
||||||
|
return "v2"
|
||||||
|
case 3:
|
||||||
|
return "v3"
|
||||||
|
case 4:
|
||||||
|
return "v4"
|
||||||
|
case 5:
|
||||||
|
return "v5"
|
||||||
|
default:
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Index20241226001UP 創建索引
|
||||||
|
func (a *CasbinAdapter) Index20241226001UP(ctx context.Context) (bool, error) {
|
||||||
|
indexes := []mongodriver.IndexModel{
|
||||||
|
{
|
||||||
|
Keys: bson.D{
|
||||||
|
{Key: "ptype", Value: 1},
|
||||||
|
},
|
||||||
|
Options: &mongodriver.IndexOptions{
|
||||||
|
Name: &[]string{"idx_ptype"}[0],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Keys: bson.D{
|
||||||
|
{Key: "ptype", Value: 1},
|
||||||
|
{Key: "v0", Value: 1},
|
||||||
|
},
|
||||||
|
Options: &mongodriver.IndexOptions{
|
||||||
|
Name: &[]string{"idx_ptype_v0"}[0],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Keys: bson.D{
|
||||||
|
{Key: "ptype", Value: 1},
|
||||||
|
{Key: "v0", Value: 1},
|
||||||
|
{Key: "v1", Value: 1},
|
||||||
|
},
|
||||||
|
Options: &mongodriver.IndexOptions{
|
||||||
|
Name: &[]string{"idx_ptype_v0_v1"}[0],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// 需要轉換為 mongo.DocumentDBWithCacheUseCase 的 CreateIndexes 方法
|
||||||
|
// 這裡簡化處理，實際需要根據你的 mongo 包裝實現
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,196 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/library/errs/code"
|
||||||
|
"backend/pkg/permission/domain"
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"backend/pkg/library/errs"
|
||||||
|
"backend/pkg/library/mongo"
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"backend/pkg/permission/domain/repository"
|
||||||
|
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/cache"
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/mon"
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
mongodriver "go.mongodb.org/mongo-driver/v2/mongo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ClientRepositoryParam struct {
|
||||||
|
Conf *mongo.Conf
|
||||||
|
CacheConf cache.CacheConf
|
||||||
|
DBOpts []mon.Option
|
||||||
|
CacheOpts []cache.Option
|
||||||
|
}
|
||||||
|
|
||||||
|
type ClientRepository struct {
|
||||||
|
DB mongo.DocumentDBWithCacheUseCase
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewClientRepository 創建客戶端倉庫實例
|
||||||
|
func NewClientRepository(param ClientRepositoryParam) repository.ClientRepository {
|
||||||
|
e := entity.Client{}
|
||||||
|
documentDB, err := mongo.MustDocumentDBWithCache(
|
||||||
|
param.Conf,
|
||||||
|
e.CollectionName(),
|
||||||
|
param.CacheConf,
|
||||||
|
param.DBOpts,
|
||||||
|
param.CacheOpts,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &ClientRepository{
|
||||||
|
DB: documentDB,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *ClientRepository) Create(ctx context.Context, client *entity.Client) error {
|
||||||
|
now := time.Now()
|
||||||
|
client.CreateTime = now
|
||||||
|
client.UpdateTime = now
|
||||||
|
id := bson.NewObjectID()
|
||||||
|
client.ID = id
|
||||||
|
rk := domain.GeClientRedisKey(id.Hex())
|
||||||
|
_, err := repo.DB.InsertOne(ctx, rk, client)
|
||||||
|
if err != nil {
|
||||||
|
// 檢查是否為重複鍵錯誤
|
||||||
|
if mongodriver.IsDuplicateKeyError(err) {
|
||||||
|
return errs.ResourceAlreadyExist(client.ClientID)
|
||||||
|
}
|
||||||
|
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *ClientRepository) GetByID(ctx context.Context, id string) (*entity.Client, error) {
|
||||||
|
var client entity.Client
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
rk := domain.GeClientRedisKey(objID.Hex())
|
||||||
|
err = repo.DB.FindOne(ctx, rk, &client, bson.M{"_id": objID})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, mongodriver.ErrNoDocuments) {
|
||||||
|
return nil, errs.ResourceNotFoundWithScope(
|
||||||
|
code.CloudEPPermission,
|
||||||
|
domain.FailedToGetByID,
|
||||||
|
"failed to get client by id")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &client, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *ClientRepository) GetByClientID(ctx context.Context, clientID string) (*entity.Client, error) {
|
||||||
|
var client entity.Client
|
||||||
|
rk := domain.GeClientRedisKey(clientID)
|
||||||
|
err := repo.DB.FindOne(ctx, rk, &client, bson.M{"client_id": clientID})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, mongodriver.ErrNoDocuments) {
|
||||||
|
return nil, errs.ResourceNotFoundWithScope(
|
||||||
|
code.CloudEPPermission,
|
||||||
|
domain.FailedToGetByClientID,
|
||||||
|
"failed to get client by client id")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &client, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *ClientRepository) Update(ctx context.Context, id string, client *entity.Client) error {
|
||||||
|
client.UpdateTime = time.Now()
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
update := bson.M{
|
||||||
|
"$set": bson.M{
|
||||||
|
"name": client.Name,
|
||||||
|
"secret": client.Secret,
|
||||||
|
"status": client.Status,
|
||||||
|
"update_time": client.UpdateTime,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
gc, err := repo.GetByID(ctx, id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
rk := domain.GeClientRedisKey(objID.Hex())
|
||||||
|
_, err = repo.DB.UpdateOne(ctx, rk, bson.M{"_id": objID}, update)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
rk = domain.GeClientRedisKey(gc.ClientID)
|
||||||
|
err = repo.DB.DelCache(ctx, rk)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *ClientRepository) Delete(ctx context.Context, id string) error {
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
gc, err := repo.GetByID(ctx, id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
rk := domain.GeClientRedisKey(gc.ClientID)
|
||||||
|
err = repo.DB.DelCache(ctx, rk)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
rk = domain.GeClientRedisKey(objID.Hex())
|
||||||
|
_, err = repo.DB.DeleteOne(ctx, rk, bson.M{"_id": objID})
|
||||||
|
if err != nil {
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *ClientRepository) List(ctx context.Context, filter repository.ClientFilter) ([]*entity.Client, error) {
|
||||||
|
query := bson.M{}
|
||||||
|
|
||||||
|
if filter.Status != nil {
|
||||||
|
query["status"] = *filter.Status
|
||||||
|
}
|
||||||
|
|
||||||
|
var clients []*entity.Client
|
||||||
|
err := repo.DB.GetClient().Find(ctx, query, &clients,
|
||||||
|
options.Find().SetLimit(int64(filter.Limit)),
|
||||||
|
options.Find().SetSkip(int64(filter.Skip)),
|
||||||
|
)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return clients, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Index20241226001UP 創建索引
|
||||||
|
func (repo *ClientRepository) Index20241226001UP(ctx context.Context) (*mongodriver.Cursor, error) {
|
||||||
|
repo.DB.PopulateIndex(ctx, "client_id", 1, true)
|
||||||
|
repo.DB.PopulateIndex(ctx, "status", 1, false)
|
||||||
|
|
||||||
|
return repo.DB.GetClient().Indexes().List(ctx)
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,209 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/library/errs/code"
|
||||||
|
"backend/pkg/permission/domain"
|
||||||
|
"backend/pkg/permission/domain/permission"
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"backend/pkg/library/errs"
|
||||||
|
"backend/pkg/library/mongo"
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"backend/pkg/permission/domain/repository"
|
||||||
|
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/cache"
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/mon"
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
mongodriver "go.mongodb.org/mongo-driver/v2/mongo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type PermissionRepositoryParam struct {
|
||||||
|
Conf *mongo.Conf
|
||||||
|
CacheConf cache.CacheConf
|
||||||
|
DBOpts []mon.Option
|
||||||
|
CacheOpts []cache.Option
|
||||||
|
}
|
||||||
|
|
||||||
|
type PermissionRepository struct {
|
||||||
|
DB mongo.DocumentDBWithCacheUseCase
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewPermissionRepository 創建權限倉庫實例
|
||||||
|
func NewPermissionRepository(param PermissionRepositoryParam) repository.PermissionRepository {
|
||||||
|
e := entity.Permission{}
|
||||||
|
documentDB, err := mongo.MustDocumentDBWithCache(
|
||||||
|
param.Conf,
|
||||||
|
e.CollectionName(),
|
||||||
|
param.CacheConf,
|
||||||
|
param.DBOpts,
|
||||||
|
param.CacheOpts,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &PermissionRepository{
|
||||||
|
DB: documentDB,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *PermissionRepository) Create(ctx context.Context, permission *entity.Permission) error {
|
||||||
|
now := time.Now()
|
||||||
|
permission.CreateTime = now
|
||||||
|
permission.UpdateTime = now
|
||||||
|
|
||||||
|
id := bson.NewObjectID()
|
||||||
|
permission.ID = id
|
||||||
|
|
||||||
|
rk := domain.GetPermissionRedisKey(id.Hex())
|
||||||
|
_, err := repo.DB.InsertOne(ctx, rk, permission)
|
||||||
|
if err != nil {
|
||||||
|
// 檢查是否為重複鍵錯誤
|
||||||
|
if mongodriver.IsDuplicateKeyError(err) {
|
||||||
|
return errs.ResourceAlreadyExist(permission.ID.Hex())
|
||||||
|
}
|
||||||
|
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *PermissionRepository) GetByID(ctx context.Context, id string) (*entity.Permission, error) {
|
||||||
|
var p entity.Permission
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
rk := domain.GetPermissionRedisKey(objID.Hex())
|
||||||
|
err = repo.DB.FindOne(ctx, rk, &p, bson.M{"_id": objID})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, mongodriver.ErrNoDocuments) {
|
||||||
|
return nil, errs.ResourceNotFoundWithScope(
|
||||||
|
code.CloudEPPermission,
|
||||||
|
domain.FailedToGetPermission,
|
||||||
|
"failed to get permission by id")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &p, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *PermissionRepository) GetByKey(ctx context.Context, httpMethod, httpPath string) (*entity.Permission, error) {
|
||||||
|
filter := bson.M{
|
||||||
|
"http_method": httpMethod,
|
||||||
|
"http_path": httpPath,
|
||||||
|
"status": permission.StatusActive,
|
||||||
|
}
|
||||||
|
|
||||||
|
var p entity.Permission
|
||||||
|
err := repo.DB.GetClient().FindOne(ctx, &p, filter)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, mongodriver.ErrNoDocuments) {
|
||||||
|
return nil, errs.ResourceNotFoundWithScope(
|
||||||
|
code.CloudEPPermission, domain.FailedToGetPermissionByKey,
|
||||||
|
"failed to get permission by key")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
return &p, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *PermissionRepository) Update(ctx context.Context, id string, permission *entity.Permission) error {
|
||||||
|
permission.UpdateTime = time.Now()
|
||||||
|
update := bson.M{
|
||||||
|
"$set": bson.M{
|
||||||
|
"parent_id": permission.ParentID,
|
||||||
|
"name": permission.Name,
|
||||||
|
"http_method": permission.HTTPMethod,
|
||||||
|
"http_path": permission.HTTPPath,
|
||||||
|
"status": permission.Status,
|
||||||
|
"type": permission.Type,
|
||||||
|
"update_time": permission.UpdateTime,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
rk := domain.GetPermissionRedisKey(id)
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = repo.DB.UpdateOne(ctx, rk, bson.M{"_id": objID}, update)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *PermissionRepository) Delete(ctx context.Context, id string) error {
|
||||||
|
rk := domain.GetPermissionRedisKey(id)
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
_, err = repo.DB.DeleteOne(ctx, rk, bson.M{"_id": objID})
|
||||||
|
if err != nil {
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *PermissionRepository) List(ctx context.Context, filter repository.PermissionFilter) ([]*entity.Permission, error) {
|
||||||
|
query := bson.M{}
|
||||||
|
|
||||||
|
if filter.Status != nil {
|
||||||
|
query["status"] = *filter.Status
|
||||||
|
}
|
||||||
|
if filter.Type != nil {
|
||||||
|
query["type"] = *filter.Type
|
||||||
|
}
|
||||||
|
if filter.ParentID != nil {
|
||||||
|
query["parent_id"] = *filter.ParentID
|
||||||
|
}
|
||||||
|
|
||||||
|
var permissions []*entity.Permission
|
||||||
|
err := repo.DB.GetClient().Find(ctx,
|
||||||
|
&permissions, query,
|
||||||
|
options.Find().SetLimit(int64(filter.Limit)),
|
||||||
|
options.Find().SetSkip(int64(filter.Skip)))
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return permissions, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *PermissionRepository) GetActivePermissions(ctx context.Context) ([]*entity.Permission, error) {
|
||||||
|
status := permission.StatusActive
|
||||||
|
filter := repository.PermissionFilter{
|
||||||
|
Status: &status,
|
||||||
|
}
|
||||||
|
|
||||||
|
return repo.List(ctx, filter)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Index20241226001UP 創建索引
|
||||||
|
func (repo *PermissionRepository) Index20241226001UP(ctx context.Context) (*mongodriver.Cursor, error) {
|
||||||
|
// 等價於 db.account.createIndex({ "login_id": 1, "platform": 1}, {unique: true})
|
||||||
|
repo.DB.PopulateMultiIndex(ctx, []string{
|
||||||
|
"http_method",
|
||||||
|
"http_path",
|
||||||
|
}, []int32{1, 1}, true)
|
||||||
|
|
||||||
|
// 等價於 db.account.createIndex({"create_at": 1})
|
||||||
|
repo.DB.PopulateIndex(ctx, "name", 1, false)
|
||||||
|
repo.DB.PopulateIndex(ctx, "status", 1, false)
|
||||||
|
repo.DB.PopulateIndex(ctx, "type", 1, false)
|
||||||
|
|
||||||
|
return repo.DB.GetClient().Indexes().List(ctx)
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,233 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/library/errs/code"
|
||||||
|
"backend/pkg/permission/domain"
|
||||||
|
"backend/pkg/permission/domain/permission"
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"backend/pkg/library/errs"
|
||||||
|
"backend/pkg/library/mongo"
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"backend/pkg/permission/domain/repository"
|
||||||
|
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/cache"
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/mon"
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
mongodriver "go.mongodb.org/mongo-driver/v2/mongo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type RoleRepositoryParam struct {
|
||||||
|
Conf *mongo.Conf
|
||||||
|
CacheConf cache.CacheConf
|
||||||
|
DBOpts []mon.Option
|
||||||
|
CacheOpts []cache.Option
|
||||||
|
}
|
||||||
|
|
||||||
|
type RoleRepository struct {
|
||||||
|
DB mongo.DocumentDBWithCacheUseCase
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewRoleRepository 創建角色倉庫實例
|
||||||
|
func NewRoleRepository(param RoleRepositoryParam) repository.RoleRepository {
|
||||||
|
e := entity.Role{}
|
||||||
|
documentDB, err := mongo.MustDocumentDBWithCache(
|
||||||
|
param.Conf,
|
||||||
|
e.CollectionName(),
|
||||||
|
param.CacheConf,
|
||||||
|
param.DBOpts,
|
||||||
|
param.CacheOpts,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &RoleRepository{
|
||||||
|
DB: documentDB,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *RoleRepository) Create(ctx context.Context, role *entity.Role) error {
|
||||||
|
now := time.Now()
|
||||||
|
role.CreateTime = now
|
||||||
|
role.UpdateTime = now
|
||||||
|
id := bson.NewObjectID()
|
||||||
|
role.ID = id
|
||||||
|
|
||||||
|
rk := domain.GetRoleRedisKeyRedisKey(id.Hex())
|
||||||
|
_, err := repo.DB.InsertOne(ctx, rk, role)
|
||||||
|
if err != nil {
|
||||||
|
// 檢查是否為重複鍵錯誤
|
||||||
|
if mongodriver.IsDuplicateKeyError(err) {
|
||||||
|
return errs.ResourceAlreadyExist(role.ClientID)
|
||||||
|
}
|
||||||
|
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *RoleRepository) GetByID(ctx context.Context, id string) (*entity.Role, error) {
|
||||||
|
var role entity.Role
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
rk := domain.GetRoleRedisKeyRedisKey(id)
|
||||||
|
err = repo.DB.FindOne(ctx, rk, &role, bson.M{"client_id": objID})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, mongodriver.ErrNoDocuments) {
|
||||||
|
return nil, errs.ResourceNotFoundWithScope(
|
||||||
|
code.CloudEPPermission,
|
||||||
|
domain.FailedToGetRoleByID,
|
||||||
|
"failed to get role by id")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &role, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *RoleRepository) GetByUID(ctx context.Context, uid string) (*entity.Role, error) {
|
||||||
|
var role entity.Role
|
||||||
|
rk := domain.GetRoleRedisKeyRedisKey(uid)
|
||||||
|
err := repo.DB.FindOne(ctx, rk, &role, bson.M{"uid": uid, "status": permission.StatusActive})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, mongodriver.ErrNoDocuments) {
|
||||||
|
return nil, errs.ResourceNotFoundWithScope(
|
||||||
|
code.CloudEPPermission,
|
||||||
|
domain.FailedToGetByUID,
|
||||||
|
"failed to get role by uid")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &role, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *RoleRepository) GetByClientAndName(ctx context.Context, clientID, name string) (*entity.Role, error) {
|
||||||
|
filter := bson.M{
|
||||||
|
"client_id": clientID,
|
||||||
|
"name": name,
|
||||||
|
"status": permission.StatusActive,
|
||||||
|
}
|
||||||
|
|
||||||
|
var role entity.Role
|
||||||
|
err := repo.DB.GetClient().FindOne(ctx, &role, filter)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, mongodriver.ErrNoDocuments) {
|
||||||
|
return nil, errs.ResourceNotFoundWithScope(
|
||||||
|
code.CloudEPPermission, domain.FailedToGetByClientAndName, "failed to get by client and name")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &role, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *RoleRepository) Update(ctx context.Context, id string, role *entity.Role) error {
|
||||||
|
role.UpdateTime = time.Now()
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
update := bson.M{
|
||||||
|
"$set": bson.M{
|
||||||
|
"name": role.Name,
|
||||||
|
"status": role.Status,
|
||||||
|
"permissions": role.Permissions,
|
||||||
|
"update_time": role.UpdateTime,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
rk := domain.GetRoleRedisKeyRedisKey(id)
|
||||||
|
|
||||||
|
_, err = repo.DB.UpdateOne(ctx, rk, bson.M{"_id": objID}, update)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *RoleRepository) Delete(ctx context.Context, id string) error {
|
||||||
|
rk := domain.GetRoleRedisKeyRedisKey(id)
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
gc, err := repo.GetByID(ctx, id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
rk = domain.GetRoleRedisKeyRedisKey(gc.UID)
|
||||||
|
err = repo.DB.DelCache(ctx, rk)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = repo.DB.DeleteOne(ctx, rk, bson.M{"_id": objID})
|
||||||
|
if err != nil {
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *RoleRepository) List(ctx context.Context, filter repository.RoleFilter) ([]*entity.Role, error) {
|
||||||
|
query := bson.M{}
|
||||||
|
|
||||||
|
if filter.ClientID != "" {
|
||||||
|
query["client_id"] = filter.ClientID
|
||||||
|
}
|
||||||
|
if filter.Status != nil {
|
||||||
|
query["status"] = *filter.Status
|
||||||
|
}
|
||||||
|
|
||||||
|
var roles []*entity.Role
|
||||||
|
err := repo.DB.GetClient().Find(ctx, &roles, query,
|
||||||
|
options.Find().SetLimit(int64(filter.Limit)),
|
||||||
|
options.Find().SetSkip(int64(filter.Skip)),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return roles, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *RoleRepository) GetRolesByClientID(ctx context.Context, clientID string) ([]*entity.Role, error) {
|
||||||
|
status := permission.StatusActive
|
||||||
|
filter := repository.RoleFilter{
|
||||||
|
ClientID: clientID,
|
||||||
|
Status: &status,
|
||||||
|
}
|
||||||
|
|
||||||
|
return repo.List(ctx, filter)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Index20241226001UP 創建索引
|
||||||
|
func (repo *RoleRepository) Index20241226001UP(ctx context.Context) (*mongodriver.Cursor, error) {
|
||||||
|
// 等價於 db.account.createIndex({ "login_id": 1, "platform": 1}, {unique: true})
|
||||||
|
repo.DB.PopulateMultiIndex(ctx, []string{
|
||||||
|
"client_id",
|
||||||
|
"name",
|
||||||
|
}, []int32{1, 1}, true)
|
||||||
|
|
||||||
|
// 等價於 db.account.createIndex({"create_at": 1})
|
||||||
|
repo.DB.PopulateIndex(ctx, "uid", 1, true)
|
||||||
|
repo.DB.PopulateIndex(ctx, "status", 1, false)
|
||||||
|
|
||||||
|
return repo.DB.GetClient().Indexes().List(ctx)
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,145 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/library/errs"
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"backend/pkg/permission/domain/repository"
|
||||||
|
"context"
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/redis"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Token Repository Implementation
|
||||||
|
|
||||||
|
type TokenRepositoryParam struct {
|
||||||
|
Redis *redis.Redis
|
||||||
|
}
|
||||||
|
|
||||||
|
type TokenRepository struct {
|
||||||
|
Redis *redis.Redis
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewTokenRepository 創建令牌倉庫實例
|
||||||
|
func NewTokenRepository(param TokenRepositoryParam) repository.TokenRepository {
|
||||||
|
return &TokenRepository{
|
||||||
|
Redis: param.Redis,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *TokenRepository) Create(ctx context.Context, token *entity.Token) error {
|
||||||
|
// 驗證數據
|
||||||
|
if err := token.Validate(); err != nil {
|
||||||
|
return errs.InvalidFormat(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
token.CreateTime = time.Now()
|
||||||
|
token.UpdateTime = time.Now()
|
||||||
|
|
||||||
|
// 在 Redis 中存儲 access token
|
||||||
|
accessKey := "token:access:" + token.AccessToken
|
||||||
|
refreshKey := "token:refresh:" + token.RefreshToken
|
||||||
|
|
||||||
|
// 設置過期時間
|
||||||
|
expiry := int(time.Until(token.ExpiresAt).Seconds())
|
||||||
|
if expiry <= 0 {
|
||||||
|
return errs.InvalidFormat("token already expired")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 存儲 access token
|
||||||
|
err := r.Redis.SetexCtx(ctx, accessKey, token.UID+":"+token.ClientID+":"+token.DeviceID, expiry)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// 存儲 refresh token (較長的過期時間)
|
||||||
|
refreshExpiry := expiry * 7 // refresh token 過期時間是 access token 的 7 倍
|
||||||
|
err = r.Redis.SetexCtx(ctx, refreshKey, token.UID+":"+token.ClientID+":"+token.DeviceID, refreshExpiry)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *TokenRepository) GetByAccessToken(ctx context.Context, accessToken string) (*entity.Token, error) {
|
||||||
|
key := "token:access:" + accessToken
|
||||||
|
value, err := r.Redis.GetCtx(ctx, key)
|
||||||
|
if err != nil {
|
||||||
|
if err == redis.Nil {
|
||||||
|
return nil, errs.NotFound("access_token")
|
||||||
|
}
|
||||||
|
return nil, errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析值
|
||||||
|
parts := strings.Split(value, ":")
|
||||||
|
if len(parts) != 3 {
|
||||||
|
return nil, errs.InvalidFormat("invalid token format")
|
||||||
|
}
|
||||||
|
|
||||||
|
return &entity.Token{
|
||||||
|
UID: parts[0],
|
||||||
|
ClientID: parts[1],
|
||||||
|
DeviceID: parts[2],
|
||||||
|
AccessToken: accessToken,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *TokenRepository) GetByRefreshToken(ctx context.Context, refreshToken string) (*entity.Token, error) {
|
||||||
|
key := "token:refresh:" + refreshToken
|
||||||
|
value, err := r.Redis.GetCtx(ctx, key)
|
||||||
|
if err != nil {
|
||||||
|
if err == redis.Nil {
|
||||||
|
return nil, errs.NotFound("refresh_token")
|
||||||
|
}
|
||||||
|
return nil, errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析值
|
||||||
|
parts := strings.Split(value, ":")
|
||||||
|
if len(parts) != 3 {
|
||||||
|
return nil, errs.InvalidFormat("invalid token format")
|
||||||
|
}
|
||||||
|
|
||||||
|
return &entity.Token{
|
||||||
|
UID: parts[0],
|
||||||
|
ClientID: parts[1],
|
||||||
|
DeviceID: parts[2],
|
||||||
|
RefreshToken: refreshToken,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *TokenRepository) Update(ctx context.Context, token *entity.Token) error {
|
||||||
|
// 驗證數據
|
||||||
|
if err := token.Validate(); err != nil {
|
||||||
|
return errs.InvalidFormat(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
token.UpdateTime = time.Now()
|
||||||
|
|
||||||
|
// 重新存儲 access token
|
||||||
|
accessKey := "token:access:" + token.AccessToken
|
||||||
|
expiry := int(time.Until(token.ExpiresAt).Seconds())
|
||||||
|
if expiry <= 0 {
|
||||||
|
return errs.InvalidFormat("token already expired")
|
||||||
|
}
|
||||||
|
|
||||||
|
err := r.Redis.SetexCtx(ctx, accessKey, token.UID+":"+token.ClientID+":"+token.DeviceID, expiry)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DatabaseErr(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *TokenRepository) Delete(ctx context.Context, id bson.ObjectID) error {
|
||||||
|
// Redis 版本不需要 ObjectID，這裡留空實現
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *TokenRepository) DeleteByUserID(ctx context.Context, uid string) error {
|
||||||
|
// 可以實現刪除用戶所有 token 的邏輯
|
||||||
|
// 這裡簡化實現
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,238 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/library/errs/code"
|
||||||
|
"backend/pkg/permission/domain"
|
||||||
|
"backend/pkg/permission/domain/permission"
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"backend/pkg/library/errs"
|
||||||
|
"backend/pkg/library/mongo"
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"backend/pkg/permission/domain/repository"
|
||||||
|
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/cache"
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/mon"
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
mongodriver "go.mongodb.org/mongo-driver/v2/mongo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserRoleRepositoryParam struct {
|
||||||
|
Conf *mongo.Conf
|
||||||
|
CacheConf cache.CacheConf
|
||||||
|
DBOpts []mon.Option
|
||||||
|
CacheOpts []cache.Option
|
||||||
|
}
|
||||||
|
|
||||||
|
type UserRoleRepository struct {
|
||||||
|
DB mongo.DocumentDBWithCacheUseCase
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewUserRoleRepository 創建用戶角色倉庫實例
|
||||||
|
func NewUserRoleRepository(param UserRoleRepositoryParam) repository.UserRoleRepository {
|
||||||
|
e := entity.UserRole{}
|
||||||
|
documentDB, err := mongo.MustDocumentDBWithCache(
|
||||||
|
param.Conf,
|
||||||
|
e.CollectionName(),
|
||||||
|
param.CacheConf,
|
||||||
|
param.DBOpts,
|
||||||
|
param.CacheOpts,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &UserRoleRepository{
|
||||||
|
DB: documentDB,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *UserRoleRepository) Create(ctx context.Context, userRole *entity.UserRole) error {
|
||||||
|
now := time.Now()
|
||||||
|
userRole.CreateTime = now
|
||||||
|
userRole.UpdateTime = now
|
||||||
|
id := bson.NewObjectID()
|
||||||
|
userRole.ID = id
|
||||||
|
|
||||||
|
rk := domain.GetUserRoleRedisKey(id.Hex())
|
||||||
|
userRole.CreateTime = time.Now()
|
||||||
|
userRole.UpdateTime = time.Now()
|
||||||
|
|
||||||
|
_, err := repo.DB.InsertOne(ctx, rk, userRole)
|
||||||
|
if err != nil {
|
||||||
|
// 檢查是否為重複鍵錯誤
|
||||||
|
if mongodriver.IsDuplicateKeyError(err) {
|
||||||
|
return errs.ResourceAlreadyExist("failed to insert user role")
|
||||||
|
}
|
||||||
|
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *UserRoleRepository) GetByID(ctx context.Context, id string) (*entity.UserRole, error) {
|
||||||
|
var userRole entity.UserRole
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
rk := domain.GetUserRoleRedisKey(id)
|
||||||
|
err = repo.DB.FindOne(ctx, rk, &userRole, bson.M{"_id": objID})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, mongodriver.ErrNoDocuments) {
|
||||||
|
return nil, errs.ResourceNotFoundWithScope(
|
||||||
|
code.CloudEPPermission,
|
||||||
|
domain.FailedToGetRoleByID,
|
||||||
|
"failed to get user role by id")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &userRole, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *UserRoleRepository) GetByUserAndRole(ctx context.Context, uid, roleUID string) (*entity.UserRole, error) {
|
||||||
|
filter := bson.M{
|
||||||
|
"uid": uid,
|
||||||
|
"role_uid": roleUID,
|
||||||
|
"status": permission.StatusActive,
|
||||||
|
}
|
||||||
|
|
||||||
|
var userRole entity.UserRole
|
||||||
|
err := repo.DB.GetClient().Find(ctx, &userRole, filter)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, mongodriver.ErrNoDocuments) {
|
||||||
|
return nil, errs.ResourceNotFoundWithScope(code.CloudEPPermission, 0, "failed to get user and role")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errs.DatabaseErrorWithScope(code.CloudEPPermission, 0, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &userRole, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *UserRoleRepository) Update(ctx context.Context, id string, userRole *entity.UserRole) error {
|
||||||
|
userRole.UpdateTime = time.Now()
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
update := bson.M{
|
||||||
|
"$set": bson.M{
|
||||||
|
"status": userRole.Status,
|
||||||
|
"update_time": userRole.UpdateTime,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
rk := domain.GetUserRoleRedisKey(id)
|
||||||
|
|
||||||
|
_, err = repo.DB.UpdateOne(ctx, rk, bson.M{"_id": objID}, update)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *UserRoleRepository) Delete(ctx context.Context, id string) error {
|
||||||
|
objID, err := bson.ObjectIDFromHex(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
rk := domain.GetUserRoleRedisKey(id)
|
||||||
|
_, err = repo.DB.DeleteOne(ctx, rk, bson.M{"_id": objID})
|
||||||
|
if err != nil {
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *UserRoleRepository) List(ctx context.Context, filter repository.UserRoleFilter) ([]*entity.UserRole, error) {
|
||||||
|
query := bson.M{}
|
||||||
|
if filter.Brand != "" {
|
||||||
|
query["brand"] = filter.Brand
|
||||||
|
}
|
||||||
|
if filter.UID != "" {
|
||||||
|
query["uid"] = filter.UID
|
||||||
|
}
|
||||||
|
if filter.RoleUID != "" {
|
||||||
|
query["role_uid"] = filter.RoleUID
|
||||||
|
}
|
||||||
|
if filter.Status != nil {
|
||||||
|
query["status"] = *filter.Status
|
||||||
|
}
|
||||||
|
|
||||||
|
var userRoles []*entity.UserRole
|
||||||
|
err := repo.DB.GetClient().Find(ctx, &userRoles, query)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
err = repo.DB.GetClient().Find(ctx,
|
||||||
|
&userRoles, query,
|
||||||
|
options.Find().SetLimit(int64(filter.Limit)),
|
||||||
|
options.Find().SetSkip(int64(filter.Skip)))
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return userRoles, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *UserRoleRepository) GetUserRolesByUID(ctx context.Context, uid string) ([]*entity.UserRole, error) {
|
||||||
|
status := permission.StatusActive
|
||||||
|
filter := repository.UserRoleFilter{
|
||||||
|
UID: uid,
|
||||||
|
Status: &status,
|
||||||
|
}
|
||||||
|
|
||||||
|
return repo.List(ctx, filter)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (repo *UserRoleRepository) DeleteByUserAndRole(ctx context.Context, uid, roleUID string) error {
|
||||||
|
filter := repository.UserRoleFilter{
|
||||||
|
UID: uid,
|
||||||
|
RoleUID: roleUID,
|
||||||
|
}
|
||||||
|
list, err := repo.List(ctx, filter)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if len(list) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, item := range list {
|
||||||
|
_ = repo.DB.DelCache(ctx, domain.GetUserRoleRedisKey(item.ID.Hex()))
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = repo.DB.GetClient().DeleteMany(ctx, filter)
|
||||||
|
if err != nil {
|
||||||
|
return errs.DBErrorWithScope(code.CloudEPPermission, err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Index20241226001UP 創建索引
|
||||||
|
func (repo *UserRoleRepository) Index20241226001UP(ctx context.Context) (*mongodriver.Cursor, error) {
|
||||||
|
// 等價於 db.account.createIndex({ "login_id": 1, "platform": 1}, {unique: true})
|
||||||
|
repo.DB.PopulateMultiIndex(ctx, []string{
|
||||||
|
"uid",
|
||||||
|
"role_uid",
|
||||||
|
}, []int32{1, 1}, true)
|
||||||
|
|
||||||
|
// 等價於 db.account.createIndex({"create_at": 1})
|
||||||
|
repo.DB.PopulateIndex(ctx, "uid", 1, false)
|
||||||
|
repo.DB.PopulateIndex(ctx, "status", 1, false)
|
||||||
|
|
||||||
|
return repo.DB.GetClient().Indexes().List(ctx)
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,207 @@
|
||||||
|
package usecase
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/library/errs/code"
|
||||||
|
"backend/pkg/permission/utils"
|
||||||
|
"context"
|
||||||
|
"crypto/rand"
|
||||||
|
"encoding/hex"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"backend/pkg/library/errs"
|
||||||
|
"backend/pkg/permission/domain/config"
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"backend/pkg/permission/domain/repository"
|
||||||
|
"backend/pkg/permission/domain/usecase"
|
||||||
|
|
||||||
|
"github.com/golang-jwt/jwt/v5"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AuthUseCaseParam struct {
|
||||||
|
ClientRepo repository.ClientRepository
|
||||||
|
TokenRepo repository.TokenRepository
|
||||||
|
JWTConfig config.JWTConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
type AuthUseCase struct {
|
||||||
|
clientRepo repository.ClientRepository
|
||||||
|
tokenRepo repository.TokenRepository
|
||||||
|
jwtConfig config.JWTConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
// MustAuthUseCase 創建認證用例實例
|
||||||
|
func MustAuthUseCase(param AuthUseCaseParam) usecase.AuthUseCase {
|
||||||
|
return &AuthUseCase{
|
||||||
|
clientRepo: param.ClientRepo,
|
||||||
|
tokenRepo: param.TokenRepo,
|
||||||
|
jwtConfig: param.JWTConfig,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *AuthUseCase) CreateToken(ctx context.Context, req usecase.CreateTokenRequest) (*usecase.TokenResponse, error) {
|
||||||
|
// 驗證客戶端
|
||||||
|
client, err := uc.clientRepo.GetByClientID(ctx, req.ClientID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !utils.IsActive(client.Status) {
|
||||||
|
return nil, errs.UserSuspended(code.CloudEPPermission, "failed to get token since user has been suspended")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根據授權類型處理
|
||||||
|
var uid string
|
||||||
|
switch req.GrantType {
|
||||||
|
case "client_credentials":
|
||||||
|
uid = "client_" + req.ClientID
|
||||||
|
case "password":
|
||||||
|
if req.Username == "" || req.Password == "" {
|
||||||
|
return nil, errs.InvalidCredentials()
|
||||||
|
}
|
||||||
|
// 這裡應該驗證用戶名密碼，簡化處理
|
||||||
|
uid = req.Username
|
||||||
|
default:
|
||||||
|
return nil, errs.InvalidFormat("unsupported grant type: " + req.GrantType)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 生成令牌
|
||||||
|
accessToken, err := uc.generateAccessToken(uid, req.ClientID, req.DeviceID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.SystemInternal("failed to generate access token: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
refreshToken, err := uc.generateRefreshToken()
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.SystemInternal("failed to generate refresh token: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// 保存令牌
|
||||||
|
token := &entity.Token{
|
||||||
|
UID: uid,
|
||||||
|
ClientID: req.ClientID,
|
||||||
|
AccessToken: accessToken,
|
||||||
|
RefreshToken: refreshToken,
|
||||||
|
DeviceID: req.DeviceID,
|
||||||
|
ExpiresAt: time.Now().Add(uc.jwtConfig.AccessExpires),
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := uc.tokenRepo.Create(ctx, token); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &usecase.TokenResponse{
|
||||||
|
AccessToken: accessToken,
|
||||||
|
RefreshToken: refreshToken,
|
||||||
|
TokenType: "Bearer",
|
||||||
|
ExpiresIn: int64(uc.jwtConfig.AccessExpires.Seconds()),
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *AuthUseCase) RefreshToken(ctx context.Context, refreshToken string) (*usecase.TokenResponse, error) {
|
||||||
|
// 查找刷新令牌
|
||||||
|
token, err := uc.tokenRepo.GetByRefreshToken(ctx, refreshToken)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if token.IsExpired() {
|
||||||
|
return nil, errs.TokenExpired()
|
||||||
|
}
|
||||||
|
|
||||||
|
// 生成新的訪問令牌
|
||||||
|
accessToken, err := uc.generateAccessToken(token.UID, token.ClientID, token.DeviceID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.SystemInternal("failed to generate access token: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新令牌
|
||||||
|
token.AccessToken = accessToken
|
||||||
|
token.ExpiresAt = time.Now().Add(uc.jwtConfig.AccessExpires)
|
||||||
|
|
||||||
|
if err := uc.tokenRepo.Update(ctx, token); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &usecase.TokenResponse{
|
||||||
|
AccessToken: accessToken,
|
||||||
|
RefreshToken: refreshToken,
|
||||||
|
TokenType: "Bearer",
|
||||||
|
ExpiresIn: int64(uc.jwtConfig.AccessExpires.Seconds()),
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *AuthUseCase) ValidateToken(ctx context.Context, accessToken string) (*usecase.TokenClaims, error) {
|
||||||
|
// 解析JWT令牌
|
||||||
|
token, err := jwt.Parse(accessToken, func(token *jwt.Token) (interface{}, error) {
|
||||||
|
if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
|
||||||
|
return nil, errs.TokenInvalid()
|
||||||
|
}
|
||||||
|
return []byte(uc.jwtConfig.Secret), nil
|
||||||
|
})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.TokenInvalid()
|
||||||
|
}
|
||||||
|
|
||||||
|
if !token.Valid {
|
||||||
|
return nil, errs.TokenInvalid()
|
||||||
|
}
|
||||||
|
|
||||||
|
claims, ok := token.Claims.(jwt.MapClaims)
|
||||||
|
if !ok {
|
||||||
|
return nil, errs.TokenInvalid()
|
||||||
|
}
|
||||||
|
|
||||||
|
uid, ok := claims["uid"].(string)
|
||||||
|
if !ok {
|
||||||
|
return nil, errs.TokenInvalid()
|
||||||
|
}
|
||||||
|
|
||||||
|
clientID, ok := claims["client_id"].(string)
|
||||||
|
if !ok {
|
||||||
|
return nil, errs.TokenInvalid()
|
||||||
|
}
|
||||||
|
|
||||||
|
deviceID, _ := claims["device_id"].(string)
|
||||||
|
|
||||||
|
return &usecase.TokenClaims{
|
||||||
|
UID: uid,
|
||||||
|
ClientID: clientID,
|
||||||
|
DeviceID: deviceID,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *AuthUseCase) Logout(ctx context.Context, accessToken string) error {
|
||||||
|
// 查找並刪除令牌
|
||||||
|
token, err := uc.tokenRepo.GetByAccessToken(ctx, accessToken)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return uc.tokenRepo.Delete(ctx, token.ID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *AuthUseCase) LogoutAllByUserID(ctx context.Context, uid string) error {
|
||||||
|
return uc.tokenRepo.DeleteByUserID(ctx, uid)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *AuthUseCase) generateAccessToken(uid, clientID, deviceID string) (string, error) {
|
||||||
|
claims := jwt.MapClaims{
|
||||||
|
"uid": uid,
|
||||||
|
"client_id": clientID,
|
||||||
|
"device_id": deviceID,
|
||||||
|
"exp": time.Now().Add(uc.jwtConfig.AccessExpires).Unix(),
|
||||||
|
"iat": time.Now().Unix(),
|
||||||
|
}
|
||||||
|
|
||||||
|
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
||||||
|
return token.SignedString([]byte(uc.jwtConfig.Secret))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *AuthUseCase) generateRefreshToken() (string, error) {
|
||||||
|
bytes := make([]byte, 32)
|
||||||
|
if _, err := rand.Read(bytes); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return hex.EncodeToString(bytes), nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,348 @@
|
||||||
|
package usecase
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/library/errs/code"
|
||||||
|
"context"
|
||||||
|
"github.com/zeromicro/go-zero/core/logx"
|
||||||
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||||||
|
|
||||||
|
"backend/pkg/library/errs"
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"backend/pkg/permission/domain/repository"
|
||||||
|
"backend/pkg/permission/domain/usecase"
|
||||||
|
|
||||||
|
"github.com/casbin/casbin/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
type PermissionUseCaseParam struct {
|
||||||
|
Enforcer *casbin.Enforcer
|
||||||
|
PermissionRepo repository.PermissionRepository
|
||||||
|
RoleRepo repository.RoleRepository
|
||||||
|
UserRoleRepo repository.UserRoleRepository
|
||||||
|
}
|
||||||
|
|
||||||
|
type PermissionUseCase struct {
|
||||||
|
enforcer *casbin.Enforcer
|
||||||
|
permissionRepo repository.PermissionRepository
|
||||||
|
roleRepo repository.RoleRepository
|
||||||
|
userRoleRepo repository.UserRoleRepository
|
||||||
|
}
|
||||||
|
|
||||||
|
// MustPermissionUseCase 創建權限用例實例
|
||||||
|
func MustPermissionUseCase(param PermissionUseCaseParam) usecase.PermissionUseCase {
|
||||||
|
return &PermissionUseCase{
|
||||||
|
enforcer: param.Enforcer,
|
||||||
|
permissionRepo: param.PermissionRepo,
|
||||||
|
roleRepo: param.RoleRepo,
|
||||||
|
userRoleRepo: param.UserRoleRepo,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *PermissionUseCase) CreatePermission(ctx context.Context, req usecase.CreatePermissionRequest) (*entity.Permission, error) {
|
||||||
|
// 驗證請求
|
||||||
|
if req.Name == "" {
|
||||||
|
return nil, errs.InvalidFormat("permission name is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
permission := &entity.Permission{
|
||||||
|
Name: req.Name,
|
||||||
|
HTTPMethod: req.HTTPMethod,
|
||||||
|
HTTPPath: req.HTTPPath,
|
||||||
|
Status: req.Status,
|
||||||
|
Type: req.Type,
|
||||||
|
}
|
||||||
|
|
||||||
|
if req.ParentID != nil {
|
||||||
|
objID, err := bson.ObjectIDFromHex(*req.ParentID)
|
||||||
|
if err != nil {
|
||||||
|
e := errs.InvalidFormat(err.Error())
|
||||||
|
return nil, e
|
||||||
|
}
|
||||||
|
permission.ID = objID
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := uc.permissionRepo.Create(ctx, permission); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return permission, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *PermissionUseCase) GetPermission(ctx context.Context, id string) (*entity.Permission, error) {
|
||||||
|
return uc.permissionRepo.GetByID(ctx, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *PermissionUseCase) UpdatePermission(ctx context.Context, req usecase.UpdatePermissionRequest) (*entity.Permission, error) {
|
||||||
|
// 獲取現有權限
|
||||||
|
permission, err := uc.permissionRepo.GetByID(ctx, req.ID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新字段
|
||||||
|
if req.Name != nil {
|
||||||
|
permission.Name = *req.Name
|
||||||
|
}
|
||||||
|
if req.HTTPMethod != nil {
|
||||||
|
permission.HTTPMethod = *req.HTTPMethod
|
||||||
|
}
|
||||||
|
if req.HTTPPath != nil {
|
||||||
|
permission.HTTPPath = *req.HTTPPath
|
||||||
|
}
|
||||||
|
if req.Status != nil {
|
||||||
|
permission.Status = *req.Status
|
||||||
|
}
|
||||||
|
if req.Type != nil {
|
||||||
|
permission.Type = *req.Type
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := uc.permissionRepo.Update(ctx, req.ID, permission); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return permission, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *PermissionUseCase) DeletePermission(ctx context.Context, id string) error {
|
||||||
|
return uc.permissionRepo.Delete(ctx, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *PermissionUseCase) ListPermissions(ctx context.Context, req usecase.ListPermissionsRequest) ([]*entity.Permission, error) {
|
||||||
|
filter := repository.PermissionFilter{
|
||||||
|
Status: req.Status,
|
||||||
|
Type: req.Type,
|
||||||
|
ParentID: req.ParentID,
|
||||||
|
Limit: req.Limit,
|
||||||
|
Skip: req.Skip,
|
||||||
|
}
|
||||||
|
|
||||||
|
return uc.permissionRepo.List(ctx, filter)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CheckUserPermission 使用 Casbin 檢查用戶權限
|
||||||
|
func (uc *PermissionUseCase) CheckUserPermission(ctx context.Context, uid, httpMethod, httpPath string) (bool, error) {
|
||||||
|
// 使用 Casbin 進行權限檢查
|
||||||
|
// sub: 用戶ID, obj: 資源路徑, act: 行為
|
||||||
|
hasPermission, err := uc.enforcer.Enforce(uid, httpPath, httpMethod)
|
||||||
|
if err != nil {
|
||||||
|
return false, errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin enforce failed: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if !hasPermission {
|
||||||
|
return false, errs.InsufficientPermission(httpMethod + ":" + httpPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// CheckRolePermission 使用 Casbin 檢查角色權限
|
||||||
|
func (uc *PermissionUseCase) CheckRolePermission(ctx context.Context, roleUID, httpMethod, httpPath string) (bool, error) {
|
||||||
|
// 使用 Casbin 進行角色權限檢查
|
||||||
|
hasPermission, err := uc.enforcer.Enforce(roleUID, httpPath, httpMethod)
|
||||||
|
if err != nil {
|
||||||
|
return false, errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin enforce failed: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if !hasPermission {
|
||||||
|
return false, errs.InsufficientPermission(httpMethod + ":" + httpPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetUserPermissions 獲取用戶的所有權限
|
||||||
|
func (uc *PermissionUseCase) GetUserPermissions(ctx context.Context, uid string) (map[string]int, error) {
|
||||||
|
// 獲取用戶的所有角色
|
||||||
|
roles, err := uc.enforcer.GetRolesForUser(uid)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.SystemInternalErrorScope(code.CloudEPPermission, "failed to get user permissions: "+err.Error())
|
||||||
|
}
|
||||||
|
permissions := make(map[string]int)
|
||||||
|
|
||||||
|
// 獲取用戶直接擁有的權限
|
||||||
|
userPolicies, err := uc.enforcer.GetPermissionsForUser(uid)
|
||||||
|
if err != nil {
|
||||||
|
logx.Infof("failed to get user permissions: " + err.Error())
|
||||||
|
}
|
||||||
|
for _, policy := range userPolicies {
|
||||||
|
if len(policy) >= 3 {
|
||||||
|
key := policy[2] + ":" + policy[1] // method:path
|
||||||
|
permissions[key] = 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 獲取通過角色繼承的權限
|
||||||
|
for _, role := range roles {
|
||||||
|
rolePolicies, err := uc.enforcer.GetPermissionsForUser(role)
|
||||||
|
if err != nil {
|
||||||
|
logx.Infof("failed to get permissions for user: " + err.Error())
|
||||||
|
}
|
||||||
|
for _, policy := range rolePolicies {
|
||||||
|
if len(policy) >= 3 {
|
||||||
|
key := policy[2] + ":" + policy[1] // method:path
|
||||||
|
permissions[key] = 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return permissions, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// BatchCheckPermissions 批量檢查權限
|
||||||
|
func (uc *PermissionUseCase) BatchCheckPermissions(ctx context.Context, uid string, permissions []usecase.PermissionCheck) (map[string]bool, error) {
|
||||||
|
results := make(map[string]bool)
|
||||||
|
|
||||||
|
for _, perm := range permissions {
|
||||||
|
key := perm.HTTPMethod + ":" + perm.HTTPPath
|
||||||
|
hasPermission, err := uc.enforcer.Enforce(uid, perm.HTTPPath, perm.HTTPMethod)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin enforce failed: "+err.Error())
|
||||||
|
}
|
||||||
|
results[key] = hasPermission
|
||||||
|
}
|
||||||
|
|
||||||
|
return results, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddPolicyForUser 為用戶添加權限策略
|
||||||
|
func (uc *PermissionUseCase) AddPolicyForUser(ctx context.Context, uid, httpPath, httpMethod string) error {
|
||||||
|
added, err := uc.enforcer.AddPolicy(uid, httpPath, httpMethod)
|
||||||
|
if err != nil {
|
||||||
|
return errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin add policy failed: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if !added {
|
||||||
|
return errs.ResourceAlreadyExistWithScope(code.CloudEPPermission, "policy already exists")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemovePolicyForUser 移除用戶的權限策略
|
||||||
|
func (uc *PermissionUseCase) RemovePolicyForUser(ctx context.Context, uid, httpPath, httpMethod string) error {
|
||||||
|
removed, err := uc.enforcer.RemovePolicy(uid, httpPath, httpMethod)
|
||||||
|
if err != nil {
|
||||||
|
return errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin remove policy failed: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if !removed {
|
||||||
|
return errs.ResourceNotFoundWithScope(code.CloudEPPermission, 0, "policy not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddRoleForUser 為用戶分配角色
|
||||||
|
func (uc *PermissionUseCase) AddRoleForUser(ctx context.Context, uid, roleUID string) error {
|
||||||
|
added, err := uc.enforcer.AddRoleForUser(uid, roleUID)
|
||||||
|
if err != nil {
|
||||||
|
return errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin add role failed: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if !added {
|
||||||
|
return errs.ResourceAlreadyExistWithScope(code.CloudEPPermission, "role already assigned")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemoveRoleForUser 移除用戶的角色
|
||||||
|
func (uc *PermissionUseCase) RemoveRoleForUser(ctx context.Context, uid, roleUID string) error {
|
||||||
|
removed, err := uc.enforcer.DeleteRoleForUser(uid, roleUID)
|
||||||
|
if err != nil {
|
||||||
|
return errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin remove role failed: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if !removed {
|
||||||
|
return errs.ResourceNotFoundWithScope(code.CloudEPPermission, 0, "role assignment not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetUsersForRole 獲取角色下的所有用戶
|
||||||
|
func (uc *PermissionUseCase) GetUsersForRole(ctx context.Context, roleUID string) ([]string, error) {
|
||||||
|
return uc.enforcer.GetUsersForRole(roleUID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetRolesForUser 獲取用戶的所有角色
|
||||||
|
func (uc *PermissionUseCase) GetRolesForUser(ctx context.Context, uid string) ([]string, error) {
|
||||||
|
return uc.enforcer.GetRolesForUser(uid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddPermissionForRole 為角色添加權限
|
||||||
|
func (uc *PermissionUseCase) AddPermissionForRole(ctx context.Context, roleUID, httpPath, httpMethod string) error {
|
||||||
|
added, err := uc.enforcer.AddPolicy(roleUID, httpPath, httpMethod)
|
||||||
|
if err != nil {
|
||||||
|
return errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin add policy failed: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if !added {
|
||||||
|
return errs.ResourceAlreadyExistWithScope(code.CloudEPPermission, "policy already exists")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemovePermissionForRole 移除角色的權限
|
||||||
|
func (uc *PermissionUseCase) RemovePermissionForRole(ctx context.Context, roleUID, httpPath, httpMethod string) error {
|
||||||
|
removed, err := uc.enforcer.RemovePolicy(roleUID, httpPath, httpMethod)
|
||||||
|
if err != nil {
|
||||||
|
return errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin remove policy failed: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if !removed {
|
||||||
|
return errs.ResourceNotFoundWithScope(code.CloudEPPermission, 0, "policy not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetPermissionsForRole 獲取角色的所有權限
|
||||||
|
func (uc *PermissionUseCase) GetPermissionsForRole(ctx context.Context, roleUID string) (map[string]int, error) {
|
||||||
|
policies, err := uc.enforcer.GetPermissionsForUser(roleUID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin get permissions failed: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
permissions := make(map[string]int)
|
||||||
|
|
||||||
|
for _, policy := range policies {
|
||||||
|
if len(policy) >= 3 {
|
||||||
|
key := policy[2] + ":" + policy[1] // method:path
|
||||||
|
permissions[key] = 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return permissions, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// CheckPatternPermission 檢查模式權限 (支援通配符)
|
||||||
|
func (uc *PermissionUseCase) CheckPatternPermission(ctx context.Context, uid, pattern, action string) (bool, error) {
|
||||||
|
hasPermission, err := uc.enforcer.Enforce(uid, pattern, action)
|
||||||
|
if err != nil {
|
||||||
|
return false, errs.SystemInternalErrorScope(code.CloudEPPermission, "casbin enforce failed: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return hasPermission, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetAllPolicies 獲取所有策略
|
||||||
|
func (uc *PermissionUseCase) GetAllPolicies(ctx context.Context) ([][]string, error) {
|
||||||
|
policies, err := uc.enforcer.GetPolicy()
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.SystemInternalErrorScope(code.CloudEPPermission, "failed to get all policies: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return policies, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetFilteredPolicies 獲取過濾後的策略
|
||||||
|
func (uc *PermissionUseCase) GetFilteredPolicies(ctx context.Context, fieldIndex int, fieldValues ...string) ([][]string, error) {
|
||||||
|
policies, err := uc.enforcer.GetFilteredPolicy(fieldIndex, fieldValues...)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.SystemInternalErrorScope(code.CloudEPPermission, "failed to get filtered policies: "+err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return policies, nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,189 @@
|
||||||
|
package usecase
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/library/errs/code"
|
||||||
|
"backend/pkg/permission/domain/permission"
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"backend/pkg/library/errs"
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"backend/pkg/permission/domain/repository"
|
||||||
|
"backend/pkg/permission/domain/usecase"
|
||||||
|
)
|
||||||
|
|
||||||
|
type RoleUseCaseParam struct {
|
||||||
|
RoleRepo repository.RoleRepository
|
||||||
|
UserRoleRepo repository.UserRoleRepository
|
||||||
|
}
|
||||||
|
|
||||||
|
type RoleUseCase struct {
|
||||||
|
roleRepo repository.RoleRepository
|
||||||
|
userRoleRepo repository.UserRoleRepository
|
||||||
|
}
|
||||||
|
|
||||||
|
// MustRoleUseCase 創建角色用例實例
|
||||||
|
func MustRoleUseCase(param RoleUseCaseParam) usecase.RoleUseCase {
|
||||||
|
return &RoleUseCase{
|
||||||
|
roleRepo: param.RoleRepo,
|
||||||
|
userRoleRepo: param.UserRoleRepo,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) CreateRole(ctx context.Context, req usecase.CreateRoleRequest) (*entity.Role, error) {
|
||||||
|
// 驗證請求
|
||||||
|
if req.ClientID == "" {
|
||||||
|
return nil, errs.InvalidFormat("client_id is required")
|
||||||
|
}
|
||||||
|
if req.Name == "" {
|
||||||
|
return nil, errs.InvalidFormat("role name is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 檢查角色名稱是否已存在
|
||||||
|
existingRole, err := uc.roleRepo.GetByClientAndName(ctx, req.ClientID, req.Name)
|
||||||
|
if err == nil && existingRole != nil {
|
||||||
|
return nil, errs.ResourceAlreadyExistWithScope(code.CloudEPPermission, req.ClientID+":"+req.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
role := &entity.Role{
|
||||||
|
ClientID: req.ClientID,
|
||||||
|
UID: req.UID,
|
||||||
|
Name: req.Name,
|
||||||
|
Status: req.Status,
|
||||||
|
Permissions: req.Permissions,
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := uc.roleRepo.Create(ctx, role); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return role, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) GetRole(ctx context.Context, id string) (*entity.Role, error) {
|
||||||
|
return uc.roleRepo.GetByID(ctx, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) GetRoleByUID(ctx context.Context, uid string) (*entity.Role, error) {
|
||||||
|
return uc.roleRepo.GetByUID(ctx, uid)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) UpdateRole(ctx context.Context, req usecase.UpdateRoleRequest) (*entity.Role, error) {
|
||||||
|
// 獲取現有角色
|
||||||
|
role, err := uc.roleRepo.GetByID(ctx, req.ID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新字段
|
||||||
|
if req.Name != nil {
|
||||||
|
// 檢查新名稱是否已存在
|
||||||
|
existingRole, err := uc.roleRepo.GetByClientAndName(ctx, role.ClientID, *req.Name)
|
||||||
|
if err == nil && existingRole != nil && existingRole.ID != role.ID {
|
||||||
|
return nil, errs.ResourceAlreadyExistWithScope(code.CloudEPPermission, role.ClientID+":"+*req.Name)
|
||||||
|
}
|
||||||
|
role.Name = *req.Name
|
||||||
|
}
|
||||||
|
if req.Status != nil {
|
||||||
|
role.Status = *req.Status
|
||||||
|
}
|
||||||
|
if req.Permissions != nil {
|
||||||
|
role.Permissions = *req.Permissions
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := uc.roleRepo.Update(ctx, req.ID, role); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return role, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) DeleteRole(ctx context.Context, id string) error {
|
||||||
|
// 獲取角色信息
|
||||||
|
role, err := uc.roleRepo.GetByID(ctx, id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
status := permission.StatusActive
|
||||||
|
// 檢查是否有用戶使用此角色
|
||||||
|
userRoles, err := uc.userRoleRepo.List(ctx, repository.UserRoleFilter{
|
||||||
|
RoleUID: role.UID,
|
||||||
|
Status: &status,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(userRoles) > 0 {
|
||||||
|
return errs.InvalidFormat("cannot delete role that is assigned to users")
|
||||||
|
}
|
||||||
|
|
||||||
|
return uc.roleRepo.Delete(ctx, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) ListRoles(ctx context.Context, req usecase.ListRolesRequest) ([]*entity.Role, error) {
|
||||||
|
filter := repository.RoleFilter{
|
||||||
|
ClientID: req.ClientID,
|
||||||
|
Status: req.Status,
|
||||||
|
Limit: req.Limit,
|
||||||
|
Skip: req.Skip,
|
||||||
|
}
|
||||||
|
|
||||||
|
return uc.roleRepo.List(ctx, filter)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) AddPermissionToRole(ctx context.Context, roleID string, permissionKey string) error {
|
||||||
|
// 獲取角色
|
||||||
|
role, err := uc.roleRepo.GetByID(ctx, roleID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 添加權限
|
||||||
|
role.AddPermission(permissionKey)
|
||||||
|
return uc.roleRepo.Update(ctx, role.ID.Hex(), role)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) RemovePermissionFromRole(ctx context.Context, roleID string, permissionKey string) error {
|
||||||
|
// 獲取角色
|
||||||
|
role, err := uc.roleRepo.GetByID(ctx, roleID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 移除權限
|
||||||
|
role.RemovePermission(permissionKey)
|
||||||
|
|
||||||
|
return uc.roleRepo.Update(ctx, role.ID.Hex(), role)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) BatchUpdateRolePermissions(ctx context.Context, roleID string, permissions entity.Permissions) error {
|
||||||
|
// 獲取角色
|
||||||
|
role, err := uc.roleRepo.GetByID(ctx, roleID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 批量更新權限
|
||||||
|
role.Permissions = permissions
|
||||||
|
|
||||||
|
return uc.roleRepo.Update(ctx, role.ID.Hex(), role)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) GetRolesByClientID(ctx context.Context, clientID string) ([]*entity.Role, error) {
|
||||||
|
return uc.roleRepo.GetRolesByClientID(ctx, clientID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *RoleUseCase) CopyRole(ctx context.Context, sourceRoleID string, req usecase.CreateRoleRequest) (*entity.Role, error) {
|
||||||
|
// 獲取源角色
|
||||||
|
sourceRole, err := uc.roleRepo.GetByID(ctx, sourceRoleID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 創建新角色，複製權限
|
||||||
|
newReq := req
|
||||||
|
newReq.Permissions = sourceRole.Permissions
|
||||||
|
|
||||||
|
return uc.CreateRole(ctx, newReq)
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,225 @@
|
||||||
|
package usecase
|
||||||
|
|
||||||
|
import (
|
||||||
|
"backend/pkg/library/errs/code"
|
||||||
|
"backend/pkg/permission/domain/permission"
|
||||||
|
"backend/pkg/permission/utils"
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"backend/pkg/library/errs"
|
||||||
|
"backend/pkg/permission/domain/entity"
|
||||||
|
"backend/pkg/permission/domain/repository"
|
||||||
|
"backend/pkg/permission/domain/usecase"
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserRoleUseCaseParam struct {
|
||||||
|
UserRoleRepo repository.UserRoleRepository
|
||||||
|
RoleRepo repository.RoleRepository
|
||||||
|
}
|
||||||
|
|
||||||
|
type UserRoleUseCase struct {
|
||||||
|
userRoleRepo repository.UserRoleRepository
|
||||||
|
roleRepo repository.RoleRepository
|
||||||
|
}
|
||||||
|
|
||||||
|
// MustUserRoleUseCase 創建用戶角色用例實例
|
||||||
|
func MustUserRoleUseCase(param UserRoleUseCaseParam) usecase.UserRoleUseCase {
|
||||||
|
return &UserRoleUseCase{
|
||||||
|
userRoleRepo: param.UserRoleRepo,
|
||||||
|
roleRepo: param.RoleRepo,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *UserRoleUseCase) AssignRole(ctx context.Context, req usecase.AssignRoleRequest) (*entity.UserRole, error) {
|
||||||
|
// 驗證請求
|
||||||
|
if req.UID == "" {
|
||||||
|
return nil, errs.InvalidFormat("uid is required")
|
||||||
|
}
|
||||||
|
if req.RoleUID == "" {
|
||||||
|
return nil, errs.InvalidFormat("role_uid is required")
|
||||||
|
}
|
||||||
|
if req.Brand == "" {
|
||||||
|
return nil, errs.InvalidFormat("brand is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 檢查角色是否存在
|
||||||
|
role, err := uc.roleRepo.GetByUID(ctx, req.RoleUID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !utils.IsActive(role.Status) {
|
||||||
|
return nil, errs.InvalidFormat("role is not active")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 檢查用戶是否已經有此角色
|
||||||
|
existingUserRole, err := uc.userRoleRepo.GetByUserAndRole(ctx, req.UID, req.RoleUID)
|
||||||
|
if err == nil && existingUserRole != nil && utils.IsActive(existingUserRole.Status) {
|
||||||
|
return nil, errs.ResourceAlreadyExistWithScope(code.CloudEPPermission, req.UID+":"+req.RoleUID)
|
||||||
|
}
|
||||||
|
|
||||||
|
userRole := &entity.UserRole{
|
||||||
|
Brand: req.Brand,
|
||||||
|
UID: req.UID,
|
||||||
|
RoleUID: req.RoleUID,
|
||||||
|
Status: permission.StatusActive,
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := uc.userRoleRepo.Create(ctx, userRole); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return userRole, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *UserRoleUseCase) RevokeRole(ctx context.Context, uid, roleUID string) error {
|
||||||
|
// 驗證參數
|
||||||
|
if uid == "" {
|
||||||
|
return errs.InvalidFormat("uid is required")
|
||||||
|
}
|
||||||
|
if roleUID == "" {
|
||||||
|
return errs.InvalidFormat("role_uid is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
return uc.userRoleRepo.DeleteByUserAndRole(ctx, uid, roleUID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *UserRoleUseCase) GetUserRole(ctx context.Context, id string) (*entity.UserRole, error) {
|
||||||
|
return uc.userRoleRepo.GetByID(ctx, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *UserRoleUseCase) UpdateUserRole(ctx context.Context, req usecase.UpdateUserRoleRequest) (*entity.UserRole, error) {
|
||||||
|
// 獲取現有用戶角色
|
||||||
|
userRole, err := uc.userRoleRepo.GetByID(ctx, req.ID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新狀態
|
||||||
|
if req.Status != nil {
|
||||||
|
userRole.Status = *req.Status
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := uc.userRoleRepo.Update(ctx, req.ID, userRole); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return userRole, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *UserRoleUseCase) ListUserRoles(ctx context.Context, req usecase.ListUserRolesRequest) ([]*entity.UserRole, error) {
|
||||||
|
filter := repository.UserRoleFilter{
|
||||||
|
Brand: req.Brand,
|
||||||
|
UID: req.UID,
|
||||||
|
RoleUID: req.RoleUID,
|
||||||
|
Status: req.Status,
|
||||||
|
Limit: req.Limit,
|
||||||
|
Skip: req.Skip,
|
||||||
|
}
|
||||||
|
|
||||||
|
return uc.userRoleRepo.List(ctx, filter)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *UserRoleUseCase) GetUserRoles(ctx context.Context, uid string) ([]*entity.UserRole, error) {
|
||||||
|
if uid == "" {
|
||||||
|
return nil, errs.InvalidFormat("uid is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
return uc.userRoleRepo.GetUserRolesByUID(ctx, uid)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *UserRoleUseCase) GetUserRoleDetails(ctx context.Context, uid string) ([]*usecase.UserRoleDetail, error) {
|
||||||
|
// 獲取用戶角色
|
||||||
|
userRoles, err := uc.GetUserRoles(ctx, uid)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var details []*usecase.UserRoleDetail
|
||||||
|
for _, userRole := range userRoles {
|
||||||
|
if !utils.IsActive(userRole.Status) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// 獲取角色詳情
|
||||||
|
role, err := uc.roleRepo.GetByUID(ctx, userRole.RoleUID)
|
||||||
|
if err != nil {
|
||||||
|
continue // 忽略獲取失敗的角色
|
||||||
|
}
|
||||||
|
|
||||||
|
detail := &usecase.UserRoleDetail{
|
||||||
|
UserRole: userRole,
|
||||||
|
Role: role,
|
||||||
|
}
|
||||||
|
details = append(details, detail)
|
||||||
|
}
|
||||||
|
|
||||||
|
return details, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *UserRoleUseCase) BatchAssignRoles(ctx context.Context, uid string, roleUIDs []string, brand string) error {
|
||||||
|
if uid == "" {
|
||||||
|
return errs.InvalidFormat("uid is required")
|
||||||
|
}
|
||||||
|
if brand == "" {
|
||||||
|
return errs.InvalidFormat("brand is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 逐個分配角色
|
||||||
|
for _, roleUID := range roleUIDs {
|
||||||
|
req := usecase.AssignRoleRequest{
|
||||||
|
Brand: brand,
|
||||||
|
UID: uid,
|
||||||
|
RoleUID: roleUID,
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := uc.AssignRole(ctx, req)
|
||||||
|
if err != nil {
|
||||||
|
// 如果是已存在錯誤，忽略繼續
|
||||||
|
e := errs.FromError(err)
|
||||||
|
if e.Is(errs.ResourceAlreadyExist()) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *UserRoleUseCase) BatchRevokeRoles(ctx context.Context, uid string, roleUIDs []string) error {
|
||||||
|
if uid == "" {
|
||||||
|
return errs.InvalidFormat("uid is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 逐個撤銷角色
|
||||||
|
for _, roleUID := range roleUIDs {
|
||||||
|
err := uc.RevokeRole(ctx, uid, roleUID)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (uc *UserRoleUseCase) ReplaceUserRoles(ctx context.Context, uid string, roleUIDs []string, brand string) error {
|
||||||
|
// 獲取用戶當前的所有角色
|
||||||
|
currentUserRoles, err := uc.GetUserRoles(ctx, uid)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 撤銷所有現有角色
|
||||||
|
for _, userRole := range currentUserRoles {
|
||||||
|
if utils.IsActive(userRole.Status) {
|
||||||
|
if err := uc.RevokeRole(ctx, uid, userRole.RoleUID); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 分配新角色
|
||||||
|
return uc.BatchAssignRoles(ctx, uid, roleUIDs, brand)
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,7 @@
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import "backend/pkg/permission/domain/permission"
|
||||||
|
|
||||||
|
func IsActive(status int) bool {
|
||||||
|
return status == permission.StatusActive
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue