update

pkg/dadjoke/handler.go

@@ -0,0 +1,131 @@
+package dadjoke
+
+import (
+	"encoding/json"
+	"net/http"
+
+	"gitlab.com/developerdurp/logger"
+	"gitlab.com/developerdurp/stdmodels"
+	"gorm.io/gorm"
+)
+
+type Handler struct {
+	db *gorm.DB
+}
+
+type DadJoke struct {
+	JOKE string `json:"joke"`
+}
+
+func NewHandler(db *gorm.DB) (*Handler, error) {
+	err := db.AutoMigrate(&DadJoke{})
+	if err != nil {
+		return nil, err
+	}
+	return &Handler{db: db}, nil
+}
+
+// GetDadJoke godoc
+//
+//	@Summary		Get dadjoke
+//	@Description	get a dad joke
+//	@Tags			DadJoke
+//	@Accept			json
+//	@Produce		application/json
+//	@Success		200	{object}	stdmodels.StandardMessage	"response"
+//	@failure		500	{object}	stdmodels.StandardError"error"
+//
+// @Security Authorization
+//
+//	@Router			/jokes/dadjoke [get]
+func (h *Handler) Get(w http.ResponseWriter, r *http.Request) {
+	joke, err := h.GetRandomDadJoke()
+
+	if err != nil {
+		stdmodels.FailureReponse("Failed to get Joke", w, http.StatusInternalServerError, []string{err.Error()})
+		return
+	}
+
+	message := stdmodels.StandardMessage{
+		Message: joke,
+	}
+
+	json.NewEncoder(w).Encode(message)
+}
+
+// PostDadJoke godoc
+//
+//	@Summary		Generate dadjoke
+//	@Description	create a dad joke
+//	@Tags			DadJoke
+//	@Accept			json
+//	@Produce		application/json
+//	@Param			joke	query		string			true	"Dad Joke you wish to enter into database"
+//	@Success		200		{object}	stdmodels.StandardMessage	"response"
+//	@failure		500	{object}	stdmodels.StandardError"error"
+//
+// @Security Authorization
+//
+//	@Router			/jokes/dadjoke [post]
+func (h *Handler) Post(w http.ResponseWriter, r *http.Request) {
+	contentType := r.Header.Get("Content-Type")
+	var req DadJoke
+
+	if contentType == "application/json" {
+		err := json.NewDecoder(r.Body).Decode(&req)
+		if err != nil {
+			logger.LogError("Failed to decode json file")
+			return
+		}
+	} else {
+		queryParams := r.URL.Query()
+		req.JOKE = queryParams.Get("joke")
+	}
+
+	err := h.PostDadJoke(req)
+	if err != nil {
+		stdmodels.FailureReponse("Failed to add joke", w, http.StatusInternalServerError, []string{err.Error()})
+		return
+	}
+
+	stdmodels.SuccessResponse("OK", w, http.StatusOK)
+}
+
+// DeleteDadJoke godoc
+//
+//	@Summary		Generate dadjoke
+//	@Description	create a dad joke
+//	@Tags			DadJoke
+//	@Accept			json
+//	@Produce		application/json
+//	@Param			joke	query		string			true	"Dad joke you wish to delete from the database"
+//	@Success		200		{object}	stdmodels.StandardMessage	"response"
+//	@failure		500	{object}	stdmodels.StandardError"error"
+//
+// @Security Authorization
+//
+//	@Router			/jokes/dadjoke [delete]
+func (h *Handler) Delete(w http.ResponseWriter, r *http.Request) {
+	contentType := r.Header.Get("Content-Type")
+	var req DadJoke
+
+	if contentType == "application/json" {
+		err := json.NewDecoder(r.Body).Decode(&req)
+		if err != nil {
+			w.WriteHeader(http.StatusInternalServerError)
+			w.Write([]byte(http.StatusText(http.StatusInternalServerError)))
+			return
+		}
+	} else {
+		queryParams := r.URL.Query()
+		req.JOKE = queryParams.Get("joke")
+	}
+
+	err := h.DeleteDadJoke(req)
+	if err != nil {
+		stdmodels.FailureReponse("Failed to delete joke", w, http.StatusInternalServerError, []string{err.Error()})
+		return
+	}
+
+	stdmodels.SuccessResponse("OK", w, http.StatusOK)
+}

pkg/dadjoke/service.go

@@ -0,0 +1,66 @@
+package dadjoke
+
+import (
+	"errors"
+	"math/rand"
+)
+
+func (h *Handler) GetRandomDadJoke() (string, error) {
+	jokes, err := h.getDadJokes()
+	if err != nil {
+		return "", err
+	}
+
+	randomIndex := rand.Intn(len(jokes))
+
+	randomElement := jokes[randomIndex]
+
+	return randomElement.JOKE, err
+}
+
+func (h *Handler) PostDadJoke(joke DadJoke) error {
+	jokes, err := h.getDadJokes()
+	if err != nil {
+		return err
+	}
+
+	found := false
+	for _, i := range jokes {
+		if i.JOKE == joke.JOKE {
+			found = true
+			break
+		}
+	}
+
+	if found {
+		return errors.New("Joke is already in database")
+	} else {
+		err = h.db.Create(&joke).Error
+		if err != nil {
+			return err
+		}
+		return nil
+	}
+}
+
+func (h *Handler) DeleteDadJoke(joke DadJoke) error {
+	check := &DadJoke{}
+	h.db.Where("joke = ?", joke.JOKE).First(check)
+	if check.JOKE == "" {
+		return errors.New("Joke does not exist")
+	}
+
+	err := h.db.Where("joke = ?", joke.JOKE).Delete(joke).Error
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (h *Handler) getDadJokes() ([]DadJoke, error) {
+	req := []DadJoke{}
+
+	err := h.db.Find(&req).Error
+
+	return req, err
+}

pkg/health/handler.go

@@ -0,0 +1,30 @@
+package health
+
+import (
+	"net/http"
+
+	"gitlab.com/developerdurp/stdmodels"
+)
+
+type Handler struct{}
+
+func NewHandler() (*Handler, error) {
+	return &Handler{}, nil
+}
+
+// getHealth godoc
+//
+//	@Summary		Generate Health status
+//	@Description	Get the health of the API
+//	@Tags			health
+//	@Accept			json
+//	@Produce		application/json
+//	@Success		200		{object}	stdmodels.StandardMessage	"response"
+//	@failure		500	{object}	stdmodels.StandardError"error"
+//
+// @Security Authorization
+//
+//	@Router			/health/gethealth [get]
+func (h *Handler) Get(w http.ResponseWriter, r *http.Request) {
+	stdmodels.SuccessResponse("OK", w, http.StatusOK)
+}

pkg/openai/handler.go

@@ -0,0 +1,103 @@
+package openai
+
+import (
+	"encoding/json"
+	"net/http"
+
+	"gitlab.com/developerdurp/stdmodels"
+)
+
+type Handler struct {
+	LlamaURL string
+}
+
+func NewHandler(LlamaURL string) (*Handler, error) {
+	return &Handler{LlamaURL: LlamaURL}, nil
+}
+
+type ChatRequest struct {
+	Message string `json:"message"`
+}
+
+// Response struct to unmarshal the JSON response
+type Response struct {
+	Response string `json:"response"`
+}
+
+// GeneralOpenAI godoc
+//
+//	@Summary		Gerneral ChatGPT
+//	@Description	Ask ChatGPT a general question
+//	@Tags			openai
+//	@Accept			json
+//	@Produce		application/json
+//	@Param			message	query		string			true	"Ask ChatGPT a general question"
+//	@Success		200	{object}	stdmodels.StandardMessage	"response"
+//	@failure		500	{object}	stdmodels.StandardError"error"
+//
+// @Security Authorization
+//
+//	@Router			/openai/general [get]
+func (h *Handler) GeneralOpenAI(w http.ResponseWriter, r *http.Request) {
+	contentType := r.Header.Get("Content-Type")
+	var req ChatRequest
+
+	if contentType == "application/json" {
+		err := json.NewDecoder(r.Body).Decode(&req)
+		if err != nil {
+			stdmodels.FailureReponse("Failed To decode content", w, http.StatusInternalServerError, []string{err.Error()})
+			return
+		}
+	} else {
+		queryParams := r.URL.Query()
+		req.Message = queryParams.Get("message")
+	}
+
+	result, err := h.createChatCompletion(req.Message, "mistral:instruct")
+	if err != nil {
+		stdmodels.FailureReponse("Failed to Send Message", w, http.StatusInternalServerError, []string{err.Error()})
+		return
+	}
+
+	stdmodels.SuccessResponse(result, w, http.StatusOK)
+}
+
+// TravelAgentOpenAI godoc
+//
+//	@Summary		Travel Agent ChatGPT
+//	@Description	Ask ChatGPT for suggestions as if it was a travel agent
+//	@Tags			openai
+//	@Accept			json
+//	@Produce		application/json
+//	@Param			message	query		string			true	"Ask ChatGPT for suggestions as a travel agent"
+//	@Success		200	{object}	stdmodels.StandardMessage	"response"
+//	@failure		500	{object}	stdmodels.StandardError"error"
+//
+// @Security Authorization
+//
+//	@Router			/openai/travelagent [get]
+func (h *Handler) TravelAgentOpenAI(w http.ResponseWriter, r *http.Request) {
+	contentType := r.Header.Get("Content-Type")
+	var req ChatRequest
+
+	if contentType == "application/json" {
+		err := json.NewDecoder(r.Body).Decode(&req)
+		if err != nil {
+			stdmodels.FailureReponse("Failed To decode content", w, http.StatusInternalServerError, []string{err.Error()})
+			return
+		}
+	} else {
+		queryParams := r.URL.Query()
+		req.Message = queryParams.Get("message")
+	}
+
+	req.Message = "I want you to act as a travel guide. I will give you my location and you will give me suggestions. " + req.Message
+
+	result, err := h.createChatCompletion(req.Message, "mistral:instruct")
+	if err != nil {
+		stdmodels.FailureReponse("Failed to Send Message", w, http.StatusInternalServerError, []string{err.Error()})
+		return
+	}
+
+	stdmodels.SuccessResponse(result, w, http.StatusOK)
+}

pkg/openai/service.go

@@ -0,0 +1,50 @@
+package openai
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+)
+
+func (h *Handler) createChatCompletion(message string, model string) (string, error) {
+	// Define the request body
+	requestBody := map[string]interface{}{
+		"model":  model,
+		"prompt": message,
+		"stream": false,
+	}
+
+	// Convert the request body to JSON
+	requestBodyBytes, err := json.Marshal(requestBody)
+	if err != nil {
+		return "", fmt.Errorf("error encoding request body: %v", err)
+	}
+
+	// Send a POST request to the specified URL with the request body
+	response, err := http.Post(
+		"http://"+h.LlamaURL+"/api/generate",
+		"application/json",
+		bytes.NewBuffer(requestBodyBytes),
+	)
+	if err != nil {
+		return "", fmt.Errorf("error sending POST request: %v", err)
+	}
+	defer response.Body.Close()
+
+	// Read the response body
+	responseBody, err := io.ReadAll(response.Body)
+	if err != nil {
+		return "", fmt.Errorf("error reading response body: %v", err)
+	}
+
+	// Unmarshal the JSON response
+	var resp Response
+	if err := json.Unmarshal(responseBody, &resp); err != nil {
+		return "", fmt.Errorf("error decoding response body: %v", err)
+	}
+
+	// Return the response
+	return resp.Response, nil
+}