package main

import (
	"fmt"
	"log"
	"net/http"
	"strings"
	"context"
	_ "strconv"
	"encoding/json"
	"github.com/jackc/pgx/v4"
)

func search(w http.ResponseWriter, r *http.Request) {
	searchkeys, exists := r.URL.Query()["s"]

	if strings.Contains(r.Header.Get("Accept"), "application/json") {
		log.Print("Treat as an Json")
	} else {
		log.Println("Treat as HTML")
	}


	var err error
	var rows pgx.Rows
	var searchKey string
	if exists {
		searchKey = searchkeys[0]
	}

	if exists && searchKey != "" {
		rows, err = pool.Query(context.Background(), "SELECT activities.id, activities.document, actors.document FROM activities as activities INNER JOIN actors as actors ON activities.document->>'actor' = actors.document->>'id' WHERE activities.normalized_tsvector @@ plainto_tsquery($1) ORDER BY activities.identifiedat DESC LIMIT 20", searchKey)
	} else {
		rows, err = pool.Query(context.Background(), "SELECT activities.id, activities.document, actors.document FROM activities as activities INNER JOIN actors as actors ON activities.document->>'actor' = actors.document->>'id' ORDER BY activities.identifiedat DESC LIMIT 20")
	}

	if err != nil {
		panic(err)
	}
	defer rows.Close()


	var activitiesJson []map[string]json.RawMessage
	for rows.Next() {
		var id int
		var activityRaw string
		var actorRaw string
		var activityJson map[string]json.RawMessage

		err = rows.Scan(&id, &activityRaw, &actorRaw)
		if err != nil {
			panic(err)
		}

		err := json.Unmarshal([]byte(activityRaw), &activityJson)
		if err != nil {
			fmt.Println(err)
		}

		activityJson["actor"] = json.RawMessage(actorRaw)
		activitiesJson = append(activitiesJson, activityJson)
	}

	data, err := json.Marshal(activitiesJson)
	if err != nil {
		log.Fatal("error marshaling combined activity: %v", err)
	}
	fmt.Fprintf(w, "%s", data)
}

func main() {
	getSettings()
	pool = getDbPool()

	http.HandleFunc("/search", search)
	log.Print("Starting HTTP inbox on port 6432")
	log.Fatal(http.ListenAndServe("0.0.0.0:6432", nil))
}