/*
 *  Copyright 2015 Anyware Services
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.ametys.plugins.sms.dao;

import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.session.ResultContext;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.apache.ibatis.session.SqlSession;

import org.ametys.core.datasource.AbstractMyBatisDAO;
import org.ametys.core.ui.Callable;
import org.ametys.plugins.sms.SMSHelper;

/**
 * DAO for manipulating SMS subscribers.
 *
 */
public class SubscriberDAO extends AbstractMyBatisDAO
{
    /** The Avalon role */
    public static final String ROLE = SubscriberDAO.class.getName();
    
    /**
     * Get all the phone numbers in the database form the list
     * @param list the id of the phone number list
     * @return A map, where keys are phone number and value are date object.
     */
    public Map<String, Date> getPhoneNumbersFromList(String list)
    {
        Map<String, Object> params = new HashMap<>();
        params.put("list", list);
        
        Map<String, Date> phoneNumbers = new HashMap<>();
        try (SqlSession session = getSession())
        {
            List<Map<String, Object>> results = session.selectList("SmsSubscribers.getPhoneNumbers", params);
            for (Map<String, Object> result : results)
            {
                phoneNumbers.put((String) result.get("PhoneNumber"), (Date) result.get("Date"));
            }
        }
        
        return phoneNumbers;
    }
    
    /**
     * Insert the phoneNumber in the database
     * @param phoneNumber the phone number 
     * @param list the id of the phone number list
     */
    public void insertNumber(String phoneNumber, String list)
    {
        try (SqlSession sqlSession = getSession())
        {
            String stmtId = "SmsSubscribers.insertNumber";
            
            Map<String, Object> params = new HashMap<>();
            params.put("phoneNumber", phoneNumber);
            params.put("list", list);
            params.put("date", new Date());
            
            // Insert the session object and get the generated ID.
            sqlSession.insert(stmtId, params);
            
            // Commit the transaction.
            sqlSession.commit();
        }
    }
    
    /**
     * Check if the phoneNumber exists in the database
     * @param phoneNumber the phone number
     * @param list the id of the list
     * @return true if exist
     */
    public boolean numberAlreadyExists(String phoneNumber, String list)
    {
        String stmtId = "SmsSubscribers.getPhoneNumbers";
        
        try (SqlSession session = getSession())
        {
            Map<String, Object> params = new HashMap<>();
            params.put("phoneNumber", phoneNumber);
            params.put("list", list);
            
            List<Map> selectList = session.selectList(stmtId, params, new RowBounds(RowBounds.NO_ROW_OFFSET, 1));
            return selectList.size() > 0;
        }
    }
    
    /**
     * Creates a SMS subscriber.
     * @param listId The id of the SMS list to subscribe to.
     * @param phoneNumber The phone number to add
     * @return The list id and the phone number of the added subscriber, or an error
     */
    @Callable
    public Map<String, String> createSubscriber(String listId, String phoneNumber)
    {
        Map<String, String> result = new HashMap<>();
        
        try
        {
            String realPhoneNumber = phoneNumber.trim();
            realPhoneNumber = SMSHelper.transformPhoneNumber(realPhoneNumber);
            if (!SMSHelper.checkPhoneNumber(realPhoneNumber) || !SMSHelper.PHONE_NUMBER_INTERNATIONAL_VALIDATOR.matcher(realPhoneNumber).matches())
            {   
                result.put("error", "wrong-format");
            }
            else if (numberAlreadyExists(realPhoneNumber, listId))
            {
                result.put("error", "already-exists");
            }
            else
            {
                insertNumber(realPhoneNumber, listId);
                result.put("phoneNumber", realPhoneNumber);
                result.put("smsListId", listId);
            }
        }
        catch (Exception e)
        {
            getLogger().error("Unable to add a new subscriber", e);
            result.put("error", "exception");
        }
        
        return result;
    }
    
    /**
     * Delete the phoneNumber in the database
     * @param phoneNumber the phone number 
     * @param list the id of the phone number list
     */
    public void deleteNumber(String phoneNumber, String list)
    {
        try (SqlSession sqlSession = getSession())
        {
            String stmtId = "SmsSubscribers.deleteNumber";
            
            Map<String, Object> params = new HashMap<>();
            if (StringUtils.isNotEmpty(phoneNumber))
            {
                params.put("phoneNumber", phoneNumber);
            }
            if (StringUtils.isNotEmpty(list))
            {
                params.put("list", list);
            }
            
            // Insert the session object and get the generated ID.
            sqlSession.delete(stmtId, params);
            
            // Commit the transaction.
            sqlSession.commit();
        }
    }
    
    /**
     * Delete every phone number of a list
     * @param list the id of the phone number list
     */
    public void deleteAllNumbers(String list)
    {
        deleteNumber(null, list);
    }
    
    /**
     * Deletes the given subscribers.
     * @param listId The The id of the SMS list.
     * @param phoneNumbers The phone numbers to delete
     * @return The list id, the numbers deleted, the numbers not deleted and/or an error.
     */
    @Callable
    public Map<String, Object> deleteSubscribers(String listId, ArrayList<String> phoneNumbers)
    {
        Map<String, Object> result = new HashMap<>();
        List<String> numbersDeleted = new LinkedList<>();
        List<String> numbersNotDeleted = new LinkedList<>();
        
        for (String phoneNumber : phoneNumbers)
        {
            try
            {
                deleteNumber(phoneNumber, listId);
                numbersDeleted.add(phoneNumber);
            }
            catch (Exception e)
            {
                getLogger().error("Unable to delete the subscriber " + phoneNumber, e);
                numbersNotDeleted.add(phoneNumber);
                result.put("error", "exception");
            }
        }
        
        result.put("listId", listId);
        result.put("numbersDeleted", numbersDeleted);
        result.put("numbersNotDeleted", numbersNotDeleted);
        
        return result;
    }
    
    /**
     * Get the number of user in a list
     * @return the number of user for each sms list
     */
    public Map<String, Number> getNbUserFromList()
    {
        Map<String, Number> userCounts = new HashMap<>();
        try (SqlSession session = getSession())
        {
            session.select("SmsSubscribers.getUserCounts", new ResultHandler<Map<String, Object>>()
            {
                @Override
                public void handleResult(ResultContext<? extends Map<String, Object>> context)
                {
                    final Map<String, Object> map = context.getResultObject();
                    userCounts.put((String) map.get("ListId"), (Number) map.get("nb"));
                }
            });
        }
        
        return userCounts;
    }
}