Java tutorial
/******************************************************************************* * Copyright 2011 Alex 'Ript' Malyshev <alexript@gmail.com> * * 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 net.autosauler.ballance.server.model; import java.util.Date; import net.autosauler.ballance.server.crypt.BCrypt; import net.autosauler.ballance.server.mongodb.Database; import net.autosauler.ballance.shared.UserRole; import org.w3c.dom.Element; import org.w3c.dom.Node; import org.w3c.dom.NodeList; import com.mongodb.BasicDBObject; import com.mongodb.DB; import com.mongodb.DBCollection; import com.mongodb.DBObject; /** * The Class User. */ public class User { final private static String USERSTABLE = "registredusers"; /** * Find. * * @param login * the login * @param domain * the domain * @return the user */ public static User find(String login, String domain) { User user = null; Database.retain(); DBObject myDoc = findObject(login, domain); if (myDoc != null) { user = new User(myDoc); } Database.release(); return user; } /** * Find object. * * @param login * the login * @return the dB object */ private static DBObject findObject(String login, String domain) { DBObject myDoc = null; DB db = Database.get(domain); if (db != null) { DBCollection coll = db.getCollection(USERSTABLE); BasicDBObject query = new BasicDBObject(); query.put("login", login); query.put("domain", domain); myDoc = coll.findOne(query); } return myDoc; } /** * Gen hash. * * @param password * the password * @return the string */ private static String genHash(String password) { return BCrypt.hashpw(password, BCrypt.gensalt()); } /** * Restore. * * @param domain2 * the domain2 * @param nodes * the nodes */ public static void restore(String domain, NodeList nodes) { Database.retain(); DB db = Database.get(domain); if (db != null) { DBCollection coll = db.getCollection(USERSTABLE); BasicDBObject query = new BasicDBObject(); query.put("domain", domain); coll.remove(query); } Database.release(); for (int i = 0; i < nodes.getLength(); i++) { User user = new User(); user.setDomain(domain); Element usernode = (Element) nodes.item(i); NodeList values = usernode.getChildNodes(); for (int c = 0; c < values.getLength(); c++) { Node v = values.item(c); if (v.getNodeType() == Node.ELEMENT_NODE) { Element val = (Element) v; if (val.getNodeName().equals("hash")) { user.setHash(val.getTextContent()); } else if (val.getNodeName().equals("login")) { user.setLogin(val.getTextContent()); } else if (val.getNodeName().equals("fullname")) { user.setUsername(val.getTextContent()); } else if (val.getNodeName().equals("roles")) { String roles = val.getTextContent(); user.setUserrole(new UserRole(Integer.parseInt(roles))); } else if (val.getNodeName().equals("createdate")) { String cd = val.getTextContent(); Long l = Long.parseLong(cd); Date d = new Date(l); user.setCreatedate(d); } else if (val.getNodeName().equals("active")) { String s = val.getTextContent(); Boolean b = Boolean.parseBoolean(s); user.setActive(b); } else if (val.getNodeName().equals("trash")) { String s = val.getTextContent(); Boolean b = Boolean.parseBoolean(s); user.setTrash(b); } } } user.create(true); } } /** * Trash user. * * @param loginanddomain * the loginanddomain * @return true, if successful */ public static boolean trashUser(String login, String domain) { boolean result = false; Database.retain(); DB db = Database.get(domain); if (db != null) { DBCollection coll = db.getCollection(USERSTABLE); BasicDBObject query = new BasicDBObject(); query.put("login", login); query.put("domain", domain); BasicDBObject obj = new BasicDBObject(); obj.put("istrash", true); obj.put("isactive", false); coll.update(query, new BasicDBObject("$set", obj)); result = true; } Database.release(); return result; } /** * Update user. * * @param proxy * the proxy * @return true, if successful */ public static boolean updateUser(net.autosauler.ballance.shared.User proxy) { boolean result = false; Database.retain(); DB db = Database.get(proxy.getDomain()); if (db != null) { DBCollection coll = db.getCollection(USERSTABLE); BasicDBObject query = new BasicDBObject(); query.put("login", proxy.getLogin()); query.put("domain", proxy.getDomain()); BasicDBObject obj = new BasicDBObject(); obj.put("isactive", proxy.isActive()); obj.put("fullname", proxy.getUsername()); obj.put("roles", proxy.getUserrole().getRole()); String password = proxy.getPassword(); if (password != null) { password = password.trim(); } if (!password.isEmpty()) { obj.put("hash", genHash(password)); } coll.update(query, new BasicDBObject("$set", obj)); result = true; } Database.release(); return result; } /** The login. */ private String login; /** The domain. */ private String domain; /** The hash. */ private String hash; /** The user full name. */ private String username; /** The user role. */ private UserRole userrole; /** The uid. */ private Long uid; /** The user's create date. */ private Date createdate; /** Is user active. */ private boolean active; /** Is user in trash. */ private boolean trash; /** * Instantiates a new user. */ public User() { } /** * Instantiates a new user from DB record. * * @param myDoc * the my doc */ public User(DBObject myDoc) { userrole = new UserRole(); setLogin((String) myDoc.get("login")); setDomain((String) myDoc.get("domain")); setHash((String) myDoc.get("hash")); setUsername((String) myDoc.get("fullname")); userrole.setRole((Integer) myDoc.get("roles")); createdate = (Date) myDoc.get("createdate"); active = (Boolean) myDoc.get("isactive"); trash = (Boolean) myDoc.get("istrash"); setUid(0L); } /** * Instantiates a new user. * * @param proxy * the proxy */ public User(net.autosauler.ballance.shared.User proxy) { setLogin(proxy.getLogin()); setDomain(proxy.getDomain()); setUsername(proxy.getUsername()); setUserrole(proxy.getUserrole()); setActive(proxy.isActive()); setPassword(proxy.getPassword()); } /** * Adds the new user to database. If login exists - false; * * @return true, if successful */ public boolean addNewUser() { User user = User.find(getLogin(), getDomain()); if (user != null) { return false; } create(); return true; } /** * Creates the. */ private void create() { create(false); } /** * Creates the new user record. * * @param restoremode * the restoremode */ private void create(boolean restoremode) { Database.retain(); DB db = Database.get(null); if (db != null) { if (!restoremode) { createdate = new Date(); active = true; trash = false; } DBCollection coll = db.getCollection(USERSTABLE); BasicDBObject doc = new BasicDBObject(); doc.put("hash", hash); doc.put("login", login); doc.put("domain", domain); doc.put("fullname", username); doc.put("roles", userrole.getRole()); doc.put("createdate", createdate); doc.put("isactive", active); doc.put("istrash", trash); coll.insert(doc); } Database.release(); } /** * @return */ public Object dump() { StringBuilder sb = new StringBuilder(); sb.append("<user>\n"); sb.append(" <hash>" + getHash() + "</hash>\n"); sb.append(" <login>" + getLogin() + "</login>\n"); sb.append(" <fullname>" + getUsername() + "</fullname>\n"); sb.append(" <roles>" + getUserrole().getRole() + "</roles>\n"); sb.append(" <createdate>" + getCreatedate().getTime() + "</createdate>\n"); sb.append(" <active>" + isActive() + "</active>\n"); sb.append(" <trash>" + isTrash() + "</trash>\n"); sb.append("</user>\n"); return sb.toString(); } /** * Gets the createdate. * * @return the createdate */ public Date getCreatedate() { return createdate; } /** * @return the domain */ public String getDomain() { return domain; } /** * Gets the hash. * * @return the hash */ public String getHash() { return hash; } /** * Gets the login. * * @return the login */ public String getLogin() { return login; } /** * Gets the proxy. * * @return the proxy */ public net.autosauler.ballance.shared.User getProxy() { net.autosauler.ballance.shared.User user = new net.autosauler.ballance.shared.User(); user.setLogin(getLogin()); user.setDomain(getDomain()); user.setUsername(getUsername()); user.setUserrole(new UserRole(getUserrole())); user.setCreatedate(getCreatedate()); user.setActive(isActive()); user.setPassword(""); return user; } /** * Gets the uid. * * @return the uid */ public Long getUid() { return uid; } /** * Gets the username. * * @return the username */ public String getUsername() { return username; } /** * Gets the userrole. * * @return the userrole */ public UserRole getUserrole() { return userrole; } /** * Gets the userrole as int. * * @return the userrole as int */ public int getUserroleAsInt() { return userrole.getRole(); } /** * Checks if is active. * * @return true, if is active */ public boolean isActive() { return active; } /** * Checks if is trash. * * @return true, if is trash */ public boolean isTrash() { return trash; } /** * Checks if is valid user password. * * @param chkpassword * the chkpassword * @return true, if is valid user */ public boolean isValidUser(String chkpassword) { return BCrypt.checkpw(chkpassword, hash); } /** * Sets the active. * * @param active * the new active */ public void setActive(boolean active) { this.active = active; } /** * Sets the createdate. * * @param createdate * the new createdate */ public void setCreatedate(Date createdate) { this.createdate = createdate; } /** * @param domain * the domain to set */ public void setDomain(String domain) { this.domain = domain.trim(); } /** * Sets the hash. * * @param hash * the new hash */ public void setHash(String hash) { this.hash = hash; } /** * Sets the login. * * @param login * the new login */ public void setLogin(String login) { this.login = login.trim(); } /** * Sets the password. * * @param password * the new password */ public void setPassword(String password) { hash = genHash(password); } /** * Sets the trash. * * @param trash * the new trash */ public void setTrash(boolean trash) { this.trash = trash; if (trash) { setActive(false); } } /** * Sets the uid. * * @param uid * the new uid */ public void setUid(Long uid) { this.uid = uid; } /** * Sets the username. * * @param username * the new username */ public void setUsername(String username) { this.username = username; } /** * Sets the userrole. * * @param userrole * the new userrole */ public void setUserrole(UserRole userrole) { this.userrole = userrole; } }