gcp/icdemo/main.go

// Copyright 2019 Google LLC
//
// 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.

// Binary icdemo is a demo of IC.
package main

import (
	"flag"
	"net/http"
	"os"
	"strings"

	"github.com/GoogleCloudPlatform/healthcare-federated-access-services/lib/httputils" /* copybara-comment: httputils */
	"github.com/GoogleCloudPlatform/healthcare-federated-access-services/lib/osenv" /* copybara-comment: osenv */
	"github.com/GoogleCloudPlatform/healthcare-federated-access-services/lib/serviceinfo" /* copybara-comment: serviceinfo */
	"github.com/GoogleCloudPlatform/healthcare-federated-access-services/lib/srcutil" /* copybara-comment: srcutil */

	glog "github.com/golang/glog" /* copybara-comment */
)

var (
	hydraURL = osenv.MustVar("HYDRA_PUBLIC_URL")
	icURL    = osenv.MustVar("IC_URL")
	project  = osenv.MustVar("PROJECT")
	srvName  = osenv.MustVar("TYPE")

	port = osenv.VarWithDefault("ICDEMO_PORT", "8091")

	csp = os.Getenv("CSP")
)

const (
	htmlFile        = "pages/icdemo/test.html"
	staticDirectory = "assets/serve/"
)

func main() {
	flag.Parse()

	serviceinfo.Project = project
	serviceinfo.Type = "icdemo"
	serviceinfo.Name = srvName

	tmpl, err := httputils.TemplateFromFiles(htmlFile)
	if err != nil {
		glog.Exitf("TemplateFromFiles(%v) failed: %v", htmlFile, err)
	}

	sb := &strings.Builder{}
	args := &pageArgs{
		HydraURL: hydraURL,
		IcURL:    icURL,
	}
	if err := tmpl.Execute(sb, args); err != nil {
		glog.Exitf("template.Execute() failed: %v", err)
	}

	http.HandleFunc("/test", httputils.NewPageHandler(sb.String(), httputils.CSPFromString(csp)))
	http.HandleFunc("/liveness_check", httputils.LivenessCheckHandler)

	http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(
		http.Dir(srcutil.Path(staticDirectory)))))

	glog.Infof("IC Demo listening on port %s", port)
	glog.Exit(http.ListenAndServe(":"+port, nil))
}

type pageArgs struct {
	HydraURL string
	IcURL    string
}