/*
 * Copyright (c) 2021 Huawei Device Co., Ltd.
 *
 * HDF is dual licensed: you can use it either under the terms of
 * the GPL, or the BSD license, at your option.
 * See the LICENSE file in the root of this repository for complete details.
 */

#include "codegen/java_client_proxy_code_emitter.h"
#include "util/logger.h"

namespace OHOS {
namespace HDI {
bool JavaClientProxyCodeEmitter::ResolveDirectory(const String& targetDirectory)
{
    if (ast_->GetASTFileType() == ASTFileType::AST_IFACE ||
        ast_->GetASTFileType() == ASTFileType::AST_ICALLBACK) {
        directory_ = GetFilePath(targetDirectory);
    } else {
        return false;
    }

    if (!File::CreateParentDir(directory_)) {
        Logger::E("CppClientInterfaceCodeEmitter", "Create '%s' failed!", directory_.string());
        return false;
    }

    return true;
}

void JavaClientProxyCodeEmitter::EmitCode()
{
    EmitProxyFile();
}

void JavaClientProxyCodeEmitter::EmitProxyFile()
{
    String filePath = String::Format("%s/%s.java", directory_.string(), FileName(proxyName_).string());
    File file(filePath, File::WRITE);
    StringBuilder sb;

    EmitLicense(sb);
    EmitPackage(sb);
    sb.Append("\n");
    EmitProxyImports(sb);
    sb.Append("\n");
    EmitProxyImpl(sb);

    String data = sb.ToString();
    file.WriteData(data.string(), data.GetLength());
    file.Flush();
    file.Close();
}

void JavaClientProxyCodeEmitter::EmitProxyImports(StringBuilder& sb)
{
    EmitProxyCorelibImports(sb);
    EmitProxySelfDefinedTypeImports(sb);
    EmitProxyDBinderImports(sb);
}

void JavaClientProxyCodeEmitter::EmitProxyCorelibImports(StringBuilder& sb)
{
    bool includeList = false;
    bool includeMap = false;
    const AST::TypeStringMap& types = ast_->GetTypes();
    for (const auto& pair : types) {
        AutoPtr<ASTType> type = pair.second;
        switch (type->GetTypeKind()) {
            case TypeKind::TYPE_LIST: {
                if (!includeList) {
                    sb.Append("import java.util.List;\n");
                    includeList = true;
                }
                break;
            }
            case TypeKind::TYPE_MAP: {
                if (!includeMap) {
                    sb.Append("import java.util.Map;\n");
                    sb.Append("import java.util.HashMap;\n");
                    includeMap = true;
                }
                break;
            }
            default:
                break;
        }
    }
}

void JavaClientProxyCodeEmitter::EmitProxySelfDefinedTypeImports(StringBuilder& sb)
{
    for (const auto& importPair : ast_->GetImports()) {
        AutoPtr<AST> import = importPair.second;
        sb.AppendFormat("import %s;\n", import->GetFullName().string());
    }
}

void JavaClientProxyCodeEmitter::EmitProxyDBinderImports(StringBuilder& sb)
{
    sb.Append("import ohos.hiviewdfx.HiLog;\n");
    sb.Append("import ohos.hiviewdfx.HiLogLabel;\n");
    sb.Append("import ohos.rpc.IRemoteObject;\n");
    sb.Append("import ohos.rpc.RemoteException;\n");
    sb.Append("import ohos.rpc.MessageParcel;\n");
    sb.Append("import ohos.rpc.MessageOption;\n");
}

void JavaClientProxyCodeEmitter::EmitProxyImpl(StringBuilder& sb)
{
    sb.AppendFormat("public class %s implements %s {\n", proxyName_.string(), interfaceName_.string());
    EmitProxyConstants(sb, g_tab);
    sb.Append("\n");
    sb.Append(g_tab).AppendFormat(
        "private static final HiLogLabel TAG = new HiLogLabel(HiLog.LOG_CORE, 0xD001510, \"%s\");\n",
        interfaceFullName_.string());
    sb.Append(g_tab).Append("private final IRemoteObject remote;\n");
    sb.Append(g_tab).Append("private static final int ERR_OK = 0;\n");
    sb.Append("\n");
    EmitProxyConstructor(sb, g_tab);
    sb.Append("\n");
    EmitProxyMethodImpls(sb, g_tab);
    sb.Append("};");
}

void JavaClientProxyCodeEmitter::EmitProxyConstants(StringBuilder& sb, const String& prefix)
{
    sb.Append(prefix).AppendFormat("private static final String DESCRIPTOR = \"%s\";\n\n",
        interfaceFullName_.string());
    EmitInterfaceMethodCommands(sb, prefix);
}

void JavaClientProxyCodeEmitter::EmitProxyConstructor(StringBuilder& sb, const String& prefix)
{
    sb.Append(prefix).AppendFormat("public %s(IRemoteObject remote) {\n", proxyName_.string());
    sb.Append(prefix + g_tab).Append("this.remote = remote;\n");
    sb.Append(prefix).Append("}\n");
    sb.Append("\n");
    sb.Append(prefix).AppendFormat("@Override\n");
    sb.Append(prefix).Append("public IRemoteObject asObject() {\n");
    sb.Append(prefix + g_tab).Append("return remote;\n");
    sb.Append(prefix).Append("}\n");
}

void JavaClientProxyCodeEmitter::EmitProxyMethodImpls(StringBuilder& sb, const String& prefix)
{
    for (size_t i = 0; i < interface_->GetMethodNumber(); i++) {
        AutoPtr<ASTMethod> method = interface_->GetMethod(i);
        EmitProxyMethodImpl(method, sb, prefix);
        if (i + 1 < interface_->GetMethodNumber()) {
            sb.Append("\n");
        }
    }
}

void JavaClientProxyCodeEmitter::EmitProxyMethodImpl(const AutoPtr<ASTMethod>& method, StringBuilder& sb,
    const String& prefix)
{
    sb.Append(prefix).Append("@Override\n");
    if (method->GetParameterNumber() == 0) {
        sb.Append(prefix).AppendFormat("public int %s() throws RemoteException ",
            MethodName(method->GetName()).string());
    } else {
        StringBuilder paramStr;
        paramStr.Append(prefix).AppendFormat("public int %s(", MethodName(method->GetName()).string());
        for (size_t i = 0; i < method->GetParameterNumber(); i++) {
            AutoPtr<ASTParameter> param = method->GetParameter(i);
            EmitInterfaceMethodParameter(param, paramStr, "");
            if (i + 1 < method->GetParameterNumber()) {
                paramStr.Append(", ");
            }
        }
        paramStr.Append(") throws RemoteException");

        sb.Append(SpecificationParam(paramStr, prefix + g_tab));
        sb.Append("\n");
    }
    EmitProxyMethodBody(method, sb, prefix);
}

void JavaClientProxyCodeEmitter::EmitInterfaceMethodParameter(const AutoPtr<ASTParameter>& param, StringBuilder& sb,
    const String& prefix)
{
    sb.Append(prefix).Append(param->EmitJavaParameter());
}

void JavaClientProxyCodeEmitter::EmitProxyMethodBody(const AutoPtr<ASTMethod>& method, StringBuilder& sb,
    const String& prefix)
{
    sb.Append(prefix).Append("{\n");
    sb.Append(prefix + g_tab).Append("MessageParcel data = MessageParcel.obtain();\n");
    sb.Append(prefix + g_tab).Append("MessageParcel reply = MessageParcel.obtain();\n");
    sb.Append(prefix + g_tab).AppendFormat("MessageOption option = new MessageOption(MessageOption.TF_SYNC);\n");
    sb.Append("\n");
    sb.Append(prefix).AppendFormat("    data.writeInterfaceToken(DESCRIPTOR);\n");

    for (size_t i = 0; i < method->GetParameterNumber(); i++) {
        AutoPtr<ASTParameter> param = method->GetParameter(i);
        param->EmitJavaWriteVar("data", sb, prefix + g_tab);
    }
    sb.Append("\n");

    sb.Append(prefix + g_tab).Append("try {\n");
    sb.Append(prefix + g_tab + g_tab).AppendFormat("if (remote.sendRequest(COMMAND_%s, data, reply, option)) {\n",
        ConstantName(method->GetName()).string());
    sb.Append(prefix + g_tab + g_tab + g_tab).Append("return 1;\n");
    sb.Append(prefix + g_tab + g_tab).Append("}\n");
    sb.Append(prefix + g_tab).Append("    reply.readException();\n");
    for (size_t i = 0; i < method->GetParameterNumber(); i++) {
        AutoPtr<ASTParameter> param = method->GetParameter(i);
        param->EmitJavaReadVar("reply", sb, prefix + g_tab + g_tab);
    }

    sb.Append(prefix + g_tab).Append("} finally {\n");
    sb.Append(prefix + g_tab + g_tab).Append("data.reclaim();\n");
    sb.Append(prefix + g_tab + g_tab).Append("reply.reclaim();\n");
    sb.Append(prefix + g_tab).Append("}\n");
    sb.Append(prefix + g_tab).Append("return 0;\n");
    sb.Append(prefix).Append("}\n");
}

void JavaClientProxyCodeEmitter::EmitLocalVariable(const AutoPtr<ASTParameter>& param, StringBuilder& sb,
    const String& prefix)
{
    AutoPtr<ASTType> type = param->GetType();
    if (type->GetTypeKind() == TypeKind::TYPE_SEQUENCEABLE) {
        sb.Append(prefix).AppendFormat("%s %s = new %s();\n",
            type->EmitJavaType(TypeMode::NO_MODE).string(), param->GetName().string(),
            type->EmitJavaType(TypeMode::NO_MODE).string());
    } else if (type->GetTypeKind() == TypeKind::TYPE_LIST) {
        sb.Append(prefix).AppendFormat("%s %s = new Array%s();\n",
            type->EmitJavaType(TypeMode::NO_MODE).string(), param->GetName().string(),
            type->EmitJavaType(TypeMode::NO_MODE).string());
    } else if (type->GetTypeKind() == TypeKind::TYPE_MAP) {
        sb.Append(prefix).AppendFormat("%s %s = new Hash%s();\n",
            type->EmitJavaType(TypeMode::NO_MODE).string(), param->GetName().string(),
            type->EmitJavaType(TypeMode::NO_MODE).string());
    } else {
        sb.Append(prefix).AppendFormat("%s %s;\n", type->EmitJavaType(TypeMode::NO_MODE).string(),
            param->GetName().string());
    }
}
} // namespace HDI
} // namespace OHOS