Packages

ejabberd

16.6.2
26.4.0 26.3.0 26.2.0 26.1.0 25.10.0 25.8.0 25.7.0 25.4.0 25.3.0 24.12.0 24.10.0 24.7.0 24.6.0 24.2.6 23.10.0 23.4.0 23.1.0 22.10.0 22.5.0 21.12.0 21.7.0 21.4.0 21.1.0 20.12.0 20.7.0 20.4.0 20.3.0 20.2.0 20.1.0 19.9.1 19.9.0 19.8.0 19.5.0 19.2.0 18.12.1 18.12.0 18.6.0 18.4.0 18.3.0 18.1.0 17.11.0 17.9.0 17.6.0 17.3.0 17.1.0 16.12.0-beta1 16.9.0 16.8.0 16.6.2 16.6.1 16.6.0 16.4.1 16.4.0 16.3.0 16.2.0 16.1.0-beta1

Robust, Ubiquitous and Massively Scalable Messaging Platform (XMPP, MQTT, SIP Server)

HexDocs

Current section

Files

Jump to
Readme
56 Versions
18 Dependencies
Dependants
1 Advisory
Files
Activity
ejabberd src mod_vcard_riak.erl
Raw

src/mod_vcard_riak.erl 

%%%-------------------------------------------------------------------

%%% @author Evgeny Khramtsov <ekhramtsov@process-one.net>

%%% @copyright (C) 2016, Evgeny Khramtsov

%%% @doc

%%%

%%% @end

%%% Created : 13 Apr 2016 by Evgeny Khramtsov <ekhramtsov@process-one.net>

%%%-------------------------------------------------------------------

-module(mod_vcard_riak).



-behaviour(mod_vcard).



%% API

-export([init/2, get_vcard/2, set_vcard/4, search/4, remove_user/2,

	 import/2]).



-include("jlib.hrl").

-include("mod_vcard.hrl").



%%%===================================================================

%%% API

%%%===================================================================

init(_Host, _Opts) ->

    ok.



get_vcard(LUser, LServer) ->

    case ejabberd_riak:get(vcard, vcard_schema(), {LUser, LServer}) of

        {ok, R} ->

            [R#vcard.vcard];

        {error, notfound} ->

            [];

        _ ->

            error

    end.



set_vcard(LUser, LServer, VCARD,

	  #vcard_search{user = {User, _},

			fn = FN,

			lfn = LFN,

			family = Family,

			lfamily = LFamily,

			given = Given,

			lgiven = LGiven,

			middle = Middle,

			lmiddle = LMiddle,

			nickname = Nickname,

			lnickname = LNickname,

			bday = BDay,

			lbday = LBDay,

			ctry = CTRY,

			lctry = LCTRY,

			locality = Locality,

			llocality = LLocality,

			email = EMail,

			lemail = LEMail,

			orgname = OrgName,

			lorgname = LOrgName,

			orgunit = OrgUnit,

			lorgunit = LOrgUnit}) ->

    US = {LUser, LServer},

    {atomic,

     ejabberd_riak:put(#vcard{us = US, vcard = VCARD},

		       vcard_schema(),

		       [{'2i', [{<<"user">>, User},

				{<<"luser">>, LUser},

				{<<"fn">>, FN},

				{<<"lfn">>, LFN},

				{<<"family">>, Family},

				{<<"lfamily">>, LFamily},

				{<<"given">>, Given},

				{<<"lgiven">>, LGiven},

				{<<"middle">>, Middle},

				{<<"lmiddle">>, LMiddle},

				{<<"nickname">>, Nickname},

				{<<"lnickname">>, LNickname},

				{<<"bday">>, BDay},

				{<<"lbday">>, LBDay},

				{<<"ctry">>, CTRY},

				{<<"lctry">>, LCTRY},

				{<<"locality">>, Locality},

				{<<"llocality">>, LLocality},

				{<<"email">>, EMail},

				{<<"lemail">>, LEMail},

				{<<"orgname">>, OrgName},

				{<<"lorgname">>, LOrgName},

				{<<"orgunit">>, OrgUnit},

				{<<"lorgunit">>, LOrgUnit}]}])}.



search(_LServer, _Data, _AllowReturnAll, _MaxMatch) ->

    [].



remove_user(LUser, LServer) ->

    {atomic, ejabberd_riak:delete(vcard, {LUser, LServer})}.



import(_LServer, #vcard{us = {LUser, LServer}, vcard = El} = VCard) ->

    #vcard_search{fn = FN,

		  lfn = LFN,

		  family = Family,

		  lfamily = LFamily,

		  given = Given,

		  lgiven = LGiven,

		  middle = Middle,

		  lmiddle = LMiddle,

		  nickname = Nickname,

		  lnickname = LNickname,

		  bday = BDay,

		  lbday = LBDay,

		  ctry = CTRY,

		  lctry = LCTRY,

		  locality = Locality,

		  llocality = LLocality,

		  email = EMail,

		  lemail = LEMail,

		  orgname = OrgName,

		  lorgname = LOrgName,

		  orgunit = OrgUnit,

		  lorgunit = LOrgUnit} =

	mod_vcard:make_vcard_search(LUser, LUser, LServer, El),

    ejabberd_riak:put(VCard, vcard_schema(),

                      [{'2i', [{<<"user">>, LUser},

                               {<<"luser">>, LUser},

                               {<<"fn">>, FN},

                               {<<"lfn">>, LFN},

                               {<<"family">>, Family},

                               {<<"lfamily">>, LFamily},

                               {<<"given">>, Given},

                               {<<"lgiven">>, LGiven},

                               {<<"middle">>, Middle},

                               {<<"lmiddle">>, LMiddle},

                               {<<"nickname">>, Nickname},

                               {<<"lnickname">>, LNickname},

                               {<<"bday">>, BDay},

                               {<<"lbday">>, LBDay},

                               {<<"ctry">>, CTRY},

                               {<<"lctry">>, LCTRY},

                               {<<"locality">>, Locality},

                               {<<"llocality">>, LLocality},

                               {<<"email">>, EMail},

                               {<<"lemail">>, LEMail},

                               {<<"orgname">>, OrgName},

                               {<<"lorgname">>, LOrgName},

                               {<<"orgunit">>, OrgUnit},

                               {<<"lorgunit">>, LOrgUnit}]}]);

import(_LServer, #vcard_search{}) ->

    ok.



%%%===================================================================

%%% Internal functions

%%%===================================================================

vcard_schema() ->

    {record_info(fields, vcard), #vcard{}}.