001    /**
002     * Copyright (c) 2000-2012 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.BaseUpgradePortletPreferences;
018    import com.liferay.portal.kernel.util.StringBundler;
019    import com.liferay.portal.kernel.util.StringPool;
020    import com.liferay.portal.kernel.util.Validator;
021    import com.liferay.portal.upgrade.v5_2_3.util.DLFileEntryTable;
022    import com.liferay.portal.upgrade.v5_2_3.util.DLFileRankTable;
023    import com.liferay.portal.upgrade.v5_2_3.util.DLFileShortcutTable;
024    import com.liferay.portal.upgrade.v5_2_3.util.DLFileVersionTable;
025    import com.liferay.portlet.PortletPreferencesFactoryUtil;
026    
027    import java.sql.SQLException;
028    
029    import javax.portlet.PortletPreferences;
030    
031    /**
032     * @author Samuel Kong
033     * @author Brian Wing Shun Chan
034     * @author Douglas Wong
035     */
036    public class UpgradeDocumentLibrary extends BaseUpgradePortletPreferences {
037    
038            @Override
039            protected void doUpgrade() throws Exception {
040                    try {
041                            runSQL("alter_column_type DLFileEntry name VARCHAR(255) null");
042    
043                            runSQL("alter_column_type DLFileRank name VARCHAR(255) null");
044    
045                            runSQL("alter_column_type DLFileShortcut toName VARCHAR(255) null");
046    
047                            runSQL("alter_column_type DLFileVersion name VARCHAR(255) null");
048                    }
049                    catch (SQLException sqle) {
050    
051                            // DLFileEntry
052    
053                            upgradeTable(
054                                    DLFileEntryTable.TABLE_NAME, DLFileEntryTable.TABLE_COLUMNS,
055                                    DLFileEntryTable.TABLE_SQL_CREATE,
056                                    DLFileEntryTable.TABLE_SQL_ADD_INDEXES);
057    
058                            // DLFileRank
059    
060                            upgradeTable(
061                                    DLFileRankTable.TABLE_NAME, DLFileRankTable.TABLE_COLUMNS,
062                                    DLFileRankTable.TABLE_SQL_CREATE,
063                                    DLFileRankTable.TABLE_SQL_ADD_INDEXES);
064    
065                            // DLFileShortcut
066    
067                            upgradeTable(
068                                    DLFileShortcutTable.TABLE_NAME,
069                                    DLFileShortcutTable.TABLE_COLUMNS,
070                                    DLFileShortcutTable.TABLE_SQL_CREATE,
071                                    DLFileShortcutTable.TABLE_SQL_ADD_INDEXES);
072    
073                            // DLFileVersion
074    
075                            upgradeTable(
076                                    DLFileVersionTable.TABLE_NAME, DLFileVersionTable.TABLE_COLUMNS,
077                                    DLFileVersionTable.TABLE_SQL_CREATE,
078                                    DLFileVersionTable.TABLE_SQL_ADD_INDEXES);
079                    }
080    
081                    // groupId
082    
083                    updateGroupId();
084    
085                    // PortletPreferences
086    
087                    updatePortletPreferences();
088            }
089    
090            @Override
091            protected String getUpdatePortletPreferencesWhereClause() {
092                    return "portletId = '20' and preferences like " +
093                            "'%<name>fileEntryColumns</name><value></value>%'";
094            }
095    
096            protected void updateGroupId() throws Exception {
097                    StringBundler sb = new StringBundler(3);
098    
099                    sb.append("update DLFileEntry set groupId = (select groupId from ");
100                    sb.append("DLFolder where DLFolder.folderId = DLFileEntry.folderId)");
101    
102                    runSQL(sb.toString());
103    
104                    sb.setIndex(0);
105    
106                    sb.append("update DLFileRank set groupId = (select groupId from ");
107                    sb.append("DLFolder where DLFolder.folderId = DLFileRank.folderId)");
108    
109                    runSQL(sb.toString());
110    
111                    sb.setIndex(0);
112    
113                    sb.append("update DLFileShortcut set groupId = (select groupId from ");
114                    sb.append("DLFolder where DLFolder.folderId = ");
115                    sb.append("DLFileShortcut.folderId)");
116    
117                    runSQL(sb.toString());
118    
119                    sb.setIndex(0);
120    
121                    sb.append("update DLFileVersion set groupId = (select groupId from ");
122                    sb.append("DLFolder where DLFolder.folderId = DLFileVersion.folderId)");
123    
124                    runSQL(sb.toString());
125            }
126    
127            @Override
128            protected String upgradePreferences(
129                            long companyId, long ownerId, int ownerType, long plid,
130                            String portletId, String xml)
131                    throws Exception {
132    
133                    PortletPreferences portletPreferences =
134                            PortletPreferencesFactoryUtil.fromXML(
135                                    companyId, ownerId, ownerType, plid, portletId, xml);
136    
137                    String fileEntryColumns = portletPreferences.getValue(
138                            "fileEntryColumns", StringPool.BLANK);
139    
140                    if (Validator.isNull(fileEntryColumns)) {
141                            portletPreferences.reset("fileEntryColumns");
142                    }
143    
144                    return PortletPreferencesFactoryUtil.toXML(portletPreferences);
145            }
146    
147    }