Application: gvSIG desktop

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 * 

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 * 

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 * 

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, 

 * MA  02110-1301, USA.

 * 

 */

/*

 * AUTHORS (In addition to CIT):

 * 2010 {Prodevelop}   {Task}

 */

package org.gvsig.installer.swing.impl.execution.model;

import java.io.File;

import java.util.Comparator;

import java.util.List;

import java.util.Set;

import java.util.TreeSet;

import javax.swing.table.AbstractTableModel;

import org.gvsig.installer.lib.api.InstallerLocator;

import org.gvsig.installer.lib.api.InstallerManager;

import org.gvsig.installer.lib.api.PackageInfo;

import org.gvsig.installer.lib.api.creation.MakePluginPackageServiceException;

import org.gvsig.installer.lib.api.execution.InstallPackageService;

import org.gvsig.installer.swing.api.SwingInstallerManager;

/**

 * @author <a href="mailto:jpiera@gvsig.org">Jorge Piera Llodr&aacute;</a>

 */

public class PackagesTableModel extends AbstractTableModel {

    private static final long serialVersionUID = 5953623592122163923L;

    private TablePackageInfo[] installerInfos = null;

    private final String[] columnNames;

    private InstallerManager installerManager = InstallerLocator

        .getInstallerManager();

    public PackagesTableModel(SwingInstallerManager swingInstallerManager,

        InstallPackageService installerExecutionService, boolean isOfficial,

        File pluginsFolder) {

        Set<TablePackageInfo> infoSet =

            new TreeSet<TablePackageInfo>(new TablePackageInfoComparator());

        for (int i = 0; i < installerExecutionService.getPackageCount(); i++) {

            PackageInfo installerInfo =

                installerExecutionService.getPackageInfo(i);

            if (installerInfo.isOfficial() == isOfficial) {

                TablePackageInfo info =

                    new TablePackageInfo(installerInfo, false, true);

                infoSet.add(info);

            }

        }

        try {

            PackageInfo[] installedPackages =

                installerManager.getInstalledPackages(pluginsFolder);

            for (int i = 0; i < installedPackages.length; i++) {

                if (installedPackages[i].isOfficial() == isOfficial) {

                    TablePackageInfo info =

                        new TablePackageInfo(installedPackages[i], true, false);

                    infoSet.add(info);

                }

            }

        } catch (MakePluginPackageServiceException e) {

            throw new RuntimeException(e);

        }

        installerInfos = infoSet.toArray(new TablePackageInfo[infoSet.size()]);

        columnNames =

            new String[] {

                "", // Check column

                swingInstallerManager.getText("id"),

                swingInstallerManager.getText("type"),

                swingInstallerManager.getText("status"),

                swingInstallerManager.getText("version"),

                swingInstallerManager.getText("build"),

                swingInstallerManager.getText("operating_system"),

                swingInstallerManager.getText("architecture"),

                swingInstallerManager.getText("java_version"),

                swingInstallerManager.getText("gvSIG_version") };

    }

    public Class<?> getColumnClass(int columnIndex) {

        switch (columnIndex) {

        case 0:

            return Boolean.class;

        default:

            return String.class;

        }

    }

    public int getColumnCount() {

        return columnNames.length;

    }

    public String getColumnName(int columnIndex) {

        return columnIndex >= columnNames.length ? ""

            : columnNames[columnIndex];

    }

    public int getRowCount() {

        return installerInfos.length;

    }

    public Object getValueAt(int rowIndex, int columnIndex) {

        TablePackageInfo tableInfo = installerInfos[rowIndex];

        PackageInfo installerInfo = tableInfo.getPackageInfo();

        switch (columnIndex) {

        case 0:

            return tableInfo.isSelected();

        case 1:

            return installerInfo.getCode();

        case 2:

            return installerInfo.getType();

        case 3:

            return installerInfo.getState();

        case 4:

            return installerInfo.getVersion();

        case 5:

            return installerInfo.getBuild();

        case 6:

            return installerInfo.getOperatingSystem();

        case 7:

            return installerInfo.getArchitecture();

        case 8:

            return installerInfo.getJavaVM();

        case 9:

            return installerInfo.getGvSIGVersion();

        default:

            return "";

        }

    }

