aboutsummaryrefslogtreecommitdiffstats
path: root/backend/internal/router/router.go
blob: 3a4d41329dd93033223a7b5031ba9736188b44af (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package router

import (
	"finance/backend/internal/api/handlers"
	"finance/backend/internal/config"
	"finance/backend/internal/database"
	"net/http"

	"github.com/gin-gonic/gin"
)

// SetupRouter configures the API routes
func SetupRouter(cfg *config.Config) *gin.Engine {
	r := gin.Default()

	// Enable CORS
	r.Use(func(c *gin.Context) {
		c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
		c.Writer.Header().Set("Access-Control-Allow-Credentials", "true")
		c.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization, accept, origin, Cache-Control, X-Requested-With")
		c.Writer.Header().Set("Access-Control-Allow-Methods", "POST, OPTIONS, GET, PUT, DELETE")

		if c.Request.Method == "OPTIONS" {
			c.AbortWithStatus(204)
			return
		}

		c.Next()
	})

	// Public utility endpoints
	r.GET("/ping", func(c *gin.Context) {
		c.JSON(http.StatusOK, gin.H{
			"message": "pong",
		})
	})

	// Add database status endpoint
	r.GET("/db-status", func(c *gin.Context) {
		// Try to get a connection from the pool
		sqlDB, err := database.DB.DB()
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{
				"status":  "error",
				"message": "Failed to get database connection",
				"error":   err.Error(),
			})
			return
		}

		// Check if database is reachable
		err = sqlDB.Ping()
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{
				"status":  "error",
				"message": "Database is not reachable",
				"error":   err.Error(),
			})
			return
		}

		c.JSON(http.StatusOK, gin.H{
			"status":  "success",
			"message": "Database connection is healthy",
		})
	})

	// Initialize handlers
	authHandler := handlers.NewAuthHandler(cfg)
	goalHandler := handlers.NewGoalHandler()
	userHandler := handlers.NewUserHandler()
	accountHandler := handlers.NewAccountHandler()
	transactionHandler := handlers.NewTransactionHandler()
	loanHandler := handlers.NewLoanHandler()

	// API v1 routes
	v1 := r.Group("/api/v1")
	{
		// Auth routes (public)
		v1.POST("/auth/signup", authHandler.Register)
		v1.POST("/auth/login", authHandler.Login)

		// Protected routes
		protected := v1.Group("")
		protected.Use(authHandler.JWTAuth)
		{
			// User routes
			protected.GET("/users/me", userHandler.GetCurrentUser)
			protected.PUT("/users/me", userHandler.UpdateCurrentUser)

			// Account routes
			protected.GET("/accounts", accountHandler.GetAccounts)
			protected.GET("/accounts/:id", accountHandler.GetAccountByID)
			protected.POST("/accounts", accountHandler.CreateAccount)
			protected.PUT("/accounts/:id", accountHandler.UpdateAccount)
			protected.DELETE("/accounts/:id", accountHandler.DeleteAccount)

			// Transaction routes
			protected.GET("/transactions", transactionHandler.GetTransactions)
			protected.GET("/transactions/:id", transactionHandler.GetTransactionByID)
			protected.POST("/transactions", transactionHandler.CreateTransaction)
			protected.PUT("/transactions/:id", transactionHandler.UpdateTransaction)
			protected.DELETE("/transactions/:id", transactionHandler.DeleteTransaction)

			// Goal routes
			protected.GET("/goals", goalHandler.GetGoals)
			protected.GET("/goals/:id", goalHandler.GetGoal)
			protected.POST("/goals", goalHandler.CreateGoal)
			protected.PUT("/goals/:id", goalHandler.UpdateGoal)
			protected.DELETE("/goals/:id", goalHandler.DeleteGoal)
			protected.PATCH("/goals/:id/progress", goalHandler.UpdateGoalProgress)

			// Loan routes
			protected.GET("/loans", loanHandler.GetLoans)
			protected.GET("/loans/:id", loanHandler.GetLoanByID)
			protected.POST("/loans", loanHandler.CreateLoan)
			protected.PUT("/loans/:id", loanHandler.UpdateLoan)
			protected.DELETE("/loans/:id", loanHandler.DeleteLoan)
		}
	}

	return r
}