googet_addrepo.go

/*
Copyright 2016 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 main

import (
	"context"
	"flag"
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"github.com/go-yaml/yaml"
	"github.com/google/googet/v2/oswrap"
	"github.com/google/googet/v2/priority"
	"github.com/google/logger"
	"github.com/google/subcommands"
)

type addRepoCmd struct {
	file     string
	priority string
}

func (*addRepoCmd) Name() string     { return "addrepo" }
func (*addRepoCmd) Synopsis() string { return "add repository" }
func (*addRepoCmd) Usage() string {
	return fmt.Sprintf(`%s addrepo [-file <repofile>] [-priority <value>] <name> <url>:
	Add repository to GooGet's repository list. 
	If -file is not set 'name.repo' will be used for the file name 
	overwriting any existing file with than name. 
	If -file is set the specified repo will be appended to that repo file, 
	creating it if it does not exist.
	If -priority is specified, the repo will be configured with this priority level.
`, filepath.Base(os.Args[0]))
}

func (cmd *addRepoCmd) SetFlags(f *flag.FlagSet) {
	f.StringVar(&cmd.file, "file", "", "repo file to add this repository to")
	f.StringVar(&cmd.priority, "priority", "", "priority level assigned to repository")
}

func (cmd *addRepoCmd) Execute(_ context.Context, f *flag.FlagSet, _ ...interface{}) subcommands.ExitStatus {
	var newEntry repoEntry
	switch f.NArg() {
	case 0, 1:
		fmt.Fprintln(os.Stderr, "Not enough arguments")
		f.Usage()
		return subcommands.ExitUsageError
	case 2:
		newEntry.Name = f.Arg(0)
		newEntry.URL = f.Arg(1)
	default:
		fmt.Fprintln(os.Stderr, "Excessive arguments")
		f.Usage()
		return subcommands.ExitUsageError
	}

	if cmd.file == "" {
		cmd.file = newEntry.Name + ".repo"
	} else {
		if !strings.HasSuffix(cmd.file, ".repo") {
			fmt.Fprintln(os.Stderr, "Repo file name must end in '.repo'")
			return subcommands.ExitUsageError
		}
	}

	if cmd.priority != "" {
		var err error
		newEntry.Priority, err = priority.FromString(cmd.priority)
		if err != nil {
			fmt.Fprintf(os.Stderr, "Unrecognized priority value: %q\n", cmd.priority)
			return subcommands.ExitUsageError
		}
	}

	content, err := yaml.Marshal([]repoEntry{newEntry})
	if err != nil {
		logger.Fatal(err)
	}

	repoPath := filepath.Join(rootDir, repoDir, cmd.file)

	if _, err := oswrap.Stat(repoPath); err != nil && os.IsNotExist(err) {
		if err := writeRepoFile(repoFile{repoPath, []repoEntry{newEntry}}); err != nil {
			logger.Fatal(err)
		}
		fmt.Printf("Wrote repo file %s with content:\n%s\n", repoPath, content)
		return subcommands.ExitSuccess
	}

	rf, err := unmarshalRepoFile(repoPath)
	if err != nil {
		logger.Fatal(err)
	}

	var res []repoEntry
	for _, re := range rf.repoEntries {
		if re.Name != newEntry.Name && re.URL != newEntry.URL {
			res = append(res, re)
		}
	}

	res = append(res, newEntry)
	rf = repoFile{rf.fileName, res}

	if err := writeRepoFile(rf); err != nil {
		logger.Fatal(err)
	}

	fmt.Printf("Appended to repo file %s with the following content:\n%s\n", repoPath, content)

	return subcommands.ExitSuccess
}