    private PackageInfo getPackageInfoAt(int rowIndex) {

        return installerInfos[rowIndex].getPackageInfo();

    }

    public String getDescriptionAt(int rowIndex) {

        return getPackageInfoAt(rowIndex).getDescription();

    }

    public boolean isCellEditable(int rowIndex, int columnIndex) {

        return (columnIndex == 0) && installerInfos[rowIndex].isInstallable();

    }

    public void setValueAt(Object aValue, int rowIndex, int columnIndex) {

        if (columnIndex == 0) {

            TablePackageInfo tableInfo = installerInfos[rowIndex];

            if (tableInfo.isInstallable()) {

                tableInfo.setSelected(Boolean.TRUE.equals(aValue));

            }

        }

    }

    public void addPackageInfosToInstall(List<PackageInfo> packageInfosToInstall) {

        for (int i = 0; i < installerInfos.length; i++) {

            TablePackageInfo tableInfo = installerInfos[i];

            if (tableInfo.isSelected()) {

                packageInfosToInstall.add(tableInfo.getPackageInfo());

            }

        }

    }

    public boolean hasAnyPackageSelected() {

        for (int i = 0; i < installerInfos.length; i++) {

            if (installerInfos[i].isSelected) {

                return true;

            }

        }

        return false;

    }

    public static final class TablePackageInfo {

        private final PackageInfo packageInfo;

        private final boolean isInstalled;

        private final boolean isInstallable;

        private boolean isSelected;

        public TablePackageInfo(PackageInfo packageInfo, boolean isInstalled,

            boolean isInstallable) {

            this(packageInfo, isInstalled, isInstallable, false);

        }

        public TablePackageInfo(PackageInfo packageInfo, boolean isInstalled,

            boolean isInstallable, boolean isSelected) {

            this.packageInfo = packageInfo;

            this.isInstalled = isInstalled;

            this.isInstallable = isInstallable;

            this.isSelected = isSelected;

        }

        public PackageInfo getPackageInfo() {

            return packageInfo;

        }

        public boolean isInstalled() {

            return isInstalled;

        }

        public boolean isInstallable() {

            return isInstallable;

        }

        public boolean isSelected() {

            return isSelected;

        }

        public void setSelected(boolean isSelected) {

            this.isSelected = isSelected;

        }

    }

    private static final class TablePackageInfoComparator implements

        Comparator<TablePackageInfo> {

        public int compare(TablePackageInfo info1, TablePackageInfo info2) {

            if (info1 == null) {

                if (info2 == null) {

                    return 0;

                }

                return -1;

            }

            if (info2 == null) {

                return 1;

            }

            PackageInfo packageInfo1 = info1.getPackageInfo();

            PackageInfo packageInfo2 = info2.getPackageInfo();

            int comparison = compareCode(packageInfo1, packageInfo2);

            if (comparison == 0) {

                comparison = compareVersion(packageInfo1, packageInfo2);

                if (comparison == 0) {

                    comparison = compareBuild(packageInfo1, packageInfo2);

                }

            }

            return comparison;

        }

        private int compareBuild(PackageInfo packageInfo1,

            PackageInfo packageInfo2) {

            return packageInfo1.getBuild() - packageInfo2.getBuild();

        }

        private int compareCode(PackageInfo packageInfo1,

            PackageInfo packageInfo2) {

            return packageInfo1.getCode().compareTo(packageInfo2.getCode());

        }

        private int compareVersion(PackageInfo packageInfo1,

            PackageInfo packageInfo2) {

            return packageInfo1.getVersion().compareTo(

                packageInfo2.getVersion());

        }

    }

    /**

     * Returns the TablePackageInfo located in the given row.

     * 

     * @param row

     *            the TablePackageInfo located in the given row

     */

    public TablePackageInfo getPackageInfo(int row) {

        return installerInfos[row];

    }

}