winappassociationwriter.java

JDesktop Integration Components （JDIC）

        boolean isActionListEmpty = true;
        boolean isFileExtensionEmpty = true;
        boolean isValid = false;
            
        // Check if actionlist is empty
        if (assoc.getActionList() != null) {
            isActionListEmpty = assoc.getActionList().isEmpty();
        }
        // Check if file extension list is empty
        if (assoc.getFileExtList() != null) {
            isFileExtensionEmpty = assoc.getFileExtList().isEmpty();
        }
    
        if (isFileExtensionEmpty && (assoc.getMimeType() == null)) { 
            isValid = false;
        } else if ((assoc.getDescription() != null) || (assoc.getIconFileName() != null) ||  
            (!isActionListEmpty)) {          
            isValid = !isFileExtensionEmpty; 
        } else {
            isValid = true;
        }
    
        if (!isValid) {
            throw new IllegalArgumentException("The given association is invalid. It should " +
                "specify both the mimeType and fileExtensionList fields to perform this operation.");
        }
    }  

    /**
     * Checks whether the given assocation is valid for unregistration.
     * If both the mimeType and fileExtensionList field is null, throw exception. 
     * 
     * @param assoc a given Association object.
     * @throws IllegalArgumentException if the given association is not valid for unregistration.
     */
    public void checkAssociationValidForUnregistration(Association assoc) 
        throws IllegalArgumentException {
        boolean isFileExtListEmpty = true;
        if (assoc.getFileExtList() != null) {
            isFileExtListEmpty = assoc.getFileExtList().isEmpty();
        }
        if ((assoc.getMimeType() == null) && isFileExtListEmpty) {
            throw new IllegalArgumentException("The given association is invalid. It should " +
                "specify both the mimeType and fileExtensionList fields to perform this " +
                "operation.");
		}
    }
  
    /**
     * Checks whether the given assocation already existed in Windows registry.
     * <PRE>
     * The evaluation will based on the following rule:
     * 1. mimetype == null && fileExt == null
     *    return false
     * 2. mimetype == null && fileExt != null
     *    return !(fileExt existed in the reg table)
     * 3. mimetype != null && fileExt == null
     *    return !(mimetype existed in the reg table)
     * 4. mimetype != null && fileExt != null
     *    return ( (mimetype existed in the reg table) &&
     *             (fileExt existed in the reg table) &&
     *             (getFileExtByMimeType(mimetype) == fileExt) &&
     *             (getMimeTypeByFileExt(fileExt) == mimetype) )
     * </PRE>
     *
     * @param assoc given association (not null)
     * @param regLevel given registry level
     * @return true if the given association existed
     *
     */
    public boolean isAssociationExist(Association assoc, int regLevel) {
        String temFileExt = null;
        String temMimeType = assoc.getMimeType();
        Iterator temFileExtIter;
        if (assoc.getFileExtList() != null) {
            temFileExtIter = assoc.getFileExtList().iterator();
        } else {
            temFileExtIter = null;
        }

        if (temFileExtIter != null) {
            if (temFileExtIter.hasNext()) {
                temFileExt = (String) temFileExtIter.next();
            }
        }

        //Check if there is the same file extension define in Win2k
        //HKEY_CURRENT_USER\\software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts
        if (WinRegistryUtil.isWin2kUserDefinedFileExtExist(temFileExt)) {
            return true;
        }

        if ((temMimeType == null) && (temFileExt == null)) {
            return false; 
        } else if ((temMimeType == null) && (temFileExt != null)) {
            return WinRegistryUtil.isFileExtExist(temFileExt, regLevel);
        } else if ((temMimeType != null) && (temFileExt == null)) {
            return WinRegistryUtil.isMimeTypeExist(temMimeType, regLevel);
        } else {
            String regMimeType = WinRegistryUtil.getMimeTypeByFileExt(temFileExt, regLevel);
            String regFileExt = WinRegistryUtil.getFileExtByMimeType(temMimeType, regLevel);
    
            return ((WinRegistryUtil.isMimeTypeExist(temMimeType, regLevel))
                    && (WinRegistryUtil.isFileExtExist(temFileExt, regLevel))
                    && (temFileExt == null
                            ? regFileExt == null
                            : temFileExt.equals(regFileExt))
                    && (temMimeType == null
                            ? regMimeType == null
                            : temMimeType.equals(regMimeType)));
        }
    }
  
    /**
     * Registers the given association info in the specified level.
     * <PRE>
     * 
     * The register process will following the rules below
     * 1. mimetype == null && fileExt != null
     *    Since fileExt does not exist in this case
     *    1.1 adds the fileExt
     *    1.2 adds description, icon file and action list for this fileExt
     * 2. mimetype != null && fileExt == null
     *    Since mimetype does not exist in this case
     *    just adds the mimetype in the reg table
     * 3. mimetype != null && fileExt != null
     *    3.1 Adds the mime type into the reg table if necessary
     *    3.2 Adds the file extension into the reg table if necessary
     *    3.3 Adds the description, icon file and action list for the fileExt
     *    3.4 Sets the mutual reference of the mime type and file extension
     * </PRE>
     *
     * @param assoc given association (not null)
     * @param regLevel given registry level
     * @throws RegisterFailedException if the operation fails.
     */
    public void registerAssociation(Association assoc, int regLevel) 
        throws RegisterFailedException {
    	//LOCAL_MACHINE. Association, in this case, should only be written into LOCAL_MACHINE rather
    	//than the CURRENT_USER.
    	boolean isOldWindows = false;
    	String osName = System.getProperty("os.name").toLowerCase();
    	if ((osName.indexOf("98") != -1) || 
    	    (osName.indexOf("me") != -1) || 
    	    (osName.indexOf("nt") != -1)) {
    		isOldWindows = true;
    	}
    	if (isOldWindows) {
    		regLevel = AppConstants.SYSTEM_LEVEL;
    	}
        BackupAssociation backupAssoc = new BackupAssociation(assoc, regLevel);
        String curMimeType = backupAssoc.getCurMimeType();
        String curFileExt = backupAssoc.getCurFileExt();
        String curDescription = assoc.getDescription();
        String curIconFileName = assoc.getIconFileName();
        List   curActionList = assoc.getActionList();
        boolean curMimeTypeExisted = backupAssoc.getCurMimeTypeExisted();
        boolean curFileExtExisted = backupAssoc.getCurFileExtExisted();
        
    
        try {
            if ((curMimeType == null) && (curFileExt != null)) {
                WinRegistryUtil.addFileExt(curFileExt, regLevel);
                if (curDescription != null) {
                    WinRegistryUtil.setDescriptionByFileExt(curDescription,
                            curFileExt, regLevel);
                }
                if (curIconFileName != null) {
                    WinRegistryUtil.setIconFileNameByFileExt(curIconFileName,
                        curFileExt, regLevel);
                }
                if (curActionList != null) {
                    WinRegistryUtil.setActionListByFileExt(curActionList,
                        curFileExt, regLevel);
                }
                //Mark the classID generator field
                WinRegistryUtil.markGeneratorByFileExt(curFileExt, regLevel);
            } else if ((curMimeType != null) && (curFileExt == null)) {
                WinRegistryUtil.addMimeType(curMimeType, regLevel);
            } else if ((curMimeType != null) && (curFileExt != null)) {
                if (!curMimeTypeExisted) {
                    WinRegistryUtil.addMimeType(curMimeType, regLevel);
                }
                if (!curFileExtExisted) {
                    WinRegistryUtil.addFileExt(curFileExt, regLevel);
                }
                if (curDescription != null) {
                    WinRegistryUtil.setDescriptionByFileExt(curDescription,
                        curFileExt, regLevel);
                }
                if (curIconFileName != null) {
                    WinRegistryUtil.setIconFileNameByFileExt(curIconFileName,
                        curFileExt, regLevel);
                }
                if (curActionList != null) {
                    WinRegistryUtil.setActionListByFileExt(curActionList,
                        curFileExt, regLevel);
                }
                //Mark the classID generator field.
                WinRegistryUtil.markGeneratorByFileExt(curFileExt, regLevel);
                WinRegistryUtil.setMutualRef(curFileExt, curMimeType, regLevel);
            }
        } catch (RegisterFailedException e) {
            restoreAssociationRegistration(backupAssoc, regLevel);
                
            throw e;
        } 
    }

    /**
     * Unregisters the given association in the specified level.
     *
     * @param assoc given association (not null)
     * @param regLevel given registry level
     * @throws RegisterFailedException if the operation fails.
     */
    public void unregisterAssociation(Association assoc, int regLevel) 
        throws RegisterFailedException {
        //Note: Windows 98, Windows ME & Windows NT will only take care of registry information from
		//LOCAL_MACHINE. Association, in this case, should only be unregistered from LOCAL_MACHINE rather
		//than the CURRENT_USER.
		boolean isOldWindows = false;
		String osName = System.getProperty("os.name").toLowerCase();
		if ((osName.indexOf("98") != -1) || 
			(osName.indexOf("me") != -1) || 
			(osName.indexOf("nt") != -1)) {
			isOldWindows = true;
		}
		if (isOldWindows) {
			regLevel = AppConstants.SYSTEM_LEVEL;
		}
		
        BackupAssociation backupAssoc = new BackupAssociation(assoc, regLevel);
        String curMimeType = backupAssoc.getCurMimeType();
        String curFileExt = backupAssoc.getCurFileExt();
        boolean curMimeTypeExisted = backupAssoc.getCurMimeTypeExisted();
        boolean curFileExtExisted = backupAssoc.getCurFileExtExisted();
    
        try {
            if (curMimeTypeExisted) {
                WinRegistryUtil.removeMimeType(curMimeType, regLevel);
            }
            if (curFileExtExisted) {
                WinRegistryUtil.removeFileExt(curFileExt, regLevel);
            }
        } catch (RegisterFailedException e) {
            restoreAssociationUnregistration(backupAssoc, regLevel);
            
            throw e;
        }
    }
}