org.entcore.timeline.services.impl.FlashMsgServiceSqlImpl.java Source code

Introduction

Here is the source code for org.entcore.timeline.services.impl.FlashMsgServiceSqlImpl.java

Source

/*
 * Copyright  "Open Digital Education", 2018
 *
 * This program is published by "Open Digital Education".
 * You must indicate the name of the software and the company in any production /contribution
 * using the software and indicate on the home page of the software industry in question,
 * "powered by Open Digital Education" with a reference to the website: https://opendigitaleducation.com/.
 *
 * This program is free software, licensed under the terms of the GNU Affero General Public License
 * as published by the Free Software Foundation, version 3 of the License.
 *
 * You can redistribute this application and/or modify it since you respect the terms of the GNU Affero General Public License.
 * If you modify the source code and then use this modified source code in your creation, you must make available the source code of your modifications.
 *
 * You should have received a copy of the GNU Affero General Public License along with the software.
 * If not, please see : <http://www.gnu.org/licenses/>. Full compliance requires reading the terms of this license and following its directives.
 */

package org.entcore.timeline.services.impl;

import static org.entcore.common.sql.Sql.parseId;
import static org.entcore.common.sql.SqlResult.*;

import java.util.List;

import org.entcore.common.service.impl.SqlCrudService;
import org.entcore.common.sql.Sql;
import org.entcore.common.user.UserInfos;
import org.entcore.timeline.services.FlashMsgService;
import io.vertx.core.Handler;
import io.vertx.core.json.JsonArray;
import io.vertx.core.json.JsonObject;
import io.vertx.core.logging.LoggerFactory;
import io.vertx.core.logging.Logger;

import fr.wseduc.webutils.Either;

public class FlashMsgServiceSqlImpl extends SqlCrudService implements FlashMsgService {

    private final String JOIN_TABLE;
    private final Logger log = LoggerFactory.getLogger(FlashMsgServiceSqlImpl.class);

    public FlashMsgServiceSqlImpl(String schema, String table) {
        super(schema, table);
        JOIN_TABLE = schema + ".messages_read";
    }

    @Override
    public void create(JsonObject data, Handler<Either<String, JsonObject>> handler) {
        sql.insert(resourceTable, data, validUniqueResultHandler(handler));
    }

    @Override
    public void update(String id, JsonObject data, Handler<Either<String, JsonObject>> handler) {
        StringBuilder sb = new StringBuilder();
        JsonArray values = new fr.wseduc.webutils.collections.JsonArray();
        for (String attr : data.fieldNames()) {
            if ("startDate".equals(attr) || "endDate".equals(attr)) {
                sb.append("\"" + attr + "\"").append(" = ?::timestamptz, ");
            } else if ("contents".equals(attr) || "profiles".equals(attr)) {
                sb.append("\"" + attr + "\"").append(" = ?::jsonb, ");
            } else {
                sb.append("\"" + attr + "\"").append(" = ?, ");
            }
            values.add(data.getValue(attr));
        }
        String query = "UPDATE " + resourceTable + " SET " + sb.toString() + "modified = NOW() " + "WHERE id = ? ";
        sql.prepared(query, values.add(parseId(id)), validRowsResultHandler(handler));
    }

    @Override
    public void deleteMultiple(List<String> ids, Handler<Either<String, JsonObject>> handler) {
        String query = "DELETE FROM " + resourceTable + " WHERE id IN " + Sql.listPrepared(ids.toArray());
        JsonArray values = new fr.wseduc.webutils.collections.JsonArray();
        for (String id : ids) {
            try {
                long idNb = Long.parseLong(id);
                values.add(idNb);
            } catch (NumberFormatException e) {
                log.error("Bad id - not a number : " + id.toString());
            }
        }

        sql.prepared(query, values, validUniqueResultHandler(handler));
    }

    @Override
    public void list(String domain, Handler<Either<String, JsonArray>> handler) {
        String query = "SELECT *, \"startDate\"::text, \"endDate\"::text " + "FROM " + resourceTable
                + " m  WHERE domain = ? ORDER BY modified DESC";

        JsonArray values = new fr.wseduc.webutils.collections.JsonArray().add(domain);
        sql.prepared(query, values, validResultHandler(handler, "contents", "profiles"));
    }

    @Override
    public void listForUser(UserInfos user, String lang, String domain,
            Handler<Either<String, JsonArray>> handler) {
        String query = "SELECT id, contents, color, \"customColor\" FROM " + resourceTable + " m "
                + "WHERE contents -> '" + lang + "' IS NOT NULL " + "AND trim(contents ->> '" + lang + "') <> '' "
                + "AND profiles ? '" + user.getType() + "' " + "AND \"startDate\" <= now() "
                + "AND \"endDate\" > now() " + "AND domain = '" + domain + "' " + "AND NOT EXISTS (SELECT * FROM "
                + JOIN_TABLE + " WHERE message_id = m.id AND user_id = '" + user.getUserId() + "') "
                + "ORDER BY modified DESC";

        sql.raw(query, validResultHandler(handler, "contents"));
    }

    @Override
    public void markAsRead(UserInfos user, String id, Handler<Either<String, JsonObject>> handler) {
        JsonObject params = new JsonObject().put("message_id", id).put("user_id", user.getUserId());

        sql.insert(JOIN_TABLE, params, validUniqueResultHandler(handler));
    }

}