001/* 002 * (C) Copyright 2006-2012 Nuxeo SA (http://nuxeo.com/) and others. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 * 016 * Contributors: 017 * <a href="mailto:tdelprat@nuxeo.com">Tiry</a> 018 * <a href="mailto:tmartins@nuxeo.com">Thierry Martins</a> 019 */ 020 021package org.nuxeo.ecm.quota.size; 022 023import org.nuxeo.ecm.core.api.DocumentModel; 024import org.nuxeo.ecm.core.api.adapter.DocumentAdapterFactory; 025import org.nuxeo.ecm.quota.QuotaUtils; 026 027/** 028 * Simple factory for {@link QuotaAwareDocument} document model adapter 029 * 030 * @author <a href="mailto:tdelprat@nuxeo.com">Tiry</a> 031 * @since 5.6 032 */ 033public class QuotaAwareDocumentFactory implements DocumentAdapterFactory { 034 035 public static QuotaAwareDocument make(DocumentModel doc, boolean save) { 036 if (!doc.hasFacet(QuotaAwareDocument.DOCUMENTS_SIZE_STATISTICS_FACET)) { 037 doc.addFacet(QuotaAwareDocument.DOCUMENTS_SIZE_STATISTICS_FACET); 038 if (save) { 039 DocumentModel origDoc = doc; 040 // set flags to disable listeners 041 QuotaUtils.disableListeners(doc); 042 doc = doc.getCoreSession().saveDocument(doc); 043 // remove flags as they could be kept in the document for a long time otherwise 044 QuotaUtils.clearContextData(doc); 045 // also remove flags from original doc, which the caller still references 046 QuotaUtils.clearContextData(origDoc); 047 } 048 } 049 return (QuotaAwareDocument) doc.getAdapter(QuotaAware.class); 050 } 051 052 @Override 053 public Object getAdapter(DocumentModel doc, Class<?> adapter) { 054 if (doc.hasFacet(QuotaAwareDocument.DOCUMENTS_SIZE_STATISTICS_FACET)) { 055 return adapter.cast(new QuotaAwareDocument(doc)); 056 } 057 return null; 058 } 059}