agola/cmd/toolbox/cmd/archive.go
Simone Gotti 4c12b03544 toolbox: expand archive/unarchive starting tilde
Expand archive sourceDir(s) and unarchive DestDir
2019-04-13 14:56:35 +02:00

77 lines
1.8 KiB
Go

// Copyright 2019 Sorint.lab
//
// 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 cmd
import (
"bufio"
"encoding/json"
"log"
"os"
"github.com/mitchellh/go-homedir"
"github.com/sorintlab/agola/internal/toolbox/archive"
"github.com/sorintlab/agola/internal/util"
"github.com/spf13/cobra"
)
var cmdArchive = &cobra.Command{
Use: "archive",
Run: archiveRun,
Short: "Archive",
}
func init() {
CmdToolbox.AddCommand(cmdArchive)
}
func archiveRun(cmd *cobra.Command, args []string) {
br := bufio.NewReader(os.Stdin)
d := json.NewDecoder(br)
a := archive.Archive{}
if err := d.Decode(&a); err != nil {
log.Fatalf("err: %v", err)
}
log.Printf("archive: %s", util.Dump(a))
var out *os.File
if a.OutFile == "" {
out = os.Stdout
} else {
var err error
out, err = os.Create(a.OutFile)
if err != nil {
log.Fatalf("error creating %s: %v", a.OutFile, err)
}
defer out.Close()
}
// expand ~ in archiveinfos SourceDir
for i := range a.ArchiveInfos {
exp, err := homedir.Expand(a.ArchiveInfos[i].SourceDir)
if err != nil {
log.Fatalf("failed to expand dir %q: %v", a.ArchiveInfos[i].SourceDir, err)
}
a.ArchiveInfos[i].SourceDir = exp
}
if err := archive.CreateTar(a.ArchiveInfos, out); err != nil {
log.Fatalf("create tar error: %v", err)
}
}