kvs_acl.erl 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. -module(kvs_acl).
  2. -copyright('Synrc Research Center s.r.o.').
  3. -compile(export_all).
  4. -include_lib("kvs/include/acls.hrl").
  5. -include_lib("kvs/include/users.hrl").
  6. -include_lib("kvs/include/groups.hrl").
  7. -include_lib("kvs/include/feeds.hrl").
  8. define_access(Accessor, Resource, Action) ->
  9. Entry = #acl_entry{ id = {Accessor, Resource},
  10. accessor= Accessor,
  11. action = Action,
  12. feed_id = Resource},
  13. case kvs:add(Entry) of {error, exist} -> kvs:put(Entry#acl_entry{action=Action}); {ok, E} -> E end.
  14. check(Keys) ->
  15. Acls = [Acl || {ok, Acl = #acl_entry{}} <- [kvs:get(acl_entry, Key) || Key <- Keys]],
  16. case Acls of [] -> none;
  17. [#acl_entry{action = Action} | _] -> Action end.
  18. check_access(#user{id = UId, type = UType}, #feed{id = FId}) ->
  19. Feed = {feed, FId},
  20. Query = [ {{user, UId}, Feed}, {{user_type, UType}, Feed}, {default, Feed}],
  21. check(Query);
  22. check_access(#user{id = UId, type = UType}, #group{id = GId}) ->
  23. Group = {group, GId},
  24. Query = [ {{user, UId}, Group}, {{user_type, UType}, Group}, {default, Group}],
  25. check(Query);
  26. check_access(#user{id = AId, type = AType}, #user{id = RId}) ->
  27. User = {user, RId},
  28. Query = [ {{user, AId}, User}, {{user_type, AType}, User}, {default, User} ],
  29. check(Query);
  30. check_access({user_type, Type}, #user{id = RId}) ->
  31. User = {user, RId},
  32. Query = [ {{user_type, Type}, User}, {default, User} ],
  33. check(Query);
  34. check_access({user_type, Type}, #feed{id = FId}) ->
  35. Feed = {feed, FId},
  36. Query = [ {{user_type, Type}, Feed}, {default, Feed} ],
  37. check(Query);
  38. check_access({user_type, Type}, #group{id = GId}) ->
  39. Group = {group, GId},
  40. Query = [{{user_type, Type}, Group}, {default, Group}],
  41. check(Query);
  42. check_access({ip, _Ip} = Accessor, {feature, _Feature} = Resource) ->
  43. Query = [{Accessor, Resource}, {default, Resource}],
  44. check(Query);
  45. check_access(#user{id = AId, type = AType}, {feature, _Feature} = R) ->
  46. Query = [ {{user, AId}, R}, {{user_type, AType}, R}, {default, R} ],
  47. check(Query);
  48. check_access(UId, {feature, _Feature} = Resource) ->
  49. case kvs:get(user, UId) of
  50. {ok, User} -> check_access(User, Resource);
  51. E -> E
  52. end.