remoting/webapp/crd/js/video_frame_recorder.js

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 /**
   6  * @fileoverview
   7  * Class implement the video frame recorder extension client.
   8  */
   9
  10 'use strict';
  11
  12 /** @suppress {duplicate} */
  13 var remoting = remoting || {};
  14
  15 /**
  16  * @constructor
  17  * @param {remoting.ClientPlugin} plugin
  18  */
  19 remoting.VideoFrameRecorder = function(plugin) {
  20   this.fileWriter_ = null;
  21   this.isRecording_ = false;
  22   this.plugin_ = plugin;
  23 };
  24
  25 /**
  26  * Starts or stops video recording.
  27  */
  28 remoting.VideoFrameRecorder.prototype.startStopRecording = function() {
  29   var data = {};
  30   if (this.isRecording_) {
  31     this.isRecording_ = false;
  32     data = { type: 'stop' }
  33
  34     chrome.fileSystem.chooseEntry(
  35         {type: 'saveFile', suggestedName: 'videoRecording.pb'},
  36         this.onFileChosen_.bind(this));
  37   } else {
  38     this.isRecording_ = true;
  39     data = { type: 'start' }
  40   }
  41   this.plugin_.sendClientMessage('video-recorder', JSON.stringify(data));
  42 }
  43
  44 /**
  45  * Returns true if the session is currently being recorded.
  46  * @return {boolean}
  47  */
  48 remoting.VideoFrameRecorder.prototype.isRecording = function() {
  49   return this.isRecording_;
  50 }
  51
  52 /**
  53  * Handles 'video-recorder' extension messages and returns true. Returns
  54  * false for all other message types.
  55  *
  56  * @param {string} type Type of extension message.
  57  * @param {Object} message The parsed extension message data.
  58  * @return {boolean}
  59  */
  60 remoting.VideoFrameRecorder.prototype.handleMessage =
  61     function(type, message) {
  62   if (type != 'video-recorder') {
  63     return false;
  64   }
  65
  66   var messageType = getStringAttr(message, 'type');
  67   var messageData = getStringAttr(message, 'data');
  68
  69   if (messageType == 'next-frame-reply') {
  70     if (!this.fileWriter_) {
  71       console.log("Received frame but have no writer");
  72       return true;
  73     }
  74     if (!messageData) {
  75       console.log("Finished receiving frames");
  76       this.fileWriter_ = null;
  77       return true;
  78     }
  79
  80     console.log("Received frame");
  81     var videoPacketString = window.atob(messageData);
  82
  83     console.log("Converted from Base64 - length:" + videoPacketString.length);
  84     var byteArrays = [];
  85
  86     for (var offset = 0; offset < videoPacketString.length; offset += 512) {
  87         var slice = videoPacketString.slice(offset, offset + 512);
  88         var byteNumbers = new Array(slice.length);
  89         for (var i = 0; i < slice.length; i++) {
  90             byteNumbers[i] = slice.charCodeAt(i);
  91         }
  92         var byteArray = new Uint8Array(byteNumbers);
  93         byteArrays.push(byteArray);
  94     }
  95
  96     console.log("Writing frame");
  97     videoPacketString = null;
  98     /**
  99      * @param {Array} parts
 100      * @return {Blob}
 101      */
 102     var makeBlob = function(parts) {
 103       return new Blob(parts);
 104     }
 105     var videoPacketBlob = makeBlob(byteArrays);
 106     byteArrays = null;
 107
 108     this.fileWriter_.write(videoPacketBlob);
 109
 110     return true;
 111   }
 112
 113   console.log("Unrecognized message: " + messageType);
 114   return true;
 115 }
 116
 117 /**
 118  * @param {Entry} entry The single file entry if multiple files are not allowed.
 119  * @param {Array<FileEntry>} fileEntries List of file entries if multiple files
 120  *     are allowed.
 121  */
 122 remoting.VideoFrameRecorder.prototype.onFileChosen_ = function(
 123     entry, fileEntries) {
 124   if (!entry) {
 125     console.log("Cancelled save of video frames.");
 126   } else {
 127     chrome.fileSystem.getDisplayPath(entry, function(path) {
 128       console.log("Saving video frames to:" + path);
 129     });
 130     entry.createWriter(this.onFileWriter_.bind(this));
 131   }
 132 }
 133
 134 /** @param {FileWriter} fileWriter */
 135 remoting.VideoFrameRecorder.prototype.onFileWriter_ = function(fileWriter) {
 136   console.log("Obtained FileWriter for video frame write");
 137   fileWriter.onwriteend = this.onWriteComplete_.bind(this);
 138   this.fileWriter_ = fileWriter;
 139   this.fetchNextFrame_();
 140 }
 141
 142 remoting.VideoFrameRecorder.prototype.onWriteComplete_ = function(e) {
 143   console.log("Video frame write complete");
 144   this.fetchNextFrame_();
 145 }
 146
 147 remoting.VideoFrameRecorder.prototype.fetchNextFrame_ = function() {
 148   console.log("Request next video frame");
 149   var data = { type: 'next-frame' }
 150   this.plugin_.sendClientMessage('video-recorder', JSON.stringify(data));
 151 }