package org.zhios.website.admin.tool;

import java.io.File;
import java.io.FileOutputStream;

import org.zhios.website.admin.data.User;
import org.zhios.website.admin.model.MD5Operator;

	
public class AddWebSiteAdmin {
	
	public static void main(String[] args) {
		if(args.length!=2){
			System.out.println("Args is illegal! Command should like 'java AddWebSiteAdmin [username] [password]'.");
		}
		else{
			String username=args[0];
			String password=args[1];
			
			if(AddAdmin(username, password)){
				System.out.println("Add website user success.");
			}else{
				System.out.println("Add website user fail.");
			}
		}
	}
	
	private static boolean AddAdmin(String pusername,String ppassword){
		boolean result=false;
		
		String password=MD5Operator.getMD5Digest(ppassword);
		if(password==null||password.isEmpty()) return result;
		
		String filepath="./sysusers.tmdb";
		try {
			File userfile=new File(filepath);
			if(userfile.exists()){
				FileOutputStream ostream=new FileOutputStream(userfile,true);
				
				User userbuffer=new User();
				userbuffer.setUserByString(pusername, password, "admin");
				ostream.write(userbuffer.userlen);
				ostream.write(userbuffer.usernamelen);
				ostream.write(userbuffer.username);
				ostream.write(userbuffer.passwordlen);
				ostream.write(userbuffer.password);
				ostream.write(userbuffer.rolelen);
				ostream.write(userbuffer.role);

				ostream.flush();
				ostream.close();
				
				result=true;
			}
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		return result;
	}
}