main.go

package main

import (
	"fmt"
	"net/http"
	"os"

	"github.com/bruvduroiu/pinecone-mock/pinecone"
)

func main() {
	router := http.NewServeMux()

	host := os.Getenv("PINECONE_MOCK_HOST")
	if host == "" {
		host = "localhost"
	}
	port := os.Getenv("PINECONE_MOCK_PORT")
	if port == "" {
		port = "8080"
	}

	handler := pinecone.Handler{
		Host: host,
		Port: port,
	}

	router.HandleFunc("POST /indexes", handler.CreateIndex)
	router.HandleFunc("GET /indexes", handler.ListIndex)
	router.HandleFunc("GET /indexes/{index_name}", handler.GetIndexByName)
	router.HandleFunc("POST /describe_index_stats", handler.DescribeIndexStats)
	router.HandleFunc("POST /vectors/upsert", handler.UpsertVectors)
	router.HandleFunc("POST /query", handler.QueryVectors)
	router.HandleFunc("GET /vectors/fetch", handler.FetchVectors)
	router.HandleFunc("POST /vectors/update", handler.UpdateVector)
	router.HandleFunc("POST /vectors/delete", handler.DeleteVector)
	router.HandleFunc("GET /vectors/list", handler.ListVectorIDs)

	addr := fmt.Sprintf("%s:%s", host, port)
	server := http.Server{
		Addr:    addr,
		Handler: router,
	}

	fmt.Println("Server listening on ", addr)
	server.ListenAndServe()
}