//
// Copyright (c) SAS Institute Inc.
//
// 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 servecmd

import (
	"errors"
	"fmt"
	"log"
	"net"
	"net/http"

	"github.com/sassoftware/relic/cmdline/shared"
	"github.com/sassoftware/relic/server/daemon"
	"github.com/spf13/cobra"

	_ "net/http/pprof"
)

var ServeCmd = &cobra.Command{
	Use:   "serve",
	Short: "Offer signing services over a HTTPS API",
	RunE:  serveCmd,
}

var argTest bool

func init() {
	shared.RootCmd.AddCommand(ServeCmd)
	ServeCmd.Flags().BoolP("force", "f", false, "(ignored)")
	ServeCmd.Flags().BoolVarP(&argTest, "test", "t", false, "Test configuration and exit")
}

func MakeServer() (*daemon.Daemon, error) {
	if err := shared.InitConfig(); err != nil {
		return nil, err
	}
	if shared.CurrentConfig.Server == nil {
		return nil, errors.New("Missing server section in configuration file")
	}
	if shared.CurrentConfig.Server.KeyFile == "" {
		return nil, errors.New("Missing keyfile option in server configuration file")
	}
	if shared.CurrentConfig.Server.CertFile == "" {
		return nil, errors.New("Missing certfile option in server configuration file")
	}
	if shared.CurrentConfig.Clients == nil {
		return nil, errors.New("Missing clients section in configuration file")
	}
	if shared.CurrentConfig.Server.Listen == "" {
		shared.CurrentConfig.Server.Listen = ":6300"
	}
	return daemon.New(shared.CurrentConfig, argTest)
}

func listenDebug() error {
	if !shared.CurrentConfig.Server.ListenDebug {
		return nil
	}
	lis, err := net.Listen("tcp", ":0")
	if err != nil {
		return err
	}
	fmt.Printf("Serving debug info on http://%s/debug/pprof/\n", lis.Addr())
	go func() {
		// pprof installs itself into the default handler on import
		err := http.Serve(lis, nil)
		if err != nil {
			fmt.Println(err)
		}
	}()
	return nil
}

func serveCmd(cmd *cobra.Command, args []string) error {
	// let journald add timestamps
	log.SetFlags(0)
	srv, err := MakeServer()
	if err != nil {
		return shared.Fail(err)
	} else if argTest {
		fmt.Println("OK")
		return nil
	}
	go watchSignals(srv)
	if err := listenDebug(); err != nil {
		return err
	}
	if err := srv.Serve(); err != nil && err != http.ErrServerClosed {
		return shared.Fail(err)
	}
	return nil
}