269 lines
8.3 KiB
Java
269 lines
8.3 KiB
Java
/* Copyright (c) 2008 Google Inc.
|
|
*
|
|
* 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 sample.youtube;
|
|
|
|
import com.google.gdata.client.media.ResumableGDataFileUploader;
|
|
import sample.util.SimpleCommandLineParser;
|
|
import com.google.gdata.client.youtube.YouTubeService;
|
|
import com.google.gdata.data.media.MediaFileSource;
|
|
import com.google.gdata.data.media.mediarss.MediaCategory;
|
|
import com.google.gdata.data.media.mediarss.MediaDescription;
|
|
import com.google.gdata.data.media.mediarss.MediaKeywords;
|
|
import com.google.gdata.data.media.mediarss.MediaTitle;
|
|
import com.google.gdata.data.youtube.VideoEntry;
|
|
import com.google.gdata.data.youtube.YouTubeMediaGroup;
|
|
import com.google.gdata.data.youtube.YouTubeNamespace;
|
|
import com.google.gdata.util.AuthenticationException;
|
|
import com.google.gdata.util.ServiceException;
|
|
import com.google.gdata.client.uploader.ProgressListener;
|
|
import com.google.gdata.client.uploader.ResumableHttpFileUploader;
|
|
|
|
import java.io.BufferedReader;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.io.InputStreamReader;
|
|
import java.io.PrintStream;
|
|
import java.net.URL;
|
|
|
|
/**
|
|
* Demonstrates YouTube Data API operation to upload large media files.
|
|
*
|
|
*
|
|
*/
|
|
public class YouTubeUploadClient {
|
|
|
|
/**
|
|
* The URL used to resumable upload
|
|
*/
|
|
public static final String RESUMABLE_UPLOAD_URL =
|
|
"http://uploads.gdata.youtube.com/resumable/feeds/api/users/default/uploads";
|
|
|
|
/** Time interval at which upload task will notify about the progress */
|
|
private static final int PROGRESS_UPDATE_INTERVAL = 1000;
|
|
|
|
/** Max size for each upload chunk */
|
|
private static final int DEFAULT_CHUNK_SIZE = 10000000;
|
|
|
|
/** Steam to print status messages to. */
|
|
PrintStream output;
|
|
|
|
/**
|
|
* Input stream for reading user input.
|
|
*/
|
|
private static final BufferedReader bufferedReader = new BufferedReader(
|
|
new InputStreamReader(System.in));
|
|
|
|
|
|
/**
|
|
* A {@link ProgressListener} implementation to track upload progress.
|
|
* The listener can track multiple uploads at the same time.
|
|
*/
|
|
private class FileUploadProgressListener implements ProgressListener {
|
|
public synchronized void progressChanged(ResumableHttpFileUploader uploader)
|
|
{
|
|
switch(uploader.getUploadState()) {
|
|
case COMPLETE:
|
|
output.println("Upload Completed");
|
|
break;
|
|
case CLIENT_ERROR:
|
|
output.println("Upload Failed");
|
|
break;
|
|
case IN_PROGRESS:
|
|
output.println(
|
|
String.format("%3.0f", uploader.getProgress() * 100) + "%");
|
|
break;
|
|
case NOT_STARTED:
|
|
output.println("Upload Not Started");
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
private YouTubeUploadClient(PrintStream out) {
|
|
this.output = out;
|
|
}
|
|
|
|
/**
|
|
* Uploads a new video to YouTube.
|
|
*
|
|
* @param service An authenticated YouTubeService object.
|
|
* @throws IOException Problems reading user input.
|
|
*/
|
|
private void uploadVideo(YouTubeService service)
|
|
throws IOException, ServiceException, InterruptedException {
|
|
|
|
output.println("First, type in the path to the movie file:");
|
|
File videoFile = new File(readLine());
|
|
if (!videoFile.exists()) {
|
|
output.println("Sorry, that video doesn't exist.");
|
|
return;
|
|
}
|
|
|
|
output.println(
|
|
"What is the MIME type of this file? (ex. 'video/quicktime' for .mov)");
|
|
MediaFileSource ms = new MediaFileSource(videoFile, readLine());
|
|
|
|
output.println("What should I call this video?");
|
|
String videoTitle = readLine();
|
|
|
|
VideoEntry newEntry = new VideoEntry();
|
|
YouTubeMediaGroup mg = newEntry.getOrCreateMediaGroup();
|
|
mg.addCategory(new MediaCategory(YouTubeNamespace.CATEGORY_SCHEME, "Tech"));
|
|
mg.setTitle(new MediaTitle());
|
|
mg.getTitle().setPlainTextContent(videoTitle);
|
|
mg.setKeywords(new MediaKeywords());
|
|
mg.getKeywords().addKeyword("gdata-test");
|
|
mg.setDescription(new MediaDescription());
|
|
mg.getDescription().setPlainTextContent(videoTitle);
|
|
|
|
FileUploadProgressListener listener = new FileUploadProgressListener();
|
|
ResumableGDataFileUploader uploader =
|
|
new ResumableGDataFileUploader.Builder(
|
|
service, new URL(RESUMABLE_UPLOAD_URL), ms, newEntry)
|
|
.title(videoTitle)
|
|
.trackProgress(listener, PROGRESS_UPDATE_INTERVAL)
|
|
.chunkSize(DEFAULT_CHUNK_SIZE)
|
|
.build();
|
|
|
|
uploader.start();
|
|
while (!uploader.isDone()) {
|
|
Thread.sleep(PROGRESS_UPDATE_INTERVAL);
|
|
}
|
|
|
|
switch(uploader.getUploadState()) {
|
|
case COMPLETE:
|
|
output.println("Uploaded successfully");
|
|
break;
|
|
case CLIENT_ERROR:
|
|
output.println("Upload Failed");
|
|
break;
|
|
default:
|
|
output.println("Unexpected upload status");
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* YouTubeUploadClient is a sample command line application that
|
|
* demonstrates how to upload large media files to youtube. This sample
|
|
* uses resumable upload feature to upload large media.
|
|
*
|
|
* @param args Used to pass the username and password of a test account.
|
|
*/
|
|
public static void main(String[] args) {
|
|
SimpleCommandLineParser parser = new SimpleCommandLineParser(args);
|
|
String username = parser.getValue("username", "user", "u");
|
|
String password = parser.getValue("password", "pass", "p");
|
|
String developerKey = parser.getValue("key", "k");
|
|
boolean help = parser.containsKey("help", "h");
|
|
|
|
if (help || username == null || password == null || developerKey == null) {
|
|
printUsage();
|
|
System.exit(1);
|
|
}
|
|
|
|
YouTubeService service = new YouTubeService("gdataSample-YouTubeAuth-1",
|
|
developerKey);
|
|
|
|
try {
|
|
service.setUserCredentials(username, password);
|
|
} catch (AuthenticationException e) {
|
|
System.out.println("Invalid login credentials.");
|
|
System.exit(1);
|
|
}
|
|
|
|
YouTubeUploadClient client = new YouTubeUploadClient(System.out);
|
|
|
|
while (true) {
|
|
try {
|
|
printMenu();
|
|
int choice = readInt();
|
|
|
|
switch(choice) {
|
|
case 1:
|
|
client.uploadVideo(service);
|
|
break;
|
|
case 0:
|
|
System.exit(1);
|
|
}
|
|
|
|
} catch (IOException e) {
|
|
// Communications error
|
|
System.err.println(
|
|
"There was a problem communicating with the service.");
|
|
e.printStackTrace();
|
|
} catch (ServiceException se) {
|
|
System.out.println("Sorry, your upload was invalid:");
|
|
System.out.println(se.getResponseBody());
|
|
se.printStackTrace();
|
|
} catch (InterruptedException ie) {
|
|
System.out.println("Upload interrupted");
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* Shows the usage of how to run the sample from the command-line.
|
|
*/
|
|
private static void printUsage() {
|
|
System.out.println("Usage: java YouTubeUploadClient.jar "
|
|
+ " --username <user@gmail.com> " + " --password <pass> "
|
|
+ " --key <developer key>");
|
|
}
|
|
|
|
/**
|
|
* Displays a menu of the main activities a user can perform.
|
|
*/
|
|
private static void printMenu() {
|
|
System.out.println("\n");
|
|
System.out.println("Choose one of the following demo options:");
|
|
System.out.println("\t1) Upload new video");
|
|
System.out.println("\t0) Exit");
|
|
}
|
|
|
|
/**
|
|
* Reads a line of text from the standard input.
|
|
*
|
|
* @throws IOException If unable to read a line from the standard input.
|
|
* @return A line of text read from the standard input.
|
|
*/
|
|
private static String readLine() throws IOException {
|
|
return bufferedReader.readLine();
|
|
}
|
|
|
|
/**
|
|
* Reads a line of text from the standard input and returns the parsed
|
|
* integer representation.
|
|
*
|
|
* @throws IOException If unable to read a line from the standard input.
|
|
* @return An integer read from the standard input.
|
|
*/
|
|
private static int readInt() throws IOException {
|
|
String input = readLine();
|
|
|
|
try {
|
|
return Integer.parseInt(input);
|
|
} catch (NumberFormatException nfe) {
|
|
return 0;
|
|
}
|
|
|
|
}
|
|
|
|
}
|