Initial commit

Author: RedSkiesReaperr

.gitignore

@@ -0,0 +1,63 @@
+# Created by https://www.toptal.com/developers/gitignore/api/go,macos
+# Edit at https://www.toptal.com/developers/gitignore?templates=go,macos
+
+### Go ###
+# If you prefer the allow list template instead of the deny list, see community template:
+# https://github.com/github/gitignore/blob/main/community/Golang/Go.AllowList.gitignore
+#
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/
+
+# Go workspace file
+go.work
+
+### macOS ###
+# General
+.DS_Store
+.AppleDouble
+.LSOverride
+
+# Icon must end with two \r
+Icon
+
+
+# Thumbnails
+._*
+
+# Files that might appear in the root of a volume
+.DocumentRevisions-V100
+.fseventsd
+.Spotlight-V100
+.TemporaryItems
+.Trashes
+.VolumeIcon.icns
+.com.apple.timemachine.donotpresent
+
+# Directories potentially created on remote AFP share
+.AppleDB
+.AppleDesktop
+Network Trash Folder
+Temporary Items
+.apdisk
+
+### macOS Patch ###
+# iCloud generated files
+*.icloud
+
+# End of https://www.toptal.com/developers/gitignore/api/go,macos
+
+bin/*
+.env

Dockerfile

@@ -0,0 +1,17 @@
+FROM golang:1.21.1-alpine as build
+
+WORKDIR /usr/src/app
+COPY go.mod go.sum ./
+RUN go mod download && go mod verify
+
+COPY . .
+RUN go build -v -o /usr/local/bin/updater ./cmd/updater/main.go
+RUN go build -v -o /usr/local/bin/watcher ./cmd/watcher/main.go
+
+# Release stage
+FROM alpine:3.18
+
+WORKDIR /bin
+COPY --from=build /usr/local/bin/updater /bin/updater
+COPY --from=build /usr/local/bin/watcher /bin/watcher
+COPY .env .

README.md

@@ -0,0 +1,16 @@
+# Notion IGDB autocomplete
+
+![demo](https://github.com/RedSkiesReaperr/notion-igdb-autocomplete/assets/64477486/02de6e81-974f-4ed1-948a-e261cbd29eba)
+
+## How does it works
+Watch the database you set up. Waits for a page with a title that matchs the following pattern: `{{GAME_TITLE}}`. Then it asks for IGDB informations and it fills the Notion page.
+
+## Requirements
+- [Docker](https://www.docker.com/products/docker-desktop/)
+
+## Setup
+1. Follow the [getting started](https://developers.notion.com/docs/create-a-notion-integration#create-your-integration-in-notion) to get your `NOTION_API_SECRET` & `NOTION_PAGE_ID`.
+2. Follow the [getting started](https://api-docs.igdb.com/#getting-started) IGDB API to get your `IGDB_CLIENT_ID` & `IGDB_SECRET`
+3. Put these in your `.env` file
+4. Complete `.env` following the `.env.example` file
+5. Run `docker-compose up` command

cmd/updater/main.go

@@ -0,0 +1,177 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"net/http"
+	"notion-igdb-autocomplete/config"
+	"notion-igdb-autocomplete/igdb"
+	"sort"
+	"time"
+
+	"github.com/agnivade/levenshtein"
+	"github.com/gin-gonic/gin"
+	"github.com/jomei/notionapi"
+)
+
+type body struct {
+	PageID string `json:"page_id,required"`
+	Search string `json:"search,required"`
+}
+
+func main() {
+	config, err := config.Load()
+	if err != nil {
+		log.Fatalf("Unable to load config: %s\n", err)
+	}
+	log.Println("Successfully loaded config!")
+
+	igdbClient, err := igdb.NewClient(config.IGDBClientID, config.IGDBSecret)
+	if err != nil {
+		log.Fatalf("Unable to create IGDB client: %s\n", err)
+	}
+	log.Println("Successfully created IGDB client!")
+
+	notionClient := notionapi.NewClient(notionapi.Token(config.NotionAPISecret))
+	log.Println("Successfully created Notion client!")
+
+	server := gin.Default()
+	server.PUT("/", func(ctx *gin.Context) {
+		var payload body
+
+		err := ctx.ShouldBindJSON(&payload)
+		if err != nil {
+			ctx.JSON(http.StatusBadRequest, gin.H{"message": err.Error()})
+			return
+		}
+
+		game, err := searchIgdbGame(payload.Search, &igdbClient)
+		if err != nil {
+			ctx.JSON(http.StatusNotFound, gin.H{"message": err.Error()})
+			return
+		}
+
+		updatedPage, err := updateNotionPage(payload.PageID, game, notionClient)
+		if err != nil {
+			ctx.JSON(http.StatusUnprocessableEntity, gin.H{"message": err.Error()})
+			return
+		}
+
+		ctx.JSON(http.StatusOK, gin.H{"data": fmt.Sprintf("Updated page %s with game %s infos", updatedPage.ID, game.Name)})
+	})
+
+	err = server.Run(fmt.Sprintf("0.0.0.0:%d", config.UpdaterPort))
+	if err != nil {
+		log.Fatalf("Unable to start server: %s\n", err)
+	}
+}
+
+func searchIgdbGame(gameName string, client *igdb.Client) (*igdb.Game, error) {
+	query := igdb.NewSearchQuery(gameName, []string{"name", "platforms.name", "first_release_date", "franchises.name", "genres.name", "cover.image_id"})
+	results, err := client.SearchGame(query)
+	if err != nil {
+		return nil, err
+	}
+
+	if len(results) <= 0 {
+		return nil, fmt.Errorf("cannot find game '%s'", gameName)
+	}
+
+	return findBestGame(gameName, results), nil
+}
+
+func updateNotionPage(pageID string, game *igdb.Game, notionClient *notionapi.Client) (*notionapi.Page, error) {
+	releaseDate := notionapi.Date(time.Unix(game.ReleaseDate, 0))
+	platforms := game.NotionPlatforms()
+	genres := game.NotionGenres()
+	franchises := game.NotionFranchises()
+
+	updateReq := notionapi.PageUpdateRequest{
+		Cover: &notionapi.Image{
+			Type: "external",
+			External: &notionapi.FileObject{
+				URL: game.CoverURL(),
+			},
+		},
+		Properties: notionapi.Properties{
+			"Title": notionapi.TitleProperty{
+				Type: notionapi.PropertyTypeTitle,
+				Title: []notionapi.RichText{
+					{Text: &notionapi.Text{Content: game.Name}},
+				},
+			},
+			"Release date": notionapi.DateProperty{
+				Type: notionapi.PropertyTypeDate,
+
+				Date: &notionapi.DateObject{
+					Start: &releaseDate,
+				},
+			},
+		},
+	}
+
+	if len(platforms) > 0 {
+		updateReq.Properties["Platforms"] = notionapi.MultiSelectProperty{
+			Type:        notionapi.PropertyTypeMultiSelect,
+			MultiSelect: game.NotionPlatforms(),
+		}
+	}
+
+	if len(franchises) > 0 {
+		updateReq.Properties["Franchises"] = notionapi.MultiSelectProperty{
+			Type:        notionapi.PropertyTypeMultiSelect,
+			MultiSelect: franchises,
+		}
+	}
+
+	if len(genres) > 0 {
+		updateReq.Properties["Genres"] = notionapi.MultiSelectProperty{
+			Type:        notionapi.PropertyTypeMultiSelect,
+			MultiSelect: game.NotionGenres(),
+		}
+	}
+
+	page, err := notionClient.Page.Update(context.Background(), notionapi.PageID(pageID), &updateReq)
+	if err != nil {
+		return nil, err
+	}
+
+	return page, nil
+}
+
+type ComparedGames []ComparedGame
+type ComparedGame struct {
+	Game  igdb.Game
+	Index int
+}
+
+// Implements interface sort.Interface
+func (cg ComparedGames) Len() int {
+	return len(cg)
+}
+
+// Implements interface sort.Interface
+func (cg ComparedGames) Swap(i, j int) {
+	cg[i], cg[j] = cg[j], cg[i]
+}
+
+// Implements interface sort.Interface
+func (cg ComparedGames) Less(i, j int) bool {
+	return cg[i].Index < cg[j].Index
+}
+
+func findBestGame(search string, games igdb.Games) *igdb.Game {
+	var comparisons ComparedGames
+
+	for _, game := range games {
+		comparisons = append(comparisons, ComparedGame{
+			Game:  game,
+			Index: levenshtein.ComputeDistance(search, game.Name),
+		})
+	}
+
+	sort.Sort(comparisons)
+
+	return &comparisons[0].Game
+}

cmd/watcher/main.go

@@ -0,0 +1,110 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"notion-igdb-autocomplete/config"
+	"strings"
+	"time"
+
+	"github.com/jomei/notionapi"
+)
+
+func main() {
+	config, err := config.Load()
+	if err != nil {
+		log.Fatalf("Unable to load config: %s\n", err)
+	} else {
+		log.Println("Successfully loaded config!")
+	}
+
+	titleCleaner := strings.NewReplacer("{{", "", "}}", "")
+
+	log.Println("Looking for pages to update...")
+	for range time.Tick(time.Duration(config.WatcherTickDelay)) {
+		pages, err := fetchPages(config.NotionAPISecret, config.NotionPageID)
+		if err != nil {
+			log.Fatalf("Unable to fetch pages: %s\n", err)
+		}
+
+		for _, obj := range pages {
+			id := obj.ID
+			title := obj.Properties["Title"].(*notionapi.TitleProperty).Title[0].Text.Content
+			cleanTitle := titleCleaner.Replace(title)
+
+			err = callUpdater(config.UpdaterURL(), updaterBody{PageID: id.String(), Search: cleanTitle})
+			if err != nil {
+				log.Printf("Unable to update page '%s': %s\n", id, err)
+			} else {
+				log.Printf("page '%s' successfully updated!", id)
+			}
+		}
+	}
+}
+
+func fetchPages(apiSecret string, databaseID string) ([]notionapi.Page, error) {
+	notion := notionapi.NewClient(notionapi.Token(apiSecret))
+	query := &notionapi.DatabaseQueryRequest{
+		Filter: &notionapi.AndCompoundFilter{
+			notionapi.PropertyFilter{
+				Property: "Title",
+				RichText: &notionapi.TextFilterCondition{
+					StartsWith: "{{",
+				},
+			},
+			notionapi.PropertyFilter{
+				Property: "Title",
+				RichText: &notionapi.TextFilterCondition{
+					EndsWith: "}}",
+				},
+			},
+		},
+	}
+
+	result, err := notion.Database.Query(context.Background(), notionapi.DatabaseID(databaseID), query)
+	if err != nil {
+		return nil, err
+	}
+
+	return result.Results, nil
+}
+
+type updaterBody struct {
+	PageID string `json:"page_id,required"`
+	Search string `json:"search,required"`
+}
+
+func callUpdater(updaterURL string, payload updaterBody) error {
+	reqBody, err := json.Marshal(payload)
+	if err != nil {
+		return err
+	}
+
+	httpClient := &http.Client{}
+	req, err := http.NewRequest("PUT", updaterURL, strings.NewReader(string(reqBody)))
+	if err != nil {
+		return err
+	}
+
+	log.Printf("Requesting update: %s\n", reqBody)
+	resp, err := httpClient.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return err
+	}
+
+	if resp.StatusCode != 200 {
+		return fmt.Errorf("%s: %s", resp.Status, body)
+	}
+
+	return nil
+}