ocm-server

main.go

package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"log"
	"net/http"
	"os/exec"
	"strings"
)

func tokenHandler(w http.ResponseWriter, r *http.Request) {
	if r.URL.Path != "/token" {
		http.Error(w, "404 not found.", http.StatusNotFound)
		return
	}

	if r.Method != "GET" {
		http.Error(w, "Method is not supported.", http.StatusNotFound)
		return
	}

	// Create the connection, and remember to close it:
	_, err := exec.LookPath("ocm")
	if err != nil {
		log.Fatal("installing ocm is in your future")
	}

	cmd := exec.Command("ocm", "token")
	cmd.Stdin = strings.NewReader("some input")
	var out bytes.Buffer
	cmd.Stdout = &out
	err = cmd.Run()
	if err != nil {
		log.Fatal(err)
	}
	token := strings.TrimSuffix(out.String(), "\n")
	data := map[string]string{
		"token": token,
	}

	w.Header().Set("Content-Type", "application/json")
	w.WriteHeader(http.StatusCreated)

	json.NewEncoder(w).Encode(data)
}

func main() {
	http.HandleFunc("/token", tokenHandler) // Update this line of code

	fmt.Printf("Starting server at port 8061\n")
	if err := http.ListenAndServe(":8061", nil); err != nil {
		log.Fatal(err)
	}
}