// Copyright 2020 Google LLC
//
// 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
//
//     https://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 codebase let's you manage codebases
package codebase

import (
 "fmt"
 "os"
 "path/filepath"

 "android.googlesource.com/platform/tools/treble.git/hacksaw/config"
)

// Add a codebase to the list of supported codebases
func Add(name string, path string) (*config.Config, error) {
 absPath, err := filepath.Abs(path)
 if err != nil {
  return nil, err
 }
 //check that the codebase path is valid before adding
 _, err = os.Stat(absPath)
 if err != nil {
  return nil, err
 }
 cfg := config.GetConfig()
 if _, ok := cfg.Codebases[name]; ok {
  return cfg, fmt.Errorf("Codebase %s already exists", name)
 }
 cfg.Codebases[name] = absPath
 return cfg, err
}

// Remove an existing codebase
func Remove(remove string) (*config.Config, error) {
 cfg := config.GetConfig()
 _, ok := cfg.Codebases[remove]
 if !ok {
  return nil, fmt.Errorf("Codebase %s not found", remove)
 }
 delete(cfg.Codebases, remove)
 if cfg.DefaultCodebase == remove {
  cfg.DefaultCodebase = ""
 }
 return cfg, nil
}

// Default gets the default codebase
func Default() string {
 cfg := config.GetConfig()
 def := cfg.DefaultCodebase
 return def
}

// SetDefault sets the default codebase
func SetDefault(def string) error {
 cfg := config.GetConfig()
 _, ok := cfg.Codebases[def]
 if !ok {
  return fmt.Errorf("Codebase %s not found", def)
 }
 cfg.DefaultCodebase = def
 return nil
}

// List all supported codebases
func List() map[string]string {
 cfg := config.GetConfig()
 return cfg.Codebases
}

// GetDir retrieves the directory of a specific workspace
func GetDir(codebase string) (string, error) {
 cfg := config.GetConfig()
 dir, ok := cfg.Codebases[codebase]
 if !ok {
  return dir, fmt.Errorf("Codebase %s not found",
   codebase)
 }
 return dir, nil
}