001/* 002 * (C) Copyright 2009 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 * Thomas Roger 018 */ 019 020package org.nuxeo.ecm.platform.importer.factories; 021 022import java.io.IOException; 023 024import org.nuxeo.ecm.core.api.CoreSession; 025import org.nuxeo.ecm.core.api.DocumentModel; 026import org.nuxeo.ecm.core.api.blobholder.BlobHolder; 027import org.nuxeo.ecm.platform.filemanager.api.FileImporterContext; 028import org.nuxeo.ecm.platform.filemanager.api.FileManager; 029import org.nuxeo.ecm.platform.importer.source.SourceNode; 030import org.nuxeo.runtime.api.Framework; 031 032/** 033 * DocumentModel factory based on the {@code FileManager}. Use the {@code FileManager} to create Folderish and Leaf 034 * Nodes. 035 * 036 * @author <a href="mailto:troger@nuxeo.com">Thomas Roger</a> 037 */ 038public class FileManagerDocumentModelFactory extends AbstractDocumentModelFactory { 039 040 protected FileManager fileManager; 041 042 @Override 043 public DocumentModel createFolderishNode(CoreSession session, DocumentModel parent, SourceNode node) 044 throws IOException { 045 FileManager fileManager = getFileManager(); 046 return fileManager.createFolder(session, node.getName(), parent.getPathAsString()); 047 } 048 049 @Override 050 public DocumentModel createLeafNode(CoreSession session, DocumentModel parent, SourceNode node) throws IOException { 051 FileManager fileManager = getFileManager(); 052 BlobHolder bh = node.getBlobHolder(); 053 FileImporterContext context = FileImporterContext.builder(session, bh.getBlob(), parent.getPathAsString()) 054 .overwrite(true) 055 .fileName(node.getName()) 056 .build(); 057 DocumentModel doc = fileManager.createOrUpdateDocument(context); 058 doc = setDocumentProperties(session, bh.getProperties(), doc); 059 return doc; 060 } 061 062 protected FileManager getFileManager() { 063 if (fileManager == null) { 064 fileManager = Framework.getService(FileManager.class); 065 } 066 return fileManager; 067 } 068 069}