feat(backend): implement

Signed-off-by: Aryan Goyal <[email protected]>

Author: ary82

.env.example

@@ -0,0 +1,3 @@
+MODE=dev
+PORT=8000
+GEMINI_KEY=

.gitignore

@@ -1,5 +1,6 @@
 main
+tmp
 .env
-extension/chromium/socratic-code/node_modules
-extension/chromium/socratic-code/.svelte-kit
-extension/chromium/socratic-code/build
+chromium/socratic-code/node_modules
+chromium/socratic-code/.svelte-kit
+chromium/socratic-code/build

cmd/app/main.go

@@ -1,7 +1,33 @@
 package main
 
-import "fmt"
+import (
+	"log"
+	"os"
+
+	"github.com/ary82/sleetcode/internal/infra"
+	"github.com/joho/godotenv"
+)
 
 func main() {
-	fmt.Println("hello")
+	mode := os.Getenv("MODE")
+	if mode != "prod" {
+		err := godotenv.Load("./.env")
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+
+	port := os.Getenv("PORT")
+	if port == "" {
+		log.Fatal("port envvar empty")
+	}
+	geminiKey := os.Getenv("GEMINI_KEY")
+	if geminiKey == "" {
+		log.Fatal("gemini_key envvar empty")
+	}
+
+	err := infra.Run(port, geminiKey)
+	if err != nil {
+		log.Fatal(err)
+	}
 }

go.mod

@@ -1,3 +1,54 @@
 module github.com/ary82/sleetcode
 
 go 1.23.1
+
+require (
+	github.com/gofiber/fiber/v2 v2.52.5
+	github.com/google/generative-ai-go v0.18.0
+	github.com/joho/godotenv v1.5.1
+	google.golang.org/api v0.186.0
+)
+
+require (
+	cloud.google.com/go v0.115.0 // indirect
+	cloud.google.com/go/ai v0.8.0 // indirect
+	cloud.google.com/go/auth v0.6.0 // indirect
+	cloud.google.com/go/auth/oauth2adapt v0.2.2 // indirect
+	cloud.google.com/go/compute/metadata v0.3.0 // indirect
+	cloud.google.com/go/longrunning v0.5.7 // indirect
+	github.com/andybalholm/brotli v1.0.5 // indirect
+	github.com/felixge/httpsnoop v1.0.4 // indirect
+	github.com/go-logr/logr v1.4.1 // indirect
+	github.com/go-logr/stdr v1.2.2 // indirect
+	github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
+	github.com/golang/protobuf v1.5.4 // indirect
+	github.com/google/s2a-go v0.1.7 // indirect
+	github.com/google/uuid v1.6.0 // indirect
+	github.com/googleapis/enterprise-certificate-proxy v0.3.2 // indirect
+	github.com/googleapis/gax-go/v2 v2.12.5 // indirect
+	github.com/klauspost/compress v1.17.0 // indirect
+	github.com/mattn/go-colorable v0.1.13 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/mattn/go-runewidth v0.0.15 // indirect
+	github.com/rivo/uniseg v0.2.0 // indirect
+	github.com/valyala/bytebufferpool v1.0.0 // indirect
+	github.com/valyala/fasthttp v1.51.0 // indirect
+	github.com/valyala/tcplisten v1.0.0 // indirect
+	go.opencensus.io v0.24.0 // indirect
+	go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.51.0 // indirect
+	go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.51.0 // indirect
+	go.opentelemetry.io/otel v1.26.0 // indirect
+	go.opentelemetry.io/otel/metric v1.26.0 // indirect
+	go.opentelemetry.io/otel/trace v1.26.0 // indirect
+	golang.org/x/crypto v0.24.0 // indirect
+	golang.org/x/net v0.26.0 // indirect
+	golang.org/x/oauth2 v0.21.0 // indirect
+	golang.org/x/sync v0.7.0 // indirect
+	golang.org/x/sys v0.21.0 // indirect
+	golang.org/x/text v0.16.0 // indirect
+	golang.org/x/time v0.5.0 // indirect
+	google.golang.org/genproto/googleapis/api v0.0.0-20240617180043-68d350f18fd4 // indirect
+	google.golang.org/genproto/googleapis/rpc v0.0.0-20240617180043-68d350f18fd4 // indirect
+	google.golang.org/grpc v1.64.1 // indirect
+	google.golang.org/protobuf v1.34.2 // indirect
+)

go.sum

@@ -0,0 +1,3 @@
+package infra
+
+const PROMPT string = `You are a teacher helping a student in solving leetcode. The problem is "%v". The solution is in "%v" programming language. The student is attempting the solution and has currently a code of %v lines. The student is attempting the follwing solution:"%v". Help the student towards the solution using Socratic teaching questions. If you feel like their solution can be optimized further, give some questions according to the Socratic Teaching method philosophy and at what line of the solution they should be asked. The questions should be insightful and encourage the student towards getting the answer on their own. There should be a maximum of one question per line of code, and the questions should not excede the number of lines.`

internal/infra/constants.go

@@ -0,0 +1,32 @@
+package infra
+
+import "fmt"
+
+type TextGenerationRequest struct {
+	Question string `json:"question"`
+	Lines    int    `json:"lines"`
+	Lang     string `json:"lang"`
+	Code     string `json:"code"`
+}
+
+type TextGenerationResponse struct {
+	Line int    `json:"line"`
+	Q    string `json:"q"`
+}
+
+func (req *TextGenerationRequest) Validate() error {
+	if req.Question == "" {
+		return fmt.Errorf("no question given")
+	}
+	if req.Lines == 0 {
+		return fmt.Errorf("0 lines solution not allowed")
+	}
+	if req.Lang == "" {
+		return fmt.Errorf("lang cannot be empty")
+	}
+	if req.Code == "" {
+		return fmt.Errorf("no code input given")
+	}
+
+	return nil
+}

internal/infra/domain.go

@@ -0,0 +1,39 @@
+package infra
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/gofiber/fiber/v2"
+	"github.com/google/generative-ai-go/genai"
+	"google.golang.org/api/option"
+)
+
+func Run(port string, geminiKey string) error {
+	app := fiber.New(fiber.Config{
+		AppName:      "Socratic code",
+		ServerHeader: "Fiber",
+	})
+
+	gClient, err := genai.NewClient(context.Background(), option.WithAPIKey(geminiKey))
+	if err != nil {
+		return err
+	}
+	textModel := gClient.GenerativeModel("gemini-1.5-flash")
+	textModel.ResponseMIMEType = "application/json"
+	textModel.ResponseSchema = &genai.Schema{
+		Type: genai.TypeArray,
+		Items: &genai.Schema{
+			Type: genai.TypeObject,
+			Properties: map[string]*genai.Schema{
+				"line": {Type: genai.TypeInteger},
+				"q":    {Type: genai.TypeString},
+			},
+		},
+	}
+
+	server := NewServer(app, textModel)
+
+	err = server.App.Listen(fmt.Sprintf(":%v", port))
+	return err
+}

internal/infra/run.go

@@ -0,0 +1,72 @@
+package infra
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+
+	"github.com/gofiber/fiber/v2"
+	"github.com/gofiber/fiber/v2/middleware/compress"
+	"github.com/gofiber/fiber/v2/middleware/logger"
+	"github.com/google/generative-ai-go/genai"
+)
+
+type Server struct {
+	App       *fiber.App
+	TextModel *genai.GenerativeModel
+}
+
+func NewServer(app *fiber.App, textModel *genai.GenerativeModel) *Server {
+	s := &Server{
+		App:       app,
+		TextModel: textModel,
+	}
+	s.registerRoutes()
+	return s
+}
+
+func (s *Server) registerRoutes() {
+	s.App.Use(logger.New())
+	s.App.Use(compress.New())
+
+	s.App.Get("/", s.getSocraticQuestions)
+}
+
+func (s *Server) getSocraticQuestions(ctx *fiber.Ctx) error {
+	req := new(TextGenerationRequest)
+
+	err := ctx.BodyParser(req)
+	if err != nil {
+		return ctx.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
+			"err": err.Error(),
+		})
+	}
+	err = req.Validate()
+	if err != nil {
+		return ctx.Status(fiber.StatusBadRequest).JSON(fiber.Map{
+			"err": err.Error(),
+		})
+	}
+
+	dynamicPrompt := fmt.Sprintf(PROMPT, req.Question, req.Lang, req.Lines, req.Code)
+	fmt.Println(dynamicPrompt)
+	res, err := s.TextModel.GenerateContent(
+		context.Background(),
+		genai.Text(dynamicPrompt),
+	)
+	if err != nil {
+		return ctx.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
+			"error": err.Error(),
+		})
+	}
+
+	resJson := []TextGenerationResponse{}
+	err = json.Unmarshal([]byte(res.Candidates[0].Content.Parts[0].(genai.Text)), &resJson)
+	if err != nil {
+		return ctx.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
+			"error": err.Error(),
+		})
+	}
+
+	return ctx.Status(fiber.StatusOK).JSON(resJson)
+}