/*
Copyright IBM Corp. 2017 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 command

import (
	"path/filepath"

	"github.com/cloudflare/cfssl/log"
	"github.com/hyperledger/fabric-ca/api"
	"github.com/hyperledger/fabric-ca/lib"
	"github.com/hyperledger/fabric-ca/util"
	"github.com/pkg/errors"
	"github.com/spf13/cobra"
)

var errInput = errors.New("Invalid usage; either --revoke.name and/or both --revoke.serial and --revoke.aki are required")

func (c *ClientCmd) newRevokeCommand() *cobra.Command {
	revokeCmd := &cobra.Command{
		Use:   "revoke",
		Short: "Revoke an identity",
		Long:  "Revoke an identity with Fabric CA server",
		// PreRunE block for this command will check to make sure enrollment
		// information exists before running the command
		PreRunE: func(cmd *cobra.Command, args []string) error {
			if len(args) > 0 {
				return errors.Errorf(extraArgsError, args, cmd.UsageString())
			}

			err := c.ConfigInit()
			if err != nil {
				return err
			}

			log.Debugf("Client configuration settings: %+v", c.clientCfg)

			return nil
		},
		RunE: func(cmd *cobra.Command, args []string) error {
			err := c.runRevoke(cmd)
			if err != nil {
				return err
			}

			return nil
		},
	}
	util.RegisterFlags(c.myViper, revokeCmd.Flags(), &c.revokeParams, nil)
	return revokeCmd
}

// The client revoke main logic
func (c *ClientCmd) runRevoke(cmd *cobra.Command) error {
	log.Debug("Entered runRevoke")

	var err error

	client := lib.Client{
		HomeDir: filepath.Dir(c.cfgFileName),
		Config:  c.clientCfg,
	}

	id, err := client.LoadMyIdentity()
	if err != nil {
		return err
	}

	// aki and serial # are required to revoke a certificate. The enrollment ID
	// is required to revoke an identity. So, either aki and serial must be
	// specified OR enrollment ID must be specified, else return an error.
	// Note that all three can be specified, in which case server will revoke
	// certificate associated with the specified aki, serial number.
	if (c.clientCfg.Revoke.Name == "") && (c.clientCfg.Revoke.AKI == "" ||
		c.clientCfg.Revoke.Serial == "") {
		cmd.Usage()
		return errInput
	}

	req := &api.RevocationRequest{
		Name:   c.clientCfg.Revoke.Name,
		Serial: c.clientCfg.Revoke.Serial,
		AKI:    c.clientCfg.Revoke.AKI,
		Reason: c.clientCfg.Revoke.Reason,
		GenCRL: c.revokeParams.GenCRL,
		CAName: c.clientCfg.CAName,
	}
	result, err := id.Revoke(req)

	if err != nil {
		return err
	}
	log.Infof("Sucessfully revoked certificates: %+v", result.RevokedCerts)

	if req.GenCRL {
		return storeCRL(c.clientCfg, result.CRL)
	}
	return nil
}