emdb/client/emdb.go

package client

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"net/http"

	"ewintr.nl/emdb/cmd/api-service/moviestore"
)

type EMDB struct {
	baseURL string
	apiKey  string
	c       *http.Client
}

func NewEMDB(baseURL string, apiKey string) *EMDB {
	return &EMDB{
		baseURL: baseURL,
		apiKey:  apiKey,
		c:       &http.Client{},
	}
}

func (e *EMDB) GetMovies() ([]moviestore.Movie, error) {
	url := fmt.Sprintf("%s/movie", e.baseURL)
	req, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		return nil, err
	}
	req.Header.Add("Authorization", e.apiKey)

	resp, err := e.c.Do(req)
	if err != nil {
		return nil, err
	}

	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
	}

	body, err := io.ReadAll(resp.Body)
	defer resp.Body.Close()

	var movies []moviestore.Movie
	if err := json.Unmarshal(body, &movies); err != nil {
		return nil, err
	}

	return movies, nil
}

func (e *EMDB) GetMovie(id string) (moviestore.Movie, error) {
	url := fmt.Sprintf("%s/movie/%s", e.baseURL, id)
	req, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		return moviestore.Movie{}, err
	}
	req.Header.Add("Authorization", e.apiKey)

	resp, err := e.c.Do(req)
	if err != nil {
		return moviestore.Movie{}, err
	}

	if resp.StatusCode != http.StatusOK {
		return moviestore.Movie{}, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
	}

	body, err := io.ReadAll(resp.Body)
	defer resp.Body.Close()

	var movie moviestore.Movie
	if err := json.Unmarshal(body, &movie); err != nil {
		return moviestore.Movie{}, err
	}

	return movie, nil
}

func (e *EMDB) CreateMovie(m moviestore.Movie) (moviestore.Movie, error) {
	body, err := json.Marshal(m)
	if err != nil {
		return moviestore.Movie{}, err
	}

	url := fmt.Sprintf("%s/movie", e.baseURL)
	req, err := http.NewRequest(http.MethodPost, url, bytes.NewReader(body))
	if err != nil {
		return moviestore.Movie{}, err
	}
	req.Header.Add("Authorization", e.apiKey)

	resp, err := e.c.Do(req)
	if err != nil {
		return moviestore.Movie{}, err
	}

	if resp.StatusCode != http.StatusOK {
		return moviestore.Movie{}, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
	}

	newBody, err := io.ReadAll(resp.Body)
	if err != nil {
		return moviestore.Movie{}, err
	}
	defer resp.Body.Close()

	var newMovie moviestore.Movie
	if err := json.Unmarshal(newBody, &newMovie); err != nil {
		return moviestore.Movie{}, err
	}

	return newMovie, nil
}

func (e *EMDB) UpdateMovie(m moviestore.Movie) (moviestore.Movie, error) {
	body, err := json.Marshal(m)
	if err != nil {
		return moviestore.Movie{}, err
	}

	url := fmt.Sprintf("%s/movie/%s", e.baseURL, m.ID)
	req, err := http.NewRequest(http.MethodPut, url, bytes.NewReader(body))
	if err != nil {
		return moviestore.Movie{}, err
	}
	req.Header.Add("Authorization", e.apiKey)

	resp, err := e.c.Do(req)
	if err != nil {
		return moviestore.Movie{}, err
	}

	if resp.StatusCode != http.StatusOK {
		return moviestore.Movie{}, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
	}

	newBody, err := io.ReadAll(resp.Body)
	if err != nil {
		return moviestore.Movie{}, err
	}
	defer resp.Body.Close()

	var newMovie moviestore.Movie
	if err := json.Unmarshal(newBody, &newMovie); err != nil {
		return moviestore.Movie{}, err
	}

	return newMovie, nil
}

