org.tinymediamanager.ui.movies.actions.MovieReadNfoAction Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tinyMediaManager Show documentation
The newest version!
/*
 * Copyright 2012 - 2019 Manuel Laggner
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.tinymediamanager.ui.movies.actions;

import java.awt.event.ActionEvent;
import java.util.ArrayList;
import java.util.List;
import java.util.ResourceBundle;

import javax.swing.JOptionPane;

import org.tinymediamanager.core.MediaFileType;
import org.tinymediamanager.core.entities.MediaFile;
import org.tinymediamanager.core.movie.connector.MovieNfoParser;
import org.tinymediamanager.core.movie.entities.Movie;
import org.tinymediamanager.core.threading.TmmTask;
import org.tinymediamanager.core.threading.TmmTaskHandle.TaskType;
import org.tinymediamanager.core.threading.TmmTaskManager;
import org.tinymediamanager.ui.MainWindow;
import org.tinymediamanager.ui.UTF8Control;
import org.tinymediamanager.ui.actions.TmmAction;
import org.tinymediamanager.ui.movies.MovieUIModule;

/**
 * The MovieReadNfoAction - to read the NFOs from all selected movies
 * 
 * @author Manuel Laggner
 */
public class MovieReadNfoAction extends TmmAction {
  private static final long           serialVersionUID = 2866581962767395824L;
  private static final ResourceBundle BUNDLE           = ResourceBundle.getBundle("messages", new UTF8Control()); //$NON-NLS-1$

  public MovieReadNfoAction() {
    putValue(NAME, BUNDLE.getString("movie.readnfo")); //$NON-NLS-1$
    putValue(SHORT_DESCRIPTION, BUNDLE.getString("movie.readnfo.desc")); //$NON-NLS-1$
  }

  @Override
  protected void processAction(ActionEvent e) {
    final List selectedMovies = new ArrayList<>(MovieUIModule.getInstance().getSelectionModel().getSelectedMovies());

    if (selectedMovies.isEmpty()) {
      JOptionPane.showMessageDialog(MainWindow.getActiveInstance(), BUNDLE.getString("tmm.nothingselected")); //$NON-NLS-1$
      return;
    }

    // rewrite selected NFOs
    TmmTaskManager.getInstance().addUnnamedTask(new TmmTask(BUNDLE.getString("movie.readnfo"), selectedMovies.size(), TaskType.BACKGROUND_TASK) {
      @Override
      protected void doInBackground() {
        int i = 0;
        for (Movie movie : selectedMovies) {
          Movie tempMovie = null;

          // process all registered NFOs
          for (MediaFile mf : movie.getMediaFiles(MediaFileType.NFO)) {
            // at the first NFO we get a movie object
            if (tempMovie == null) {
              try {
                tempMovie = MovieNfoParser.parseNfo(mf.getFileAsPath()).toMovie();
              }
              catch (Exception ignored) {
              }
              continue;
            }

            // every other NFO gets merged into that temp. movie object
            if (tempMovie != null) {
              try {
                tempMovie.merge(MovieNfoParser.parseNfo(mf.getFileAsPath()).toMovie());
              }
              catch (Exception ignored) {
              }
            }
          }

          // did we get movie data from our NFOs
          if (tempMovie != null) {
            // force merge it to the actual movie object
            movie.forceMerge(tempMovie);
            movie.saveToDb();
          }

          publishState(++i);
          if (cancel) {
            break;
          }
        }
      }
    });
  }
}