remove google drive integration and plex refresher implementation
This commit is contained in:
parent
f8757b1206
commit
fc1e3d8daa
|
|
@ -2,5 +2,3 @@
|
|||
/target/
|
||||
.DS_Store
|
||||
/MangranaCommons.iml
|
||||
/tokens/
|
||||
/src/main/resources/credentials.json
|
||||
|
|
|
|||
21
pom.xml
21
pom.xml
|
|
@ -6,7 +6,7 @@
|
|||
|
||||
<groupId>tv.mangrana</groupId>
|
||||
<artifactId>mangrana-commons</artifactId>
|
||||
<version>4.1.3</version>
|
||||
<version>6.0.0</version>
|
||||
|
||||
<properties>
|
||||
<maven.compiler.source>8</maven.compiler.source>
|
||||
|
|
@ -43,25 +43,6 @@
|
|||
<version>4.4</version>
|
||||
</dependency>
|
||||
|
||||
<!-- Google API -->
|
||||
<dependency>
|
||||
<groupId>com.google.apis</groupId>
|
||||
<artifactId>google-api-services-drive</artifactId>
|
||||
<version>v3-rev20220508-1.32.1</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.google.api-client</groupId>
|
||||
<artifactId>google-api-client</artifactId>
|
||||
<version>1.35.1</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.google.oauth-client</groupId>
|
||||
<artifactId>google-oauth-client-jetty</artifactId>
|
||||
<version>1.34.1</version>
|
||||
</dependency>
|
||||
|
||||
<!-- UT -->
|
||||
<dependency>
|
||||
<groupId>org.testng</groupId>
|
||||
|
|
|
|||
|
|
@ -1,85 +0,0 @@
|
|||
package org.o7planning.googledrive.example;
|
||||
|
||||
|
||||
import com.google.api.client.auth.oauth2.Credential;
|
||||
import com.google.api.client.extensions.java6.auth.oauth2.AuthorizationCodeInstalledApp;
|
||||
import com.google.api.client.extensions.jetty.auth.oauth2.LocalServerReceiver;
|
||||
import com.google.api.client.googleapis.auth.oauth2.GoogleAuthorizationCodeFlow;
|
||||
import com.google.api.client.googleapis.auth.oauth2.GoogleClientSecrets;
|
||||
import com.google.api.client.googleapis.javanet.GoogleNetHttpTransport;
|
||||
import com.google.api.client.http.HttpTransport;
|
||||
import com.google.api.client.json.JsonFactory;
|
||||
import com.google.api.client.json.gson.GsonFactory;
|
||||
import com.google.api.client.util.store.FileDataStoreFactory;
|
||||
import com.google.api.services.drive.Drive;
|
||||
import com.google.api.services.drive.DriveScopes;
|
||||
|
||||
import java.io.FileNotFoundException;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.io.InputStreamReader;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
|
||||
import static tv.mangrana.utils.Output.log;
|
||||
|
||||
public class GoogleDriveUtils {
|
||||
|
||||
private static final String APPLICATION_NAME = "Google Drive API Java Quickstart";
|
||||
|
||||
private static final JsonFactory JSON_FACTORY = GsonFactory.getDefaultInstance();
|
||||
|
||||
private static final List<String> SCOPES = Collections.singletonList(DriveScopes.DRIVE);
|
||||
|
||||
private static final String CREDENTIALS_FILE_PATH = "/credentials.json";
|
||||
|
||||
private static final String TOKENS_DIRECTORY_PATH = "tokens";
|
||||
|
||||
// Global instance of the HTTP transport.
|
||||
private static HttpTransport HTTP_TRANSPORT;
|
||||
|
||||
private static Drive _driveService;
|
||||
|
||||
static {
|
||||
try {
|
||||
HTTP_TRANSPORT = GoogleNetHttpTransport.newTrustedTransport();
|
||||
} catch (Throwable t) {
|
||||
t.printStackTrace();
|
||||
System.exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
public static Credential getCredentials() throws IOException {
|
||||
|
||||
// Load client secrets.
|
||||
InputStream in = GoogleDriveUtils.class.getResourceAsStream(CREDENTIALS_FILE_PATH);
|
||||
if (in == null) {
|
||||
throw new FileNotFoundException("Resource not found: " + CREDENTIALS_FILE_PATH);
|
||||
}
|
||||
GoogleClientSecrets clientSecrets = GoogleClientSecrets.load(JSON_FACTORY, new InputStreamReader(in));
|
||||
|
||||
// Build flow and trigger user authorization request.
|
||||
GoogleAuthorizationCodeFlow flow = new GoogleAuthorizationCodeFlow.Builder(
|
||||
HTTP_TRANSPORT, JSON_FACTORY, clientSecrets, SCOPES)
|
||||
.setDataStoreFactory(new FileDataStoreFactory(new java.io.File(TOKENS_DIRECTORY_PATH)))
|
||||
.setAccessType("offline")
|
||||
.build();
|
||||
LocalServerReceiver receiver = new LocalServerReceiver.Builder().setPort(8888).build();
|
||||
Credential credential = new AuthorizationCodeInstalledApp(flow, receiver).authorize("user");
|
||||
//returns an authorized Credential object.
|
||||
return credential;
|
||||
}
|
||||
|
||||
public static Drive getDriveService() throws IOException {
|
||||
if (_driveService != null) {
|
||||
return _driveService;
|
||||
}
|
||||
Credential credential = getCredentials();
|
||||
//
|
||||
_driveService = new Drive.Builder(HTTP_TRANSPORT, JSON_FACTORY, credential) //
|
||||
.setApplicationName(APPLICATION_NAME).build();
|
||||
log("Google Drive service initialized");
|
||||
return _driveService;
|
||||
}
|
||||
|
||||
}
|
||||
|
|
@ -1,36 +0,0 @@
|
|||
package tv.mangrana.google.api.client;
|
||||
|
||||
import tv.mangrana.exception.NoElementFoundException;
|
||||
import com.google.api.services.drive.Drive;
|
||||
import com.google.api.services.drive.model.TeamDrive;
|
||||
import org.o7planning.googledrive.example.GoogleDriveUtils;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.List;
|
||||
import java.util.Optional;
|
||||
|
||||
import static tv.mangrana.utils.Output.log;
|
||||
|
||||
public class QuickTester {
|
||||
|
||||
public static void main(String[] args) throws IOException, NoElementFoundException {
|
||||
new QuickTester()
|
||||
.listTDs();
|
||||
}
|
||||
|
||||
public QuickTester() throws IOException {
|
||||
service = GoogleDriveUtils.getDriveService();
|
||||
}
|
||||
|
||||
Drive service;
|
||||
|
||||
private void listTDs() throws IOException, NoElementFoundException {
|
||||
List<TeamDrive> list = Optional.ofNullable(
|
||||
service.teamdrives().list().execute().getTeamDrives())
|
||||
.orElseThrow(() -> new NoElementFoundException("No TDs found :("));
|
||||
|
||||
list.forEach(td -> log(td.getName()));
|
||||
log("Ok, if you have been seeing some TDs in the output, that means GoogleDriveUtils works \uD83D\uDC4D");
|
||||
}
|
||||
|
||||
}
|
||||
|
|
@ -1,162 +0,0 @@
|
|||
package tv.mangrana.google.api.client.gateway;
|
||||
|
||||
import tv.mangrana.exception.NoElementFoundException;
|
||||
import com.google.api.services.drive.Drive;
|
||||
import com.google.api.services.drive.model.File;
|
||||
import com.google.api.services.drive.model.FileList;
|
||||
import org.apache.commons.collections4.CollectionUtils;
|
||||
import org.o7planning.googledrive.example.GoogleDriveUtils;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.*;
|
||||
|
||||
import static tv.mangrana.utils.Output.log;
|
||||
|
||||
public class GoogleDriveApiGateway {
|
||||
|
||||
Drive service;
|
||||
|
||||
public enum GoogleElementType {FOLDER, VIDEO}
|
||||
|
||||
public GoogleDriveApiGateway() throws IOException {
|
||||
service = GoogleDriveUtils.getDriveService();
|
||||
}
|
||||
|
||||
public File lookupElementById(String elementId) throws IOException {
|
||||
return service.files()
|
||||
.get(elementId)
|
||||
.setSupportsTeamDrives(true)
|
||||
.execute();
|
||||
}
|
||||
|
||||
public File lookupElementByName(String elementName, GoogleElementType type, String relatedTeamDriveId) throws IOException, NoElementFoundException {
|
||||
String query = "name = '" + elementName.replace("'","\\'") + "'"
|
||||
+ " and trashed=false"
|
||||
+ getTypeFilterQuery(type);
|
||||
|
||||
FileList fileList =
|
||||
service.files()
|
||||
.list()
|
||||
.setCorpora("drive")
|
||||
.setTeamDriveId(relatedTeamDriveId)
|
||||
.setIncludeItemsFromAllDrives(true)
|
||||
.setSupportsTeamDrives(true)
|
||||
.setQ(query)
|
||||
.execute();
|
||||
|
||||
List<File> files = Optional.ofNullable(
|
||||
fileList.getFiles())
|
||||
.orElseThrow(() -> new NoElementFoundException("element not found by name: " + elementName));
|
||||
|
||||
if (CollectionUtils.isNotEmpty(files)) {
|
||||
if (files.size() > 1) {
|
||||
log("WARN: There are more than one matching element!!!");
|
||||
files.forEach(fl -> log("> element with name {0} an id {1}", fl.getName(), fl.getId()));
|
||||
}
|
||||
return files.get(0);
|
||||
} else {
|
||||
throw new NoElementFoundException("no elements in the list xO");
|
||||
}
|
||||
}
|
||||
|
||||
public List<File> getChildrenFromParent(File parent, boolean onlyFolders) {
|
||||
String query =
|
||||
"trashed=false and "+
|
||||
(onlyFolders ? "mimeType = 'application/vnd.google-apps.folder' and " : "")+
|
||||
"'"+parent.getId()+"' in parents";
|
||||
|
||||
return getChildrenCommonCall(query);
|
||||
}
|
||||
|
||||
public File getChildFromParentByName(String name, File parent, boolean onlyFolder) throws NoElementFoundException {
|
||||
String query = "name = '" + name.replace("'","\\'") + "'" +
|
||||
" and trashed=false and "+
|
||||
(onlyFolder ? "mimeType = 'application/vnd.google-apps.folder' and " : "")+
|
||||
"'"+parent.getId()+"' in parents";
|
||||
List<File> children = getChildrenCommonCall(query);
|
||||
if (children.isEmpty()) throw new NoElementFoundException("no elements in the list xO");
|
||||
if (children.size() > 1) log("WARNING: more than one element here not expected for <{0}> and parent <{1}>", name, parent.getName());
|
||||
return children.get(0);
|
||||
}
|
||||
|
||||
private List<File> getChildrenCommonCall(String query) {
|
||||
List<File> fullFileList = new ArrayList<>();
|
||||
String pageToken = null;
|
||||
do {
|
||||
try {
|
||||
FileList fileList = service.files()
|
||||
.list()
|
||||
.setQ(query)
|
||||
.setIncludeItemsFromAllDrives(true)
|
||||
.setSupportsTeamDrives(true)
|
||||
.setFields("nextPageToken, files(id, name)")
|
||||
.setPageToken(pageToken)
|
||||
.setOrderBy("name")
|
||||
.execute();
|
||||
|
||||
fullFileList.addAll(fileList.getFiles());
|
||||
pageToken = fileList.getNextPageToken();
|
||||
} catch (IOException e) {
|
||||
log("ERROR during api call");
|
||||
e.printStackTrace();
|
||||
}
|
||||
} while (pageToken != null);
|
||||
|
||||
return fullFileList;
|
||||
}
|
||||
|
||||
public void copyFile(String fileId, String destinationFolderId) throws IOException {
|
||||
File newFileReference = new File();
|
||||
newFileReference.setParents(Collections.singletonList(destinationFolderId));
|
||||
service.files()
|
||||
.copy(fileId, newFileReference)
|
||||
.setSupportsTeamDrives(true)
|
||||
.execute();
|
||||
}
|
||||
|
||||
public void moveFile(File file, String destinationFolderId, String newFileName) throws IOException {
|
||||
String parentId = getParent(file.getId());
|
||||
File newFileReference = new File();
|
||||
newFileReference.setName(newFileName);
|
||||
service.files()
|
||||
.update(file.getId(), newFileReference)
|
||||
.setSupportsTeamDrives(true)
|
||||
.setAddParents(destinationFolderId)
|
||||
.setRemoveParents(parentId)
|
||||
.setFields("id, name, parents")
|
||||
.execute();
|
||||
}
|
||||
|
||||
private String getParent(String fileId) throws IOException {
|
||||
File file = service.files()
|
||||
.get(fileId)
|
||||
.setFields("parents")
|
||||
.setSupportsTeamDrives(true)
|
||||
.execute();
|
||||
return file.getParents().get(0);
|
||||
}
|
||||
|
||||
public File createFolder(String name, String parentFolderId) throws IOException {
|
||||
File fileMetadata = new File();
|
||||
fileMetadata.setName(name);
|
||||
fileMetadata.setMimeType("application/vnd.google-apps.folder");
|
||||
fileMetadata.setParents(Collections.singletonList(parentFolderId));
|
||||
return service
|
||||
.files()
|
||||
.create(fileMetadata)
|
||||
.setSupportsTeamDrives(true)
|
||||
.execute();
|
||||
}
|
||||
|
||||
private String getTypeFilterQuery(GoogleElementType type) {
|
||||
switch (type) {
|
||||
case VIDEO:
|
||||
return " and mimeType contains 'video'";
|
||||
case FOLDER:
|
||||
return " and mimeType = 'application/vnd.google-apps.folder'";
|
||||
default:
|
||||
return "";
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
|
@ -1,6 +1,7 @@
|
|||
package tv.mangrana.jobs;
|
||||
|
||||
import org.apache.commons.lang.StringUtils;
|
||||
import tv.mangrana.exception.IncorrectWorkingReferencesException;
|
||||
|
||||
import java.io.FileWriter;
|
||||
import java.io.IOException;
|
||||
|
|
@ -10,7 +11,9 @@ import java.nio.file.Paths;
|
|||
import java.text.MessageFormat;
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.stream.Stream;
|
||||
|
||||
|
|
@ -26,8 +29,11 @@ public class JobsFileStorage {
|
|||
static final String ELEMENT_NAME = "element";
|
||||
|
||||
public final String JOB_LINE_FORMAT;
|
||||
private static final String RESUME_FILE = JobFileManager.getResumeFile();
|
||||
|
||||
public JobsFileStorage() {
|
||||
private final List<Map<String, String>> linesInfo;
|
||||
|
||||
public JobsFileStorage() throws IncorrectWorkingReferencesException {
|
||||
String preFormat = "{COMPLETED_DATE}: {5} | {JOB_TYPE}: {0} | {HASH}: {1} | {ARR_ID}: {2} | {INTERNET_DB_ID}: {3} | {ELEMENT_NAME}: {4}";
|
||||
JOB_LINE_FORMAT = preFormat
|
||||
.replace("{COMPLETED_DATE}", COMPLETED_DATE)
|
||||
|
|
@ -36,8 +42,36 @@ public class JobsFileStorage {
|
|||
.replace("{ARR_ID}", ARR_ID)
|
||||
.replace("{INTERNET_DB_ID}", INTERNET_DB_ID)
|
||||
.replace("{ELEMENT_NAME}", ELEMENT_NAME);
|
||||
|
||||
linesInfo = getInfoFromFile();
|
||||
}
|
||||
|
||||
private List<Map<String, String>> getInfoFromFile() throws IncorrectWorkingReferencesException {
|
||||
List<Map<String, String>> newLinesInfo = new ArrayList<>();
|
||||
try (Stream<String> linesStream = Files.lines(Paths.get(RESUME_FILE))) {
|
||||
linesStream
|
||||
.filter(probableLine -> StringUtils.isNotEmpty(probableLine) && probableLine.contains("|"))
|
||||
.forEach(line -> newLinesInfo.add(line2Map(line)));
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
throw new IncorrectWorkingReferencesException("A problem occurred when trying to get jobs file info");
|
||||
}
|
||||
return newLinesInfo;
|
||||
}
|
||||
|
||||
private Map<String, String> line2Map(String line) {
|
||||
Map<String, String> mappedLine = new HashMap<>();
|
||||
String[] infoList = line.split("\\|");
|
||||
for (String elementInfo : infoList) {
|
||||
if (!elementInfo.trim().startsWith(COMPLETED_DATE)) {
|
||||
String[] fieldValue = elementInfo.split(":");
|
||||
String field = fieldValue[0].trim();
|
||||
String value = fieldValue[1].trim();
|
||||
mappedLine.put(field, value);
|
||||
}
|
||||
}
|
||||
return mappedLine;
|
||||
}
|
||||
String resumeFile = JobFileManager.getResumeFile();
|
||||
|
||||
public void persistCompleted(String type, String downloadId, int arrId, int iId, String element, LocalDateTime time){
|
||||
String jobLine = MessageFormat.format(JOB_LINE_FORMAT,
|
||||
|
|
@ -55,7 +89,7 @@ public class JobsFileStorage {
|
|||
}
|
||||
|
||||
void addLine(String jobLine) {
|
||||
try (FileWriter fw = new FileWriter(resumeFile, true);
|
||||
try (FileWriter fw = new FileWriter(RESUME_FILE, true);
|
||||
PrintWriter pw = new PrintWriter(fw)) {
|
||||
|
||||
pw.println(jobLine);
|
||||
|
|
@ -65,31 +99,11 @@ public class JobsFileStorage {
|
|||
}
|
||||
|
||||
public String getIIDByElement(String element) {
|
||||
try (Stream<String> linesStream = Files.lines(Paths.get(resumeFile))) {
|
||||
return linesStream
|
||||
.map(this::line2Map)
|
||||
.filter(mappedLine -> mappedLine.get(ELEMENT_NAME).equals(element))
|
||||
return linesInfo.stream()
|
||||
.filter(mappedLine -> element.equals(mappedLine.get(ELEMENT_NAME)))
|
||||
.findAny()
|
||||
.map(found -> found.get(INTERNET_DB_ID))
|
||||
.orElse(null);
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
private Map<String, String> line2Map(String line) {
|
||||
Map<String, String> mappedLine = new HashMap<>();
|
||||
String[] infoList = line.split("\\|");
|
||||
for (String elementInfo : infoList) {
|
||||
if (!elementInfo.trim().startsWith(COMPLETED_DATE)) {
|
||||
String[] fieldValue = elementInfo.split(":");
|
||||
String field = fieldValue[0].trim();
|
||||
String value = fieldValue[1].trim();
|
||||
mappedLine.put(field, value);
|
||||
}
|
||||
}
|
||||
return mappedLine;
|
||||
}
|
||||
|
||||
String formatTime(LocalDateTime time) {
|
||||
|
|
|
|||
|
|
@ -1,119 +0,0 @@
|
|||
package tv.mangrana.plex.url;
|
||||
|
||||
import org.apache.http.HttpEntity;
|
||||
import org.apache.http.client.methods.CloseableHttpResponse;
|
||||
import org.apache.http.client.methods.HttpUriRequest;
|
||||
import org.apache.http.client.methods.RequestBuilder;
|
||||
import org.apache.http.impl.client.CloseableHttpClient;
|
||||
import org.apache.http.impl.client.HttpClients;
|
||||
import org.w3c.dom.Document;
|
||||
import org.xml.sax.SAXException;
|
||||
import tv.mangrana.config.CommonConfigFileLoader;
|
||||
import tv.mangrana.utils.EasyLogger;
|
||||
|
||||
import javax.xml.parsers.DocumentBuilder;
|
||||
import javax.xml.parsers.DocumentBuilderFactory;
|
||||
import javax.xml.parsers.ParserConfigurationException;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.net.URI;
|
||||
import java.net.URISyntaxException;
|
||||
|
||||
import static tv.mangrana.config.CommonConfigFileLoader.CommonProjectConfiguration.*;
|
||||
import static tv.mangrana.utils.Output.log;
|
||||
import static tv.mangrana.utils.rest.APIInterface.ProtocolURLMark.HTTPS;
|
||||
|
||||
public class PlexCommandLauncher {
|
||||
|
||||
private final EasyLogger logger;
|
||||
private final CommonConfigFileLoader<?> config;
|
||||
private final PlexLibrarySectionsResolver sectionResolver;
|
||||
|
||||
public PlexCommandLauncher(CommonConfigFileLoader<?> config) {
|
||||
this.config = config;
|
||||
this.logger = new EasyLogger();
|
||||
this.sectionResolver = new PlexLibrarySectionsResolver(this);
|
||||
}
|
||||
|
||||
// public static void main(String[] args) throws IncorrectWorkingReferencesException {
|
||||
// String toRefresh="/tv/Series/C/City on a Hill (2019)";
|
||||
// new PlexCommandLauncher(new CommonConfigFileLoader()).scanByPath(toRefresh);
|
||||
// }
|
||||
|
||||
boolean scanByPath(String plexPathToRefresh, String plexMountPath) {
|
||||
boolean ok = true;
|
||||
String plexRefreshURL = getPlexRefreshURL(plexPathToRefresh, plexMountPath);
|
||||
if (plexRefreshURL==null) return false;
|
||||
try (CloseableHttpClient httpclient = HttpClients.createDefault()) {
|
||||
|
||||
HttpUriRequest httpGET = RequestBuilder.get()
|
||||
.setUri(new URI(plexRefreshURL))
|
||||
.addParameter("path", plexPathToRefresh)
|
||||
.addParameter("X-Plex-Token", config.getConfig(PLEX_TOKEN))
|
||||
.build();
|
||||
httpclient.execute(httpGET);
|
||||
@SuppressWarnings("unused")
|
||||
String urlWithTokenHidden = httpGET.getURI().toString().replaceFirst(config.getConfig(PLEX_TOKEN), "__plex_token__");
|
||||
logger.nLog("Launched URL command: {0}", httpGET.getURI().toString());
|
||||
} catch (Exception e) {
|
||||
logger.nHLog("Some error has happened using the URL <{0}>", plexRefreshURL);
|
||||
e.printStackTrace();
|
||||
ok = false;
|
||||
}
|
||||
return ok;
|
||||
}
|
||||
|
||||
public Document retrieveSectionsInfo() {
|
||||
try (CloseableHttpClient httpclient = HttpClients.createDefault()) {
|
||||
String plexSectionsURL = getPlexSectionsURL();
|
||||
HttpUriRequest httpGET = RequestBuilder.get()
|
||||
.setUri(new URI(plexSectionsURL))
|
||||
.addParameter("X-Plex-Token", config.getConfig(PLEX_TOKEN))
|
||||
.build();
|
||||
|
||||
try (CloseableHttpResponse httpResponse = httpclient.execute(httpGET)) {
|
||||
|
||||
final HttpEntity entity = httpResponse.getEntity();
|
||||
if (entity != null) {
|
||||
try (InputStream inputStream = entity.getContent()) {
|
||||
DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
|
||||
DocumentBuilder builder = factory.newDocumentBuilder();
|
||||
return builder.parse(inputStream);
|
||||
} catch (ParserConfigurationException | SAXException e) {
|
||||
log("could not get the section information from plex url");
|
||||
}
|
||||
}
|
||||
}
|
||||
log("launched url command: "+httpGET.getURI().toString().replaceFirst(config.getConfig(PLEX_TOKEN), "__plex_token__"));
|
||||
} catch (URISyntaxException | IOException e) {
|
||||
log("could not refresh plex artist because of "+e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
private String getPlexRefreshURL(String fullDestinationPath, String plexMountPath) {
|
||||
try {
|
||||
String sectionId = sectionResolver.resolveSectionByPath(fullDestinationPath, plexMountPath);
|
||||
if (sectionId==null) {
|
||||
log("Could not resolve the section of the element in plex for "+fullDestinationPath);
|
||||
return null;
|
||||
} else {
|
||||
log("Captured section <{0}> of the element in plex for {1}", sectionId, fullDestinationPath);
|
||||
}
|
||||
String host = config.getConfig(PLEX_HOST);
|
||||
String uriFormat = config.getConfig(PLEX_SECTION_REFRESH_URI);
|
||||
String uri = uriFormat.replaceFirst("\\{section_id}", sectionId);
|
||||
return HTTPS.getMark() + host + uri;
|
||||
} catch (Exception e) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
private String getPlexSectionsURL() {
|
||||
String host = HTTPS.getMark() + config.getConfig(PLEX_HOST);
|
||||
String uri = config.getConfig(PLEX_SECTIONS_LIST_URI);
|
||||
return host + uri;
|
||||
}
|
||||
|
||||
}
|
||||
|
|
@ -1,57 +0,0 @@
|
|||
package tv.mangrana.plex.url;
|
||||
|
||||
|
||||
import com.sun.org.apache.xerces.internal.dom.DeferredElementImpl;
|
||||
import org.w3c.dom.Document;
|
||||
import org.w3c.dom.Node;
|
||||
import org.w3c.dom.NodeList;
|
||||
import tv.mangrana.config.CommonConfigFileLoader;
|
||||
import tv.mangrana.utils.Output;
|
||||
|
||||
import javax.xml.xpath.XPath;
|
||||
import javax.xml.xpath.XPathConstants;
|
||||
import javax.xml.xpath.XPathExpressionException;
|
||||
import javax.xml.xpath.XPathFactory;
|
||||
|
||||
public class PlexLibrarySectionsResolver {
|
||||
|
||||
private final PlexCommandLauncher commandLauncher;
|
||||
private Document sectionsInfo;
|
||||
|
||||
public PlexLibrarySectionsResolver(PlexCommandLauncher commandLauncher) {
|
||||
this.commandLauncher = commandLauncher;
|
||||
}
|
||||
|
||||
private void initSectionsIfNecessary() {
|
||||
if (sectionsInfo==null) {
|
||||
sectionsInfo = commandLauncher.retrieveSectionsInfo();
|
||||
}
|
||||
}
|
||||
|
||||
public String resolveSectionByPath(String fullDestinationPath, String plexMountPath) {
|
||||
String keyFolder = fullDestinationPath.replaceFirst(plexMountPath,"").split("/")[1];
|
||||
initSectionsIfNecessary();
|
||||
XPath xPath = XPathFactory.newInstance().newXPath();
|
||||
String startingLocationText = plexMountPath.concat("/").concat(keyFolder).concat("/");
|
||||
String directoryNodeOfLocation = getDirectoryKeyValue(sectionsInfo, xPath, startingLocationText);
|
||||
if (directoryNodeOfLocation == null) {
|
||||
startingLocationText = plexMountPath.concat("/").concat(keyFolder);
|
||||
//Output.log("but going to retry with {0}", startingLocationText);
|
||||
return getDirectoryKeyValue(sectionsInfo, xPath, startingLocationText);
|
||||
} else
|
||||
return directoryNodeOfLocation;
|
||||
}
|
||||
|
||||
private String getDirectoryKeyValue(Document xmlDocument, XPath xPath, String startingLocationText) {
|
||||
String expression = "/MediaContainer/Directory/Location[starts-with(@path, '"+ startingLocationText +"')]";
|
||||
try {
|
||||
NodeList candidatesNodes = (NodeList) xPath.compile(expression).evaluate(xmlDocument, XPathConstants.NODESET);
|
||||
Node directoryNodeOfLocation = candidatesNodes.item(0).getParentNode();
|
||||
return ((DeferredElementImpl) directoryNodeOfLocation).getAttribute("key");
|
||||
} catch (XPathExpressionException | NullPointerException e) {
|
||||
//Output.log("could not resolve the section of the element in plex "+startingLocationText);
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
}
|
||||
|
|
@ -1,38 +0,0 @@
|
|||
package tv.mangrana.plex.url;
|
||||
|
||||
import tv.mangrana.config.CommonConfigFileLoader;
|
||||
|
||||
import static tv.mangrana.config.CommonConfigFileLoader.CommonProjectConfiguration.*;
|
||||
|
||||
public class PlexRefresher {
|
||||
|
||||
private final PlexCommandLauncher commandLauncher;
|
||||
private final CommonConfigFileLoader<?> config;
|
||||
|
||||
public PlexRefresher(CommonConfigFileLoader<?> config) {
|
||||
this.commandLauncher = new PlexCommandLauncher(config);
|
||||
this.config = config;
|
||||
}
|
||||
|
||||
public boolean scanSerieByPath(String fullDestinationPath) {
|
||||
try {
|
||||
String sonarrPathStarter = config.getConfig(SONARR_PATHS_STARTER);
|
||||
String plexMountPath = config.getConfig(PLEX_SERIES_PATHS_STARTER);
|
||||
String path2Refresh = fullDestinationPath.replaceFirst(sonarrPathStarter, plexMountPath);
|
||||
return commandLauncher.scanByPath(path2Refresh, plexMountPath);
|
||||
} catch (Exception e) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
public boolean scanMovieByPath(String fullDestinationPath) {
|
||||
try {
|
||||
String radarrPathStarter = config.getConfig(RADARR_PATHS_STARTER);
|
||||
String plexMountPath = config.getConfig(PLEX_MOVIES_PATHS_STARTER);
|
||||
String pathToRefresh = fullDestinationPath.replaceFirst(radarrPathStarter, plexMountPath);
|
||||
return commandLauncher.scanByPath(pathToRefresh, plexMountPath);
|
||||
} catch (Exception e) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -1,8 +1,9 @@
|
|||
package tv.mangrana.radarr.api.client.gateway;
|
||||
|
||||
import tv.mangrana.radarr.api.schema.command.RefreshMoviesCommand;
|
||||
import tv.mangrana.radarr.api.schema.movie.Movie2Add;
|
||||
import tv.mangrana.radarr.api.schema.movie.MovieFile;
|
||||
import tv.mangrana.radarr.api.schema.movie.MovieResource;
|
||||
import tv.mangrana.radarr.api.schema.queue.Movie;
|
||||
import tv.mangrana.radarr.api.schema.queue.QueueResourcePagingResource;
|
||||
import tv.mangrana.utils.rest.APIInterface;
|
||||
|
||||
|
|
@ -34,7 +35,12 @@ public interface RadarrAPIInterface extends APIInterface {
|
|||
@GET
|
||||
@Path("/movie")
|
||||
@Produces({ MediaType.APPLICATION_JSON })
|
||||
List<MovieResource> movieLookupByTMDBid(@QueryParam("tmdbId") int tmdbId, @QueryParam("apikey") String apikey);
|
||||
List<MovieResource> getMovieByTMDBid(@QueryParam("tmdbId") int tmdbId, @QueryParam("apikey") String apikey);
|
||||
|
||||
@GET
|
||||
@Path("/movie/lookup/tmdb")
|
||||
@Produces({ MediaType.APPLICATION_JSON })
|
||||
MovieResource movieLookupByTMDBid(@QueryParam("tmdbId") int tmdbId, @QueryParam("apikey") String apikey);
|
||||
|
||||
@POST
|
||||
@Path("/command")
|
||||
|
|
@ -46,6 +52,26 @@ public interface RadarrAPIInterface extends APIInterface {
|
|||
@Consumes({ MediaType.APPLICATION_JSON })
|
||||
void updateMovie(MovieResource movie, @PathParam("id") int movieId, @QueryParam("apikey") String apikey);
|
||||
|
||||
@POST
|
||||
@Path("/movie")
|
||||
@Consumes({ MediaType.APPLICATION_JSON })
|
||||
void addMovie(Movie2Add movie, @QueryParam("apikey") String apikey);
|
||||
|
||||
@POST
|
||||
@Path("/movie/import")
|
||||
@Consumes({ MediaType.APPLICATION_JSON })
|
||||
void importMovie(MovieResource movie, @QueryParam("apikey") String apikey);
|
||||
|
||||
@GET
|
||||
@Path("/moviefile")
|
||||
@Produces({ MediaType.APPLICATION_JSON })
|
||||
List<MovieFile> getFileByMovieId(@QueryParam("movieId") int movieId, @QueryParam("apikey") String apikey);
|
||||
|
||||
@PUT
|
||||
@Path("/moviefile/{id}")
|
||||
@Consumes({ MediaType.APPLICATION_JSON })
|
||||
void updateFile(MovieFile file, @PathParam("id") int fileId, @QueryParam("apikey") String apikey);
|
||||
|
||||
@PUT
|
||||
@Path("/movie/{id}")
|
||||
@Consumes({ MediaType.APPLICATION_JSON })
|
||||
|
|
|
|||
|
|
@ -1,25 +1,33 @@
|
|||
package tv.mangrana.radarr.api.client.gateway;
|
||||
|
||||
import tv.mangrana.config.CommonConfigFileLoader;
|
||||
import tv.mangrana.exception.TooMuchTriesException;
|
||||
import tv.mangrana.radarr.api.schema.command.RefreshMoviesCommand;
|
||||
import tv.mangrana.radarr.api.schema.movie.Movie2Add;
|
||||
import tv.mangrana.radarr.api.schema.movie.MovieFile;
|
||||
import tv.mangrana.radarr.api.schema.movie.MovieResource;
|
||||
import tv.mangrana.radarr.api.schema.queue.QueueResourcePagingResource;
|
||||
import tv.mangrana.utils.EasyLogger;
|
||||
import tv.mangrana.utils.RetryEngine;
|
||||
import tv.mangrana.utils.RetryEngine.Action;
|
||||
import tv.mangrana.utils.rest.APIProxyBuilderSingleton;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.function.Supplier;
|
||||
|
||||
import static tv.mangrana.config.CommonConfigFileLoader.CommonProjectConfiguration.RADARR_API_HOST;
|
||||
import static tv.mangrana.config.CommonConfigFileLoader.CommonProjectConfiguration.RADARR_API_KEY;
|
||||
import static tv.mangrana.utils.Output.log;
|
||||
|
||||
public class RadarrApiGateway {
|
||||
|
||||
protected final EasyLogger logger;
|
||||
private final String apiKey;
|
||||
private final RadarrAPIInterface proxy;
|
||||
|
||||
public RadarrApiGateway(CommonConfigFileLoader<?> config) {
|
||||
apiKey = config.getConfig(RADARR_API_KEY);
|
||||
proxy = APIProxyBuilderSingleton.getRadarrInterface(config.getConfig(RADARR_API_HOST));
|
||||
logger = new EasyLogger();
|
||||
}
|
||||
|
||||
public QueueResourcePagingResource getQueue() {
|
||||
|
|
@ -28,33 +36,192 @@ public class RadarrApiGateway {
|
|||
|
||||
public MovieResource getMovieById(Integer movieId) {
|
||||
MovieResource movie = proxy.getMovieById(movieId, apiKey);
|
||||
log("retrieved movie from radarr with id "+movieId);
|
||||
logger.nLog("retrieved movie from radarr with id "+movieId);
|
||||
return movie;
|
||||
}
|
||||
|
||||
public List<MovieResource> movieLookupByTMDBid (int tmdbId) {
|
||||
public MovieResource getMovieByTMDBid (int tmdbId) throws TooMuchTriesException {
|
||||
Supplier<MovieResource> movieLookup = () -> {
|
||||
try {
|
||||
List<MovieResource> retrieved = proxy.getMovieByTMDBid(tmdbId, apiKey);
|
||||
if (!retrieved.isEmpty())
|
||||
return retrieved.get(0);
|
||||
else return null;
|
||||
} catch (Exception e) {
|
||||
logger.nLog("failed movieLookupByTMDBid for {0}. Will be retried.", String.valueOf(tmdbId));
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
};
|
||||
return createMovieRetryer("MovieLookupByTMDBid")
|
||||
.tryUntilGotDesired(movieLookup, 10);
|
||||
}
|
||||
|
||||
public MovieResource movieLookupByTMDBid (int tmdbId) throws TooMuchTriesException {
|
||||
Supplier<MovieResource> movieLookup = () -> {
|
||||
try {
|
||||
return proxy.movieLookupByTMDBid(tmdbId, apiKey);
|
||||
} catch (Exception e) {
|
||||
logger.nLog("failed movieLookupByTMDBid for {0}. Will be retried.", String.valueOf(tmdbId));
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
};
|
||||
RetryEngine<MovieResource> retrier = new RetryEngine<>("MovieLookupByTMDBid", 2, logger::nLog);
|
||||
return retrier.tryUntilGotDesired(movieLookup, 10);
|
||||
}
|
||||
|
||||
public boolean existMovieInRadarr(int tmdbId) throws TooMuchTriesException {
|
||||
Supplier<Boolean> checkExistence = () -> {
|
||||
try {
|
||||
List<MovieResource> retrieved = proxy.getMovieByTMDBid(tmdbId, apiKey);
|
||||
return retrieved.isEmpty() ? Boolean.FALSE : Boolean.TRUE;
|
||||
} catch (Exception e) {
|
||||
logger.nLog("failed getMovieByTMDBid for {0}. Will be retried.", String.valueOf(tmdbId));
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
};
|
||||
return createPredicateRetryer("MovieLookupByTMDBid")
|
||||
.tryUntilGotDesired(checkExistence, 10);
|
||||
}
|
||||
|
||||
public void removeQueueItem(int itemId) {
|
||||
proxy.removeQueueItem(itemId, false, apiKey);
|
||||
log("removed item from queue successfully: "+itemId);
|
||||
logger.nLog("removed item from queue successfully: "+itemId);
|
||||
}
|
||||
|
||||
public void refreshMovie(int movieId) {
|
||||
public void refreshMovie(int movieId) throws TooMuchTriesException {
|
||||
Supplier<RetryEngine.Action> refresh = () -> {
|
||||
try {
|
||||
proxy.refreshMoviesCommand(new RefreshMoviesCommand(movieId), apiKey);
|
||||
log("refreshed movie with id "+movieId);
|
||||
return Action.PERFORMED;
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
};
|
||||
createActionRetryer("RefreshMovie")
|
||||
.tryUntilGotDesired(refresh, 10);
|
||||
|
||||
logger.nLog("refreshed movie with id "+movieId);
|
||||
}
|
||||
|
||||
public void updateMovie(MovieResource movie){
|
||||
public void updateMovie(MovieResource movie) throws TooMuchTriesException {
|
||||
Supplier<RetryEngine.Action> refresh = () -> {
|
||||
try {
|
||||
proxy.updateMovie(movie, movie.getId(), apiKey);
|
||||
return Action.PERFORMED;
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
};
|
||||
createActionRetryer("updateMovie")
|
||||
.tryUntilGotDesired(refresh, 10);
|
||||
|
||||
logger.nLog("updated movie in Radarr with id "+movie.getId());
|
||||
}
|
||||
|
||||
public void relocateMovie(MovieResource movie){
|
||||
proxy.relocateMovie(movie, movie.getId(), true, apiKey);
|
||||
}
|
||||
|
||||
public List<MovieResource> movieLookupByTitle (String title) {
|
||||
return proxy.movieLookupByTitle(title, apiKey);
|
||||
public MovieResource movieLookupByTitle (String title) throws TooMuchTriesException {
|
||||
Supplier<MovieResource> movieLookup = () -> {
|
||||
try {
|
||||
List<MovieResource> retrieved = proxy.movieLookupByTitle(title, apiKey);
|
||||
if (!retrieved.isEmpty())
|
||||
return retrieved.get(0);
|
||||
else return null;
|
||||
} catch (Exception e) {
|
||||
logger.nLog("failed movieLookupByTitle for {0}. Will be retried.", title);
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
};
|
||||
return createMovieRetryer("MovieLookupByTitle")
|
||||
.tryUntilGotDesired(movieLookup, 10);
|
||||
}
|
||||
|
||||
public void importMovie(MovieResource movie) throws TooMuchTriesException {
|
||||
Supplier<RetryEngine.Action> refresh = () -> {
|
||||
try {
|
||||
proxy.importMovie(movie, apiKey);
|
||||
return Action.PERFORMED;
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
};
|
||||
createActionRetryer("addMovieMovie")
|
||||
.tryUntilGotDesired(refresh, 10);
|
||||
|
||||
logger.nLog("add movie in Radarr with id "+movie.getId());
|
||||
}
|
||||
|
||||
public void addMovie(Movie2Add movie) {
|
||||
proxy.addMovie(movie, apiKey);
|
||||
}
|
||||
|
||||
public MovieFile getFileByMovieId (int movieId) throws TooMuchTriesException {
|
||||
Supplier<MovieFile> movieLookup = () -> {
|
||||
try {
|
||||
List<MovieFile> retrieved = proxy.getFileByMovieId(movieId, apiKey);
|
||||
if (!retrieved.isEmpty())
|
||||
return retrieved.get(0);
|
||||
else return null;
|
||||
} catch (Exception e) {
|
||||
logger.nLog("failed movieLookupByTitle for {0}. Will be retried.", String.valueOf(movieId));
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
};
|
||||
RetryEngine<MovieFile> retryer = new RetryEngine<>("MovieLookupByTitle", 1, logger::nLog);
|
||||
return retryer.tryUntilGotDesired(movieLookup, 10);
|
||||
}
|
||||
|
||||
public boolean existFileByMovieId (int movieId) throws TooMuchTriesException {
|
||||
Supplier<Boolean> movieLookup = () -> {
|
||||
try {
|
||||
List<MovieFile> retrieved = proxy.getFileByMovieId(movieId, apiKey);
|
||||
return retrieved.isEmpty() ? Boolean.FALSE : Boolean.TRUE;
|
||||
} catch (Exception e) {
|
||||
logger.nLog("failed movieLookupByTitle for {0}. Will be retried.", String.valueOf(movieId));
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
};
|
||||
RetryEngine<Boolean> retryer = new RetryEngine<>("getFileByMovieId", 1, logger::nLog);
|
||||
return retryer.tryUntilGotDesired(movieLookup, 10);
|
||||
}
|
||||
|
||||
public void updateFile(MovieFile file) throws TooMuchTriesException {
|
||||
Supplier<RetryEngine.Action> refresh = () -> {
|
||||
try {
|
||||
proxy.updateFile(file, file.getId(), apiKey);
|
||||
return Action.PERFORMED;
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
};
|
||||
createActionRetryer("updateMovie")
|
||||
.tryUntilGotDesired(refresh, 10);
|
||||
|
||||
logger.nLog("updated file in Radarr with id "+file.getId());
|
||||
}
|
||||
|
||||
private RetryEngine<MovieResource> createMovieRetryer(String title) {
|
||||
return new RetryEngine<>(title,1,logger::nLog);
|
||||
}
|
||||
|
||||
private RetryEngine<Action> createActionRetryer(String title) {
|
||||
return new RetryEngine<>(title,2,logger::nLog);
|
||||
}
|
||||
|
||||
private RetryEngine<Boolean> createPredicateRetryer(String title) {
|
||||
return new RetryEngine<>(title,1,logger::nLog);
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
|||
|
|
@ -832,3 +832,4 @@ public class MovieResource {
|
|||
}
|
||||
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -36,6 +36,10 @@ public class RetryEngine<D> {
|
|||
}
|
||||
}
|
||||
|
||||
public enum Action {
|
||||
PERFORMED
|
||||
}
|
||||
|
||||
public RetryEngine(String title, int minutesToWait, Consumer<String> logger) {
|
||||
this(title, minutesToWait, null, logger);
|
||||
}
|
||||
|
|
|
|||
|
|
@ -34,4 +34,11 @@ public class StringCaptor {
|
|||
return "Temporada ".concat(episodeInfo.substring(1,3));
|
||||
}
|
||||
|
||||
public static int getTMDBFromFile(String path) throws IncorrectWorkingReferencesException {
|
||||
String tmdbId = Optional.ofNullable(
|
||||
StringCaptor.getMatchingSubstring(path.contains("/")?path.substring(path.lastIndexOf('/')):path, "\\{tmdb-(.*)\\}"))
|
||||
.orElseThrow(() -> new IncorrectWorkingReferencesException("Couldn't find tmdb from: "+path));
|
||||
return Integer.parseInt(tmdbId);
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue