kotlin-native/runtime/src/main/cpp/Console.cpp

/*
 * Copyright 2010-2017 JetBrains s.r.o.
 *
 * 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 <cstdio>
#include <string>
#include <vector>

#include "KAssert.h"
#include "Memory.h"
#include "Natives.h"
#include "KString.h"
#include "Porting.h"
#include "Types.h"
#include "Exceptions.h"
#ifdef KONAN_ANDROID
#include "CompilerConstants.hpp"
#endif

#include "utf8.h"

using namespace kotlin;

namespace {

std::string kStringToUtf8(KConstRef message) {
    if (message->type_info() != theStringTypeInfo) {
        ThrowClassCastException(message, theStringTypeInfo);
    }
    return kotlin::to_string<KStringConversionMode::REPLACE_INVALID>(message);
}

} // namespace

extern "C" {

// io/Console.kt
void Kotlin_io_Console_print(KConstRef message) {
    // TODO: system stdout must be aware about UTF-8.
    auto utf8 = kStringToUtf8(message);
    kotlin::ThreadStateGuard guard(kotlin::ThreadState::kNative);
    konan::consoleWriteUtf8(utf8.c_str(), utf8.size());
}

void Kotlin_io_Console_printToStdErr(KConstRef message) {
    // TODO: system stderr must be aware about UTF-8.
    auto utf8 = kStringToUtf8(message);
    kotlin::ThreadStateGuard guard(kotlin::ThreadState::kNative);
    konan::consoleErrorUtf8(utf8.c_str(), utf8.size());
}

void Kotlin_io_Console_println(KConstRef message) {
    Kotlin_io_Console_print(message);
#ifndef KONAN_ANDROID
    Kotlin_io_Console_println0();
#else
    // On Android single print produces logcat entry, so no need in linefeed.
    if (!kotlin::compiler::printToAndroidLogcat()) {
        Kotlin_io_Console_println0();
    }
#endif
}

void Kotlin_io_Console_printlnToStdErr(KConstRef message) {
    Kotlin_io_Console_printToStdErr(message);
#ifndef KONAN_ANDROID
    Kotlin_io_Console_println0ToStdErr();
#else
    // On Android single print produces logcat entry, so no need in linefeed.
    if (!kotlin::compiler::printToAndroidLogcat()) {
        Kotlin_io_Console_println0ToStdErr();
    }
#endif
}

void Kotlin_io_Console_println0() {
    kotlin::ThreadStateGuard guard(kotlin::ThreadState::kNative);
    konan::consoleWriteUtf8("\n", 1);
}

void Kotlin_io_Console_println0ToStdErr() {
    kotlin::ThreadStateGuard guard(kotlin::ThreadState::kNative);
    konan::consoleErrorUtf8("\n", 1);
}

OBJ_GETTER0(Kotlin_io_Console_readLine) {
    char data[4096];
    int32_t result;
    {
        kotlin::ThreadStateGuard guard(kotlin::ThreadState::kNative);
        result = konan::consoleReadUtf8(data, sizeof(data));
    }
    if (result < 0) {
        RETURN_OBJ(nullptr);
    }
    RETURN_RESULT_OF(CreateStringFromCString, data);
}

OBJ_GETTER0(Kotlin_io_Console_readlnOrNull) {
    std::vector<char> data;
    data.reserve(16);
    bool isEOF = false;
    bool isError = false;
    {
        kotlin::ThreadStateGuard guard(kotlin::ThreadState::kNative);
        while (true) {
            int result = fgetc(stdin);

            if (result == EOF || result == '\n') {
                isEOF = (result == EOF);
                isError = (ferror(stdin) != 0);
                break;
            }

            data.push_back(result);
        }
    }
    if (isError) {
        ThrowIllegalStateException();
    }
    if (!isEOF && !data.empty() && data.back() == '\r') { // CRLF
        data.pop_back();
    }
    if (data.empty() && isEOF) {
        RETURN_OBJ(nullptr);
    }
    RETURN_RESULT_OF(CreateStringFromUtf8, data.data(), data.size());
}

} // extern "C"