/*
 * Juick
 * Copyright (C) 2008-2011, Ugnich Anton
 *
 * 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/>.
 */
package com.juick.json;

import com.juick.Message;
import java.text.DateFormat;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.*;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

/**
 *
 * @author Ugnich Anton
 */
public class MessageSerializer extends JSONSerializer<Message> {
    
    UserSerializer userSerializer = new UserSerializer();
    PlaceSerializer placeSerializer = new PlaceSerializer();

    @Override
    public Message deserialize(JSONObject json) throws JSONException {
        com.juick.Message jmsg = new com.juick.Message();
        jmsg.setMID(json.getInt("mid"));
        if (json.has("rid")) {
            jmsg.setRID(json.getInt("rid"));
        }
        if (json.has("replyto")) {
            jmsg.ReplyTo = json.getInt("replyto");
        }

        jmsg.FriendsOnly = json.has("friendsonly");
        jmsg.ReadOnly = json.has("readonly");

        jmsg.Text = json.getString("body").replace("&quot;", "\"");
        jmsg.setUser(userSerializer.deserialize(json.getJSONObject("user")));

        try {
            DateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            df.setTimeZone(TimeZone.getTimeZone("UTC"));
            jmsg.Timestamp = df.parse(json.getString("timestamp"));
        } catch (ParseException e) {
        }

        if (json.has("tags")) {
            JSONArray tags = json.getJSONArray("tags");
            for (int n = 0; n < tags.length(); n++) {
                jmsg.Tags.add(tags.getString(n).replace("&quot;", "\""));
            }
        }

        if (json.has("replies")) {
            jmsg.Replies = json.getInt("replies");
        }

        if (json.has("photo")) {
            jmsg.Photo = json.getJSONObject("photo").getString("small");
        }

        return jmsg;
    }

    @Override
    public JSONObject serialize(Message msg) {
        JSONObject json = new JSONObject();

        try {
            if (msg.getMID() > 0) {
                json.put("mid", msg.getMID());
            }
            if (msg.getRID() > 0) {
                json.put("rid", msg.getRID());
            }
            if (msg.ReplyTo > 0) {
                json.put("replyto", msg.ReplyTo);
            }
            if (msg.FriendsOnly) {
                json.put("friendsonly", 1);
            }
            if (msg.ReadOnly) {
                json.put("readonly", 1);
            }
            if (msg.Text != null) {
                json.put("body", msg.Text);
            }
            if (msg.Timestamp != null) {
                json.put("timestamp", msg.Timestamp);
            } else if (msg.TimestampString != null) {
                json.put("timestamp", msg.TimestampString);
            }
            if (msg.getUser() != null) {
                json.put("user", userSerializer.serialize(msg.getUser()));
            }
            if (msg.Tags != null && msg.Tags.size() > 0) {
                json.put("tags", new JSONArray(msg.Tags));
            }
            if (msg.Replies > 0) {
                json.put("replies", msg.Replies);
            }
            if (msg.Place != null) {
                json.put("place", placeSerializer.serialize(msg.Place));
            }
            if (msg.AttachmentType != null) {
                String fname = msg.getMID() + (msg.getRID() > 0 ? "-" + msg.getRID() : "") + "." + msg.AttachmentType;
                JSONObject photo = new JSONObject();
                photo.put("thumbnail", "http://i.juick.com/ps/" + fname);
                photo.put("small", "http://i.juick.com/photos-512/" + fname);
                photo.put("medium", "http://i.juick.com/photos-1024/" + fname);
                json.put("photo", photo);
            }
        } catch (JSONException e) {
        }

        return json;
    }
}