func (e *EMDB) GetReviews(movieID string) ([]moviestore.Review, error) {
	url := fmt.Sprintf("%s/movie/%s/review", e.baseURL, movieID)
	req, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		return nil, err
	}
	req.Header.Add("Authorization", e.apiKey)

	resp, err := e.c.Do(req)
	if err != nil {
		return nil, err
	}

	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
	}

	body, err := io.ReadAll(resp.Body)
	defer resp.Body.Close()

	var reviews []moviestore.Review
	if err := json.Unmarshal(body, &reviews); err != nil {
		return nil, err
	}

	return reviews, nil
}

func (e *EMDB) GetReview(id string) (moviestore.Review, error) {
	url := fmt.Sprintf("%s/review/%s", e.baseURL, id)
	req, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		return moviestore.Review{}, err
	}
	req.Header.Add("Authorization", e.apiKey)

	resp, err := e.c.Do(req)
	if err != nil {
		return moviestore.Review{}, err
	}

	if resp.StatusCode != http.StatusOK {
		return moviestore.Review{}, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
	}

	body, err := io.ReadAll(resp.Body)
	defer resp.Body.Close()

	var review moviestore.Review
	if err := json.Unmarshal(body, &review); err != nil {
		return moviestore.Review{}, err
	}

	return review, nil
}

func (e *EMDB) GetNextUnratedReview() (moviestore.Review, error) {
	url := fmt.Sprintf("%s/review/unrated/next", e.baseURL)
	req, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		return moviestore.Review{}, err
	}
	req.Header.Add("Authorization", e.apiKey)

	resp, err := e.c.Do(req)
	if err != nil {
		return moviestore.Review{}, err
	}

	if resp.StatusCode != http.StatusOK {
		return moviestore.Review{}, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
	}

	body, err := io.ReadAll(resp.Body)
	defer resp.Body.Close()

	var review moviestore.Review
	if err := json.Unmarshal(body, &review); err != nil {
		return moviestore.Review{}, err
	}

	return review, nil
}

func (e *EMDB) UpdateReview(review moviestore.Review) error {
	body, err := json.Marshal(review)
	if err != nil {
		return err
	}

	url := fmt.Sprintf("%s/review/%s", e.baseURL, review.ID)
	req, err := http.NewRequest(http.MethodPut, url, bytes.NewReader(body))
	if err != nil {
		return err
	}
	req.Header.Add("Authorization", e.apiKey)

	resp, err := e.c.Do(req)
	if err != nil {
		return err
	}

	if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusCreated {
		return fmt.Errorf("unexpected status code: %d", resp.StatusCode)
	}

	return nil
}

func (e *EMDB) CreateJob(movieID, action string) error {
	j := struct {
		MovieID string
		Action  string
	}{
		MovieID: movieID,
		Action:  action,
	}

	body, err := json.Marshal(j)
	if err != nil {
		return err
	}

	url := fmt.Sprintf("%s/job", e.baseURL)
	req, err := http.NewRequest(http.MethodPost, url, bytes.NewReader(body))
	if err != nil {
		return err
	}
	req.Header.Add("Authorization", e.apiKey)

	resp, err := e.c.Do(req)
	if err != nil {
		return err
	}

	if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusCreated {
		return fmt.Errorf("unexpected status code: %d", resp.StatusCode)
	}

	return nil
}

func (e *EMDB) GetNextAIJob() (moviestore.Job, error) {
	url := fmt.Sprintf("%s/job/next-ai", e.baseURL)
	req, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		return moviestore.Job{}, err
	}
	req.Header.Add("Authorization", e.apiKey)

	resp, err := e.c.Do(req)
	if err != nil {
		return moviestore.Job{}, err
	}

	if resp.StatusCode != http.StatusOK {
		return moviestore.Job{}, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
	}

	body, err := io.ReadAll(resp.Body)
	defer resp.Body.Close()

	var j moviestore.Job
	if err := json.Unmarshal(body, &j); err != nil {
		return moviestore.Job{}, err
	}

	return j, nil
}