/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 <log4cxx/logstring.h>
#include <log4cxx/helpers/fileoutputstream.h>
#include <log4cxx/helpers/exception.h>
#include <log4cxx/helpers/bytebuffer.h>
#include <apr_file_io.h>
#include <log4cxx/helpers/transcoder.h>
#if !defined(LOG4CXX)
#define LOG4CXX 1
#endif
#include <log4cxx/helpers/aprinitializer.h>

using namespace log4cxx;
using namespace log4cxx::helpers;

IMPLEMENT_LOG4CXX_OBJECT(FileOutputStream)

FileOutputStream::FileOutputStream(const LogString& filename,
    bool append) : pool(), fileptr(open(filename, append, pool)) {
}

FileOutputStream::FileOutputStream(const logchar* filename,
    bool append) : pool(), fileptr(open(filename, append, pool)) {
}

apr_file_t* FileOutputStream::open(const LogString& filename,
    bool append, Pool& pool) {
    apr_fileperms_t perm = APR_OS_DEFAULT;
    apr_int32_t flags = APR_WRITE | APR_CREATE;
    if (append) {
        flags |= APR_APPEND;
    } else {
        flags |= APR_TRUNCATE;
    }
    File fn;
    fn.setPath(filename);
    apr_file_t* fileptr = 0;
    apr_status_t stat = fn.open(&fileptr, flags, perm, pool);
    if (stat != APR_SUCCESS) {
      throw IOException(stat);
    }
    return fileptr;
}

FileOutputStream::~FileOutputStream() {
  if (fileptr != NULL && !APRInitializer::isDestructed) {
    apr_file_close(fileptr);
  }
}

void FileOutputStream::close(Pool& /* p */) {
  if (fileptr != NULL) {
    apr_status_t stat = apr_file_close(fileptr);
    if (stat != APR_SUCCESS) {
        throw IOException(stat);
    }
    fileptr = NULL;
  }
}

void FileOutputStream::flush(Pool& /* p */) {
}

void FileOutputStream::write(ByteBuffer& buf, Pool& /* p */ ) {
  if (fileptr == NULL) {
     throw IOException(-1);
  }
  size_t nbytes = buf.remaining();
  size_t pos = buf.position();
  const char* data = buf.data();
  while(nbytes > 0) {
    apr_status_t stat = apr_file_write(
      fileptr, data + pos, &nbytes);
    if (stat != APR_SUCCESS) {
      throw IOException(stat);
    }
    pos += nbytes;
    buf.position(pos);
    nbytes = buf.remaining();
  }
}