// Copyright 2015 The Cobalt Authors. All Rights Reserved.
//
// 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.

#include "cobalt/dom/source_buffer.h"

#include <vector>

#include "base/compiler_specific.h"
#include "base/file_path.h"
#include "base/logging.h"
#include "base/path_service.h"
#include "cobalt/dom/dom_exception.h"
#include "cobalt/dom/media_source.h"

namespace cobalt {
namespace dom {

SourceBuffer::SourceBuffer(const scoped_refptr<MediaSource>& media_source,
                           const std::string& id)
    : media_source_(media_source), id_(id), timestamp_offset_(0) {
  DCHECK(media_source_);
  DCHECK(!id_.empty());
}

scoped_refptr<TimeRanges> SourceBuffer::buffered(
    script::ExceptionState* exception_state) const {
  if (!media_source_) {
    DOMException::Raise(DOMException::kInvalidStateErr, exception_state);
    // Return value should be ignored.
    return NULL;
  }

  return media_source_->GetBuffered(this, exception_state);
}

double SourceBuffer::timestamp_offset(
    script::ExceptionState* exception_state) const {
  UNREFERENCED_PARAMETER(exception_state);
  return timestamp_offset_;
}

void SourceBuffer::set_timestamp_offset(
    double offset, script::ExceptionState* exception_state) {
  if (!media_source_) {
    DOMException::Raise(DOMException::kInvalidStateErr, exception_state);
    return;
  }

  if (media_source_->SetTimestampOffset(this, offset, exception_state)) {
    timestamp_offset_ = offset;
  }
}

void SourceBuffer::Append(const script::Handle<script::Uint8Array>& data,
                          script::ExceptionState* exception_state) {
  if (!media_source_) {
    DOMException::Raise(DOMException::kInvalidStateErr, exception_state);
  }

  if (!data.IsEmpty() && data->Length()) {
    media_source_->Append(this, data->Data(), static_cast<int>(data->Length()),
                          exception_state);
  }
}

void SourceBuffer::Abort(script::ExceptionState* exception_state) {
  if (!media_source_) {
    DOMException::Raise(DOMException::kInvalidStateErr, exception_state);
    return;
  }

  media_source_->Abort(this, exception_state);
}

void SourceBuffer::Close() { media_source_ = NULL; }

void SourceBuffer::TraceMembers(script::Tracer* tracer) {
  EventTarget::TraceMembers(tracer);

  tracer->Trace(media_source_);
}

}  // namespace dom
}  // namespace cobalt
