Port gitsign-attest to cobra subcommand. (#195)

Signed-off-by: Billy Lynch <billy@chainguard.dev>

Signed-off-by: Billy Lynch <billy@chainguard.dev>

Author: wlynch

cmd/gitsign-attest/main.go

@@ -22,7 +22,8 @@ gitsign [flags]
 
 ### SEE ALSO
 
+* [gitsign attest](gitsign_attest.md)	 - add attestations to Git objects
 * [gitsign show](gitsign_show.md)	 - Show source predicate information
 * [gitsign version](gitsign_version.md)	 - print Gitsign version
 
-###### Auto generated by spf13/cobra on 21-Nov-2022
+###### Auto generated by spf13/cobra on 26-Nov-2022

docs/cli/gitsign.md

@@ -0,0 +1,22 @@
+## gitsign attest
+
+add attestations to Git objects
+
+```
+gitsign attest [flags]
+```
+
+### Options
+
+```
+  -f, --filepath string   attestation filepath
+  -h, --help              help for attest
+      --objtype string    [commit | tree] - Git object type to attest (default "commit")
+      --type string       specify a predicate type (slsaprovenance|link|spdx|spdxjson|cyclonedx|vuln|custom) or an URI (default "custom")
+```
+
+### SEE ALSO
+
+* [gitsign](gitsign.md)	 - Keyless Git signing with Sigstore!
+
+###### Auto generated by spf13/cobra on 26-Nov-2022

docs/cli/gitsign_attest.md

@@ -26,4 +26,4 @@ gitsign show [revision] [flags]
 
 * [gitsign](gitsign.md)	 - Keyless Git signing with Sigstore!
 
-###### Auto generated by spf13/cobra on 21-Nov-2022
+###### Auto generated by spf13/cobra on 26-Nov-2022

docs/cli/gitsign_show.md

@@ -16,4 +16,4 @@ gitsign version [flags]
 
 * [gitsign](gitsign.md)	 - Keyless Git signing with Sigstore!
 
-###### Auto generated by spf13/cobra on 21-Nov-2022
+###### Auto generated by spf13/cobra on 26-Nov-2022

docs/cli/gitsign_version.md

@@ -30,7 +30,7 @@ f44de7a (HEAD -> main) commit
 2b0ff1e commit 1
 760568f initial commit
 $ gitsign-attest -f test.json
-$ gitsign-attest -f spdx.sbom -type spdx
+$ gitsign-attest -f spdx.sbom --type spdx
 $ git checkout refs/attestations/commits
 $ tree
 .
@@ -51,7 +51,7 @@ preserve attestations for squash commits, or between sub-directories.
 ```sh
 $ git log --oneline --format="Commit: %h  Tree: %t" -1
 Commit: edd19d9  Tree: 853a6ca
-$ gitsign-attest -f test.json -t
+$ gitsign-attest -f test.json --objtype tree
 $ git checkout refs/attestations/trees
 $ tree .
 .

cmd/gitsign-attest/internal/attest/attest.go renamed to internal/attest/attest.go

@@ -0,0 +1,114 @@
+// Copyright 2022 The Sigstore Authors
+//
+// 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 attest
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/go-git/go-git/v5"
+	cosignopts "github.com/sigstore/cosign/cmd/cosign/cli/options"
+	"github.com/sigstore/cosign/cmd/cosign/cli/sign"
+	"github.com/sigstore/cosign/pkg/cosign"
+	"github.com/sigstore/gitsign/internal/attest"
+	"github.com/sigstore/gitsign/internal/config"
+	"github.com/spf13/cobra"
+)
+
+const (
+	attCommitRef = "refs/attestations/commits"
+	attTreeRef   = "refs/attestations/trees"
+
+	FlagObjectTypeCommit = "commit"
+	FlagObjectTypeTree   = "tree"
+)
+
+type options struct {
+	Config *config.Config
+
+	FlagObjectType      string
+	FlagPath            string
+	FlagAttestationType string
+}
+
+func (o *options) AddFlags(cmd *cobra.Command) {
+	cmd.Flags().StringVar(&o.FlagObjectType, "objtype", FlagObjectTypeCommit, "[commit | tree] - Git object type to attest")
+	cmd.Flags().StringVarP(&o.FlagPath, "filepath", "f", "", "attestation filepath")
+	cmd.Flags().StringVar(&o.FlagAttestationType, "type", "", `specify a predicate type (slsaprovenance|link|spdx|spdxjson|cyclonedx|vuln|custom) or an URI (default "custom")`)
+}
+
+func (o *options) Run(ctx context.Context) error {
+	repo, err := git.PlainOpen(".")
+	if err != nil {
+		return fmt.Errorf("error opening repo: %w", err)
+	}
+
+	head, err := repo.Head()
+	if err != nil {
+		return fmt.Errorf("error getting repository head: %w", err)
+	}
+
+	// If we're attaching the attestation to a tree, resolve the tree SHA.
+	sha := head.Hash()
+	refName := attCommitRef
+	if o.FlagObjectType == FlagObjectTypeTree {
+		commit, err := repo.CommitObject(head.Hash())
+		if err != nil {
+			return fmt.Errorf("error getting tree: %w", err)
+		}
+		sha = commit.TreeHash
+
+		refName = attTreeRef
+	}
+
+	sv, err := sign.SignerFromKeyOpts(ctx, "", "", cosignopts.KeyOpts{
+		FulcioURL:    o.Config.Fulcio,
+		RekorURL:     o.Config.Rekor,
+		OIDCIssuer:   o.Config.Issuer,
+		OIDCClientID: o.Config.ClientID,
+	})
+	if err != nil {
+		return fmt.Errorf("getting signer: %w", err)
+	}
+	defer sv.Close()
+
+	attestor := attest.NewAttestor(repo, sv, cosign.TLogUploadInTotoAttestation)
+
+	out, err := attestor.WriteFile(ctx, refName, sha, o.FlagPath, o.FlagAttestationType)
+	if err != nil {
+		return err
+	}
+	fmt.Println(out)
+
+	return nil
+}
+
+func New(cfg *config.Config) *cobra.Command {
+	o := &options{
+		Config: cfg,
+	}
+	cmd := &cobra.Command{
+		Use:   "attest",
+		Short: "add attestations to Git objects",
+		Args:  cobra.ArbitraryArgs,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := context.Background()
+			return o.Run(ctx)
+		},
+	}
+	o.AddFlags(cmd)
+
+	return cmd
+}

cmd/gitsign-attest/internal/attest/attest_test.go renamed to internal/attest/attest_test.go

@@ -18,6 +18,7 @@ package root
 import (
 	"github.com/spf13/cobra"
 
+	"github.com/sigstore/gitsign/internal/commands/attest"
 	"github.com/sigstore/gitsign/internal/commands/show"
 	"github.com/sigstore/gitsign/internal/commands/version"
 	"github.com/sigstore/gitsign/internal/config"
@@ -82,6 +83,7 @@ func New(cfg *config.Config) *cobra.Command {
 
 	rootCmd.AddCommand(version.New(cfg))
 	rootCmd.AddCommand(show.New(cfg))
+	rootCmd.AddCommand(attest.New(cfg))
 	o.AddFlags(rootCmd)
 
 	return rootCmd