compiler/src/main/kotlin/android/databinding/tool/writer/DataBinderWriter.kt

/*
 * Copyright (C) 2015 The Android Open Source Project
 * 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.
 */

package android.databinding.tool.writer

import android.databinding.tool.LayoutBinder

class DataBinderWriter(val pkg: String, val projectPackage: String, val className: String,
        val layoutBinders : List<LayoutBinder>, val minSdk : kotlin.Int) {
    fun write(brWriter : BRWriter) = kcode("") {
        nl("package $pkg;")
        nl("import $projectPackage.BR;")
        nl("class $className {") {
            tab("final static int TARGET_MIN_SDK = ${minSdk};")
            nl("")
            tab("public $className() {") {
            }
            tab("}")
            nl("")
            tab("public android.databinding.ViewDataBinding getDataBinder(android.databinding.DataBindingComponent bindingComponent, android.view.View view, int layoutId) {") {
                tab("switch(layoutId) {") {
                    layoutBinders.groupBy{it.getLayoutname()}.forEach {
                        val firstVal = it.value.get(0)
                        tab("case ${firstVal.getModulePackage()}.R.layout.${firstVal.getLayoutname()}:") {
                            if (it.value.size() == 1) {
                                if (firstVal.isMerge()) {
                                    tab("return new ${firstVal.getPackage()}.${firstVal.getImplementationName()}(bindingComponent, new android.view.View[]{view});")
                                } else {
                                    tab("return ${firstVal.getPackage()}.${firstVal.getImplementationName()}.bind(view, bindingComponent);")
                                }
                            } else {
                                // we should check the tag to decide which layout we need to inflate
                                tab("{") {
                                    tab("final Object tag = view.getTag();")
                                    tab("if(tag == null) throw new java.lang.RuntimeException(\"view must have a tag\");")
                                    it.value.forEach {
                                        tab("if (\"${it.getTag()}_0\".equals(tag)) {") {
                                            if (it.isMerge()) {
                                                tab("return new ${it.getPackage()}.${it.getImplementationName()}(bindingComponent, new android.view.View[]{view});")
                                            } else {
                                                tab("return new ${it.getPackage()}.${it.getImplementationName()}(bindingComponent, view);")
                                            }
                                        } tab("}")
                                    }
                                }tab("}")
                            }

                        }
                    }
                }
                tab("}")
                tab("return null;")
            }
            tab("}")

            tab("android.databinding.ViewDataBinding getDataBinder(android.databinding.DataBindingComponent bindingComponent, android.view.View[] views, int layoutId) {") {
                tab("switch(layoutId) {") {
                    layoutBinders.filter{it.isMerge()}.groupBy{it.getLayoutname()}.forEach {
                        val firstVal = it.value.get(0)
                        tab("case ${firstVal.getModulePackage()}.R.layout.${firstVal.getLayoutname()}:") {
                            if (it.value.size() == 1) {
                                tab("return new ${firstVal.getPackage()}.${firstVal.getImplementationName()}(bindingComponent, views);")
                            } else {
                                // we should check the tag to decide which layout we need to inflate
                                tab("{") {
                                    tab("final Object tag = views[0].getTag();")
                                    tab("if(tag == null) throw new java.lang.RuntimeException(\"view must have a tag\");")
                                    it.value.forEach {
                                        tab("if (\"${it.getTag()}_0\".equals(tag)) {") {
                                            tab("return new ${it.getPackage()}.${it.getImplementationName()}(bindingComponent, views);")
                                        } tab("}")
                                    }
                                }tab("}")
                            }
                        }
                    }
                }
                tab("}")
                tab("return null;")
            }
            tab("}")

            tab("int getLayoutId(String tag) {") {
                tab("if (tag == null) {") {
                    tab("return 0;");
                }
                tab("}")
                // String.hashCode is well defined in the API so we can rely on it being the same on the device and the host machine
                tab("final int code = tag.hashCode();");
                tab("switch(code) {") {
                    layoutBinders.groupBy {"${it.getTag()}_0".hashCode()}.forEach {
                        tab("case ${it.key}:") {
                            it.value.forEach {
                                tab("if(tag.equals(\"${it.getTag()}_0\"))") {
                                    tab("return ${it.getModulePackage()}.R.layout.${it.getLayoutname()};")
                                }
                            }
                            tab("break;")
                        }

                    }
                }
                tab("}")
                tab("return 0;")
            }
            tab("}")

            tab("String convertBrIdToString(int id) {") {
                tab("if (id < 0 || id >= InnerBrLookup.sKeys.length) {") {
                    tab("return null;")
                } tab("}")
                tab("return InnerBrLookup.sKeys[id];")
            } tab("}")

            tab("private static class InnerBrLookup {") {
                tab("static String[] sKeys = new String[]{") {
                    tab("\"_all\"")
                    brWriter.indexedProps.forEach {
                        tab(",\"${it.value}\"")
                    }
                }.app("};")
            } tab("}")
        }
        nl("}")
    }.generate()
}