162 lines
4.6 KiB
Erlang
162 lines
4.6 KiB
Erlang
%%%----------------------------------------------------------------------
|
|
%%% File : translate.erl
|
|
%%% Author : Alexey Shchepin <alexey@process-one.net>
|
|
%%% Purpose : Localization helper
|
|
%%% Created : 6 Jan 2003 by Alexey Shchepin <alexey@process-one.net>
|
|
%%%
|
|
%%%
|
|
%%% ejabberd, Copyright (C) 2002-2013 ProcessOne
|
|
%%%
|
|
%%% This program is free software; you can redistribute it and/or
|
|
%%% modify it under the terms of the GNU General Public License as
|
|
%%% published by the Free Software Foundation; either version 2 of the
|
|
%%% License, or (at your option) any later version.
|
|
%%%
|
|
%%% This program is distributed in the hope that it will be useful,
|
|
%%% but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
%%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
%%% General Public License for more details.
|
|
%%%
|
|
%%% You should have received a copy of the GNU General Public License
|
|
%%% along with this program; if not, write to the Free Software
|
|
%%% Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
|
|
%%% 02111-1307 USA
|
|
%%%
|
|
%%%----------------------------------------------------------------------
|
|
|
|
-module(translate).
|
|
|
|
-author('alexey@process-one.net').
|
|
|
|
-export([start/0, load_dir/1, load_file/2,
|
|
translate/2]).
|
|
|
|
-include("ejabberd.hrl").
|
|
-include("logger.hrl").
|
|
|
|
start() ->
|
|
ets:new(translations, [named_table, public]),
|
|
Dir = case os:getenv("EJABBERD_MSGS_PATH") of
|
|
false ->
|
|
case code:priv_dir(ejabberd) of
|
|
{error, _} -> ?MSGS_DIR;
|
|
Path -> filename:join([Path, "msgs"])
|
|
end;
|
|
Path -> Path
|
|
end,
|
|
load_dir(iolist_to_binary(Dir)),
|
|
ok.
|
|
|
|
-spec load_dir(binary()) -> ok.
|
|
|
|
load_dir(Dir) ->
|
|
case file:list_dir(Dir) of
|
|
{ok, Files} ->
|
|
MsgFiles = lists:filter(fun (FN) ->
|
|
case length(FN) > 4 of
|
|
true ->
|
|
string:substr(FN, length(FN) - 3)
|
|
== ".msg";
|
|
_ -> false
|
|
end
|
|
end,
|
|
Files),
|
|
lists:foreach(fun (FNS) ->
|
|
FN = list_to_binary(FNS),
|
|
LP = ascii_tolower(str:substr(FN, 1,
|
|
byte_size(FN) - 4)),
|
|
L = case str:tokens(LP, <<".">>) of
|
|
[Language] -> Language;
|
|
[Language, _Project] -> Language
|
|
end,
|
|
load_file(L, <<Dir/binary, "/", FN/binary>>)
|
|
end,
|
|
MsgFiles),
|
|
ok;
|
|
{error, Reason} -> ?ERROR_MSG("~p", [Reason])
|
|
end.
|
|
|
|
load_file(Lang, File) ->
|
|
case file:consult(File) of
|
|
{ok, Terms} ->
|
|
lists:foreach(fun ({Orig, Trans}) ->
|
|
Trans1 = case Trans of
|
|
<<"">> -> Orig;
|
|
_ -> Trans
|
|
end,
|
|
ets:insert(translations,
|
|
{{Lang, iolist_to_binary(Orig)},
|
|
iolist_to_binary(Trans1)})
|
|
end,
|
|
Terms);
|
|
%% Code copied from ejabberd_config.erl
|
|
{error,
|
|
{_LineNumber, erl_parse, _ParseMessage} = Reason} ->
|
|
ExitText = iolist_to_binary([File,
|
|
" approximately in the line ",
|
|
file:format_error(Reason)]),
|
|
?ERROR_MSG("Problem loading translation file ~n~s",
|
|
[ExitText]),
|
|
exit(ExitText);
|
|
{error, Reason} ->
|
|
ExitText = iolist_to_binary([File, ": ",
|
|
file:format_error(Reason)]),
|
|
?ERROR_MSG("Problem loading translation file ~n~s",
|
|
[ExitText]),
|
|
exit(ExitText)
|
|
end.
|
|
|
|
-spec translate(binary(), binary()) -> binary().
|
|
|
|
translate(Lang, Msg) ->
|
|
LLang = ascii_tolower(Lang),
|
|
case ets:lookup(translations, {LLang, Msg}) of
|
|
[{_, Trans}] -> Trans;
|
|
_ ->
|
|
ShortLang = case str:tokens(LLang, <<"-">>) of
|
|
[] -> LLang;
|
|
[SL | _] -> SL
|
|
end,
|
|
case ShortLang of
|
|
<<"en">> -> Msg;
|
|
LLang -> translate(Msg);
|
|
_ ->
|
|
case ets:lookup(translations, {ShortLang, Msg}) of
|
|
[{_, Trans}] -> Trans;
|
|
_ -> translate(Msg)
|
|
end
|
|
end
|
|
end.
|
|
|
|
translate(Msg) ->
|
|
case ?MYLANG of
|
|
<<"en">> -> Msg;
|
|
Lang ->
|
|
LLang = ascii_tolower(Lang),
|
|
case ets:lookup(translations, {LLang, Msg}) of
|
|
[{_, Trans}] -> Trans;
|
|
_ ->
|
|
ShortLang = case str:tokens(LLang, <<"-">>) of
|
|
[] -> LLang;
|
|
[SL | _] -> SL
|
|
end,
|
|
case ShortLang of
|
|
<<"en">> -> Msg;
|
|
Lang -> Msg;
|
|
_ ->
|
|
case ets:lookup(translations, {ShortLang, Msg}) of
|
|
[{_, Trans}] -> Trans;
|
|
_ -> Msg
|
|
end
|
|
end
|
|
end
|
|
end.
|
|
|
|
ascii_tolower(B) ->
|
|
iolist_to_binary(ascii_tolower_s(binary_to_list(B))).
|
|
|
|
ascii_tolower_s([C | Cs]) when C >= $A, C =< $Z ->
|
|
[C + ($a - $A) | ascii_tolower_s(Cs)];
|
|
ascii_tolower_s([C | Cs]) -> [C | ascii_tolower_s(Cs)];
|
|
ascii_tolower_s([]) -> [].
|