Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import in.shop2020.model.v1.user.Address;
import in.shop2020.model.v1.user.PrivateDealUser;
import in.shop2020.model.v1.user.User;
import in.shop2020.model.v1.user.UserCommunicationException;
import in.shop2020.model.v1.user.UserContextService.Client;
import in.shop2020.util.DesEncrypter;
import in.shop2020.thrift.clients.HelperClient;
import in.shop2020.thrift.clients.UserClient;

import java.util.ArrayList;
import java.util.List;
import java.util.Random;

import org.apache.log4j.Logger;
import org.apache.shiro.SecurityUtils;
import org.apache.thrift.TException;
import org.apache.thrift.transport.TTransportException;

/**
 * @author vikas
 *
 */
@SuppressWarnings("serial")
public class UserInfoController extends BaseController {
    private static Logger log = Logger.getLogger(Class.class);
    private String id;
    private long userId;
    private User user;
    private List<Address> userAddresses;
    private Address primaryAdddress;
    private String trustLevelDelta;
    private PrivateDealUser privateDealUser;
    private String isPrivateDealUserActive;
    private static final String chars = "abcdefghijklmonpqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
    private static final Random random = new Random();
    private static final int LENGTH = 10;


    private in.shop2020.util.DesEncrypter desEncrypter = new in.shop2020.util.DesEncrypter("saholic");

    public String index() throws Exception {
        UserClient userServiceClient = new UserClient();
        in.shop2020.model.v1.user.UserContextService.Client userClient = userServiceClient.getClient();
        user = userClient.getUserById(userId);
        userAddresses = user.getAddresses();

        //
        // For cases where Default Address is NULL in db, we get 0 as long here
        // Skipping such cases to avoid exception. Ideally UserContextService should simply
        // return null in such cases.
        //
        if (user.getDefaultAddressId() != 0) {
            primaryAdddress = userClient.getAddressById(user.getDefaultAddressId());
        }

        setPrivateDealUser(userClient.getPrivateDealUser(userId));

        return INDEX;
    }

    public String update() throws Exception {
        userId = Long.parseLong(id);
        userContextServiceClient = new UserClient().getClient();
        userContextServiceClient.increaseTrustLevel(userId, Double.parseDouble(trustLevelDelta));
        return index();
    }

    public String addPrivateDealUser() throws Exception{
        if (canViewBulkOrderEnquiry()){
            userContextServiceClient = new UserClient().getClient();
            userContextServiceClient.addPrivateDealUser(userId);
        }
        return index();
    }

    public String changeStatusOfPrivateDealUser() throws Exception{
        if (canViewBulkOrderEnquiry()){
            userContextServiceClient = new UserClient().getClient();
            userContextServiceClient.changePrivateDealUserStatus(userId, Boolean.valueOf(isPrivateDealUserActive));
        }
        return index();
    }

    public String resetPrivateDealUserPassword() throws Exception{
        log.info("Inside reset password"+userId);
        if (canViewBulkOrderEnquiry()){
            userContextServiceClient = new UserClient().getClient();
            user = userContextServiceClient.getUserById(userId);
            String newPassword = generateNewPassword();
            String encryptedPassword =   desEncrypter.encrypt(newPassword);
            if(userContextServiceClient.forgotPassword(user.getEmail(), encryptedPassword)){
                if(mailNewPassword(user.getEmail(), newPassword)){
                    ;
                }else{
                    throw new UserCommunicationException();
                }
            }else{
                throw new UserCommunicationException();
            }
        }
        return index();
    }

    private boolean mailNewPassword(String emailId, String newPassword) {
        List<String> toList = new ArrayList<String>();
        toList.add(emailId);

        try {
            HelperClient helperClient = new HelperClient("helper_service_server","helper_service_server_port");
            helperClient.getClient().saveUserEmailForSending(toList, "help@saholic.com", "Password reset request", "Your new password is: " + newPassword, "", "ForgotPassword", new ArrayList<String>(), new ArrayList<String>(), 1);
        } catch (Exception e) {
            log.error("Unexpected error while mailing the new password");
            return false;
        }
        return true;
    }

    private static String generateNewPassword() {
        char[] buf = new char[LENGTH];
        for (int i = 0; i < buf.length; i++) {
            buf[i] = chars.charAt(random.nextInt(chars.length()));
        }
        return new String(buf);
    }

    public void setUserId(String userId) {
        try {
            this.userId = Long.parseLong(userId);
        }
        catch (NumberFormatException e) {
            log.error(e);
        }
    }

    public boolean isTrustLevelEditable() {
        return SecurityUtils.getSubject().hasRole("TeamLead") && SecurityUtils.getSubject().hasRole("Outbound");
    }

    public Long getUserId() {
        return userId;
    }

    public User getUser() {
        return user;
    }

    public List<Address> getUserAddresses() {
        return userAddresses;
    }

    public Address getPrimaryAdddress() {
        return primaryAdddress;
    }

    public String getTrustLevelDelta() {
        return trustLevelDelta;
    }

    public void setTrustLevelDelta(String trustLevelDelta) {
        this.trustLevelDelta = trustLevelDelta;
    }

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    public void setPrivateDealUser(PrivateDealUser privateDealUser) {
        this.privateDealUser = privateDealUser;
    }

    public PrivateDealUser getPrivateDealUser() {
        return privateDealUser;
    }

    public void setIsPrivateDealUserActive(String isPrivateDealUserActive) {
        this.isPrivateDealUserActive = isPrivateDealUserActive;
    }

    public String getIsPrivateDealUserActive() {
        return isPrivateDealUserActive;
    }
}