/ucengine/src/models/uce_file.erl

%%
%%  U.C.Engine - Unified Collaboration Engine
%%  Copyright (C) 2011 af83
%%
%%  This program is free software: you can redistribute it and/or modify
%%  it under the terms of the GNU Affero General Public License as published by
%%  the Free Software Foundation, either version 3 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 Affero General Public License for more details.
%%
%%  You should have received a copy of the GNU Affero General Public License
%%  along with this program.  If not, see <http://www.gnu.org/licenses/>.
%%
-module(uce_file).

-export([add/2, list/3, get/2, delete/2]).

-include("uce.hrl").

add(Domain, #uce_file{location=Location, name=Name} = File) ->
    case uce_meeting:exists(Domain, Location) of
        false ->
            throw({error, not_found});
        true ->
            Extension = filename:extension(Name),
            Base64Name = binary_to_list(base64:encode(unicode:characters_to_binary(Name))),
            Rnd = utils:random(),
            {Id, Mime} =
                case Extension of
                    "" ->
                        {Base64Name ++ "_" ++ Rnd,
                         "text/plain"};
                    _ ->
                        {Base64Name ++ "_" ++ Rnd ++ Extension,
                         yaws_api:mime_type(Name)}
                end,
            (db:get(?MODULE, Domain)):add(Domain, File#uce_file{id=Id, mime=Mime})
    end.

list(Domain, Id, Order) ->
    case uce_meeting:exists(Domain, Id) of
        false ->
            throw({error, not_found});
        true ->
            (db:get(?MODULE, Domain)):list(Domain, Id, Order)
    end.

get(Domain, Id) ->
    (db:get(?MODULE, Domain)):get(Domain, Id).

delete(Domain, Id) ->
    case get(Domain, Id) of
        {error, Reason} ->
            {error, Reason};
        {ok, _} ->
            (db:get(?MODULE, Domain)):delete(Domain, Id)
    end.