// Copyright 2022 Google Inc. 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.

package config

import (
 "strings"

 "android/soong/android"
)

var (
 linuxArmCflags = []string{
  "-march=armv7a",
 }

 linuxArm64Cflags = []string{}

 linuxArmLdflags = []string{
  "-march=armv7a",
  "-Wl,--compress-debug-sections=zstd",
 }

 linuxArm64Ldflags = []string{
  "-Wl,--compress-debug-sections=zstd",
 }
)

func init() {
 pctx.StaticVariable("LinuxArmCflags", strings.Join(linuxArmCflags, " "))
 pctx.StaticVariable("LinuxArm64Cflags", strings.Join(linuxArm64Cflags, " "))
 pctx.StaticVariable("LinuxArmLdflags", strings.Join(linuxArmLdflags, " "))
 pctx.StaticVariable("LinuxArm64Ldflags", strings.Join(linuxArm64Ldflags, " "))

 pctx.StaticVariable("LinuxArmYasmFlags", "-f elf32 -m arm")
 pctx.StaticVariable("LinuxArm64YasmFlags", "-f elf64 -m aarch64")

}

// Musl arm+arm64
type toolchainLinuxArm struct {
 toolchain32Bit
 toolchainLinux
}

type toolchainLinuxArm64 struct {
 toolchain64Bit
 toolchainLinux
}

func (t *toolchainLinuxArm) Name() string {
 return "arm"
}

func (t *toolchainLinuxArm64) Name() string {
 return "arm64"
}

func (t *toolchainLinuxArm) Cflags() string {
 return "${config.LinuxCflags} ${config.LinuxArmCflags}"
}

func (t *toolchainLinuxArm) Cppflags() string {
 return ""
}

func (t *toolchainLinuxArm64) Cflags() string {
 return "${config.LinuxCflags} ${config.LinuxArm64Cflags}"
}

func (t *toolchainLinuxArm64) Cppflags() string {
 return ""
}

func (t *toolchainLinuxArm) Ldflags(ctx ToolchainFlagsContext) FlagsWithDeps {
 return FlagsWithDeps{
  Flags: "${config.LinuxLdflags} ${config.LinuxArmLdflags}",
 }
}

func (t *toolchainLinuxArm64) Ldflags(ctx ToolchainFlagsContext) FlagsWithDeps {
 return FlagsWithDeps{
  Flags: "${config.LinuxLdflags} ${config.LinuxArm64Ldflags}",
 }
}

func (t *toolchainLinuxArm) YasmFlags() string {
 return "${config.LinuxArmYasmFlags}"
}

func (t *toolchainLinuxArm64) YasmFlags() string {
 return "${config.LinuxArm64YasmFlags}"
}

func (toolchainLinuxArm) LibclangRuntimeLibraryArch() string {
 return "arm"
}

func (toolchainLinuxArm64) LibclangRuntimeLibraryArch() string {
 return "arm64"
}

func (t *toolchainLinuxArm) InstructionSetFlags(isa string) (string, error) {
 // TODO: Is no thumb OK?
 return t.toolchainBase.InstructionSetFlags("")
}

type toolchainLinuxMuslArm struct {
 toolchainLinuxArm
 toolchainMusl
}

type toolchainLinuxMuslArm64 struct {
 toolchainLinuxArm64
 toolchainMusl
}

func (t *toolchainLinuxMuslArm) ClangTriple() string {
 return "arm-linux-musleabihf"
}

func (t *toolchainLinuxMuslArm) Cflags() string {
 return t.toolchainLinuxArm.Cflags() + " " + t.toolchainMusl.Cflags()
}

func (t *toolchainLinuxMuslArm) Ldflags(ctx ToolchainFlagsContext) FlagsWithDeps {
 return t.toolchainLinuxArm.Ldflags(ctx).Append(t.toolchainMusl.Ldflags(ctx))
}

func (t *toolchainLinuxMuslArm64) ClangTriple() string {
 return "aarch64-linux-musl"
}

func (t *toolchainLinuxMuslArm64) Cflags() string {
 return t.toolchainLinuxArm64.Cflags() + " " + t.toolchainMusl.Cflags()
}

func (t *toolchainLinuxMuslArm64) Ldflags(ctx ToolchainFlagsContext) FlagsWithDeps {
 return t.toolchainLinuxArm64.Ldflags(ctx).Append(t.toolchainMusl.Ldflags(ctx))
}

var toolchainLinuxMuslArmSingleton Toolchain = &toolchainLinuxMuslArm{}
var toolchainLinuxMuslArm64Singleton Toolchain = &toolchainLinuxMuslArm64{}

func linuxMuslArmToolchainFactory(arch android.Arch) Toolchain {
 return toolchainLinuxMuslArmSingleton
}

func linuxMuslArm64ToolchainFactory(arch android.Arch) Toolchain {
 return toolchainLinuxMuslArm64Singleton
}

func init() {
 registerToolchainFactory(android.LinuxMusl, android.Arm, linuxMuslArmToolchainFactory)
 registerToolchainFactory(android.LinuxMusl, android.Arm64, linuxMuslArm64ToolchainFactory)
}