me.philnate.textmanager.entities.Document.java Source code

Introduction

Here is the source code for me.philnate.textmanager.entities.Document.java

Source

/**
 *   textManager, a GUI for managing bills for texter jobs
 *
 *   Copyright (C) 2013 philnate
 *
 *   This file is part of textManager.
 *
 *   textManager 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 3 of the
 *   License, or (at your option) any later version.
 *
 *   textManager 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 textManager. If not, see <http://www.gnu.org/licenses/>.
 */
package me.philnate.textmanager.entities;

import static java.lang.String.format;
import static me.philnate.textmanager.utils.DB.docs;
import static me.philnate.textmanager.utils.DB.ds;

import java.io.File;
import java.io.IOException;

import me.philnate.textmanager.utils.WordCount;

import org.bson.types.ObjectId;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.code.morphia.annotations.Id;
import com.google.code.morphia.query.Query;
import com.mongodb.gridfs.GridFSInputFile;

public class Document {
    @Id
    private ObjectId id;
    private ObjectId document;
    private long wordCount;
    private String title;
    private ObjectId billingItemId;

    private static Logger LOG = LoggerFactory.getLogger(Document.class);

    public ObjectId getId() {
        return id;
    }

    public Document setId(ObjectId id) {
        this.id = id;
        return this;
    }

    public ObjectId getDocument() {
        return document;
    }

    public Document setDocument(ObjectId document) {
        this.document = document;
        return this;
    }

    public long getWordCount() {
        return wordCount;
    }

    public Document setWordCount(long wordCount) {
        this.wordCount = wordCount;
        return this;
    }

    public ObjectId getBillingItemId() {
        return billingItemId;
    }

    public String getTitle() {
        return title;
    }

    public Document setTitle(String title) {
        this.title = title;
        return this;
    }

    @Override
    public String toString() {
        return this.title + " (" + this.wordCount + ")";
    }

    public static Document loadAndSave(File file) {
        GridFSInputFile gFile;
        try {
            gFile = docs.createFile(file);
            gFile.save();
            LOG.debug(format("Reading file  '%s' for new Document", file.getAbsolutePath()));
            Document doc = new Document();
            doc.setDocument((ObjectId) gFile.getId()).setTitle(file.getName());
            return doc.setWordCount(WordCount.countFile(file));
        } catch (IOException e1) {
            return null;
        }
    }

    public void delete() {
        LOG.debug(format("deleting document '%s'", document));
        docs.remove(document);
    }

    public static Query<Document> find() {
        return ds.find(Document.class);
    }

    public static Document find(ObjectId id) {
        LOG.debug(format("Searching for Document with ObjectId '%s'", id));
        return ds.get(Document.class, id);
    }
}