001    /**
002     * Copyright (c) 2000-2010 Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portal.upgrade.v5_2_3;
016    
017    import com.liferay.portal.kernel.upgrade.util.UpgradeTable;
018    import com.liferay.portal.kernel.upgrade.util.UpgradeTableFactoryUtil;
019    import com.liferay.portal.kernel.util.StringBundler;
020    import com.liferay.portal.kernel.util.StringPool;
021    import com.liferay.portal.kernel.util.Validator;
022    import com.liferay.portal.upgrade.BaseUpgradePortletPreferences;
023    import com.liferay.portal.upgrade.v5_2_3.util.DLFileEntryTable;
024    import com.liferay.portal.upgrade.v5_2_3.util.DLFileRankTable;
025    import com.liferay.portal.upgrade.v5_2_3.util.DLFileShortcutTable;
026    import com.liferay.portal.upgrade.v5_2_3.util.DLFileVersionTable;
027    import com.liferay.portlet.PortletPreferencesImpl;
028    import com.liferay.portlet.PortletPreferencesSerializer;
029    
030    /**
031     * @author Samuel Kong
032     * @author Brian Wing Shun Chan
033     * @author Douglas Wong
034     */
035    public class UpgradeDocumentLibrary extends BaseUpgradePortletPreferences {
036    
037            protected void doUpgrade() throws Exception {
038                    try {
039                            runSQL("alter_column_type DLFileEntry name VARCHAR(255) null");
040                    }
041                    catch (Exception e) {
042    
043                            // DLFileEntry
044    
045                            UpgradeTable upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
046                                    DLFileEntryTable.TABLE_NAME, DLFileEntryTable.TABLE_COLUMNS);
047    
048                            upgradeTable.setCreateSQL(DLFileEntryTable.TABLE_SQL_CREATE);
049    
050                            upgradeTable.updateTable();
051                    }
052    
053                    try {
054                            runSQL("alter_column_type DLFileRank name VARCHAR(255) null");
055                    }
056                    catch (Exception e) {
057    
058                            // DLFileRank
059    
060                            UpgradeTable upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
061                                    DLFileRankTable.TABLE_NAME, DLFileRankTable.TABLE_COLUMNS);
062    
063                            upgradeTable.setCreateSQL(DLFileRankTable.TABLE_SQL_CREATE);
064    
065                            upgradeTable.updateTable();
066                    }
067    
068                    try {
069                            runSQL("alter_column_type DLFileShortcut toName VARCHAR(255) null");
070                    }
071                    catch (Exception e) {
072    
073                            // DLFileShortcut
074    
075                            UpgradeTable upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
076                                    DLFileShortcutTable.TABLE_NAME,
077                                    DLFileShortcutTable.TABLE_COLUMNS);
078    
079                            upgradeTable.setCreateSQL(DLFileShortcutTable.TABLE_SQL_CREATE);
080    
081                            upgradeTable.updateTable();
082                    }
083    
084                    try {
085                            runSQL("alter_column_type DLFileVersion name VARCHAR(255) null");
086                    }
087                    catch (Exception e) {
088    
089                            // DLFileVersion
090    
091                            UpgradeTable upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
092                                    DLFileVersionTable.TABLE_NAME,
093                                    DLFileVersionTable.TABLE_COLUMNS);
094    
095                            upgradeTable.setCreateSQL(DLFileVersionTable.TABLE_SQL_CREATE);
096    
097                            upgradeTable.updateTable();
098                    }
099    
100                    // groupId
101    
102                    updateGroupId();
103    
104                    // PortletPreferences
105    
106                    updatePortletPreferences();
107            }
108    
109            protected String getUpdatePortletPreferencesWhereClause() {
110                    return "portletId = '20' and preferences like " +
111                            "'%<name>fileEntryColumns</name><value></value>%'";
112            }
113    
114            protected void updateGroupId() throws Exception {
115                    StringBundler sb = new StringBundler(3);
116    
117                    sb.append("update DLFileEntry set groupId = (select groupId from ");
118                    sb.append("DLFolder where DLFolder.folderId = DLFileEntry.folderId)");
119    
120                    runSQL(sb.toString());
121    
122                    sb.setIndex(0);
123    
124                    sb.append("update DLFileRank set groupId = (select groupId from ");
125                    sb.append("DLFolder where DLFolder.folderId = DLFileRank.folderId)");
126    
127                    runSQL(sb.toString());
128    
129                    sb.setIndex(0);
130    
131                    sb.append("update DLFileShortcut set groupId = (select groupId from ");
132                    sb.append("DLFolder where DLFolder.folderId = ");
133                    sb.append("DLFileShortcut.folderId)");
134    
135                    runSQL(sb.toString());
136    
137                    sb.setIndex(0);
138    
139                    sb.append("update DLFileVersion set groupId = (select groupId from ");
140                    sb.append("DLFolder where DLFolder.folderId = DLFileVersion.folderId)");
141    
142                    runSQL(sb.toString());
143            }
144    
145            protected String upgradePreferences(
146                            long companyId, long ownerId, int ownerType, long plid,
147                            String portletId, String xml)
148                    throws Exception {
149    
150                    PortletPreferencesImpl preferences =
151                            PortletPreferencesSerializer.fromXML(
152                                    companyId, ownerId, ownerType, plid, portletId, xml);
153    
154                    String fileEntryColumns = preferences.getValue(
155                            "fileEntryColumns", StringPool.BLANK);
156    
157                    if (Validator.isNull(fileEntryColumns)) {
158                            preferences.reset("fileEntryColumns");
159                    }
160    
161                    return PortletPreferencesSerializer.toXML(preferences);
162            }
163    
164    }