161 | {body} 162 |
163 | ) 164 | }) 165 | FormMessage.displayName = "FormMessage" 166 | 167 | export { 168 | useFormField, 169 | Form, 170 | FormItem, 171 | FormLabel, 172 | FormControl, 173 | FormDescription, 174 | FormMessage, 175 | FormField, 176 | } 177 | -------------------------------------------------------------------------------- /client/src/components/ui/input.tsx: -------------------------------------------------------------------------------- 1 | import * as React from "react" 2 | 3 | import { cn } from "@/lib/utils" 4 | 5 | const Input = React.forwardRefCard Footer
84 |Node: %s
91 |IP: %s
92 |Metric: %s
93 |Current Value: %s
94 |Threshold: %s
95 |Timestamp: %s
96 |This alert was generated by VPS Pilot monitoring system.
98 | 99 | 100 | `, alert.NodeName, alert.NodeIp, alert.Metric, alert.CurrentValue, alert.Threshold, alert.Timestamp.Format(time.RFC1123)) 101 | 102 | m.SetBody("text/html", body) 103 | 104 | // Create dialer 105 | d := gomail.NewDialer(host, port, username, password) 106 | d.TLSConfig = &tls.Config{InsecureSkipVerify: true} 107 | 108 | // Send the email 109 | if err := d.DialAndSend(m); err != nil { 110 | return fmt.Errorf("failed to send email: %v", err) 111 | } 112 | 113 | fmt.Println("Alert sent to email:", toEmail) 114 | return nil 115 | } 116 | 117 | // SendSlackAlert sends alert notifications via Slack webhook 118 | func SendSlackAlert(webhookURL string, alert AlertMsg) error { 119 | // Format the alert message for Slack 120 | message := fmt.Sprintf( 121 | ":rotating_light: *ALERT* :rotating_light:\n*Node:* `%s`\n*IP:* `%s`\n*Metric:* *%s*\n*Current Value:* `%s`\n*Threshold:* `%s`\n*Timestamp:* %s", 122 | alert.NodeName, alert.NodeIp, alert.Metric, alert.CurrentValue, alert.Threshold, alert.Timestamp.Format(time.RFC1123), 123 | ) 124 | 125 | // Prepare the Slack payload 126 | payload := map[string]interface{}{ 127 | "text": message, 128 | "attachments": []map[string]interface{}{ 129 | { 130 | "color": "danger", 131 | "fields": []map[string]interface{}{ 132 | { 133 | "title": "Node", 134 | "value": alert.NodeName, 135 | "short": true, 136 | }, 137 | { 138 | "title": "IP Address", 139 | "value": alert.NodeIp, 140 | "short": true, 141 | }, 142 | { 143 | "title": "Metric", 144 | "value": alert.Metric, 145 | "short": true, 146 | }, 147 | { 148 | "title": "Current Value", 149 | "value": alert.CurrentValue, 150 | "short": true, 151 | }, 152 | { 153 | "title": "Threshold", 154 | "value": alert.Threshold, 155 | "short": true, 156 | }, 157 | { 158 | "title": "Timestamp", 159 | "value": alert.Timestamp.Format(time.RFC1123), 160 | "short": true, 161 | }, 162 | }, 163 | }, 164 | }, 165 | } 166 | 167 | // Marshal the payload to JSON 168 | body, err := json.Marshal(payload) 169 | if err != nil { 170 | return err 171 | } 172 | 173 | // Send the POST request to the Slack webhook 174 | resp, err := http.Post(webhookURL, "application/json", bytes.NewBuffer(body)) 175 | if err != nil { 176 | return err 177 | } 178 | defer resp.Body.Close() 179 | 180 | // Check for successful response 181 | if resp.StatusCode >= 200 && resp.StatusCode < 300 { 182 | fmt.Println("Alert sent to Slack!") 183 | } else { 184 | return fmt.Errorf("failed to send Slack alert, status code: %d", resp.StatusCode) 185 | } 186 | 187 | return nil 188 | } 189 | -------------------------------------------------------------------------------- /server/internal/tcpserver/server.go: -------------------------------------------------------------------------------- 1 | package tcpserver 2 | 3 | import ( 4 | "context" 5 | "encoding/gob" 6 | "fmt" 7 | "net" 8 | "strings" 9 | 10 | "github.com/sanda0/vps_pilot/internal/db" 11 | ) 12 | 13 | var AgentConnections map[string]net.Conn 14 | 15 | func StartTcpServer(ctx context.Context, repo *db.Repo, port string) { 16 | 17 | var statChan = make(chan Msg, 100) 18 | var monitorChan = make(chan Msg, 100) 19 | 20 | AgentConnections = make(map[string]net.Conn) 21 | listener, err := net.Listen("tcp", ":"+port) 22 | if err != nil { 23 | return 24 | } 25 | defer listener.Close() 26 | fmt.Println("TCP server Listening on port", port) 27 | 28 | go StoreSystemStats(ctx, repo, statChan) 29 | go MontiorAlerts(ctx, repo, monitorChan) 30 | 31 | for { 32 | conn, err := listener.Accept() 33 | if err != nil { 34 | return 35 | } 36 | AgentConnections[conn.RemoteAddr().String()] = conn 37 | go handleRequest(ctx, repo, conn, statChan, monitorChan) 38 | } 39 | } 40 | 41 | func handleRequest(ctx context.Context, repo *db.Repo, conn net.Conn, statChan chan Msg, monitorChan chan Msg) { 42 | defer conn.Close() 43 | fmt.Println("New connection from", conn.RemoteAddr()) 44 | 45 | decoder := gob.NewDecoder(conn) 46 | encoder := gob.NewEncoder(conn) 47 | var msg Msg 48 | for { 49 | err := decoder.Decode(&msg) 50 | if err != nil { 51 | break 52 | } 53 | if msg.Msg == "connected" { 54 | ip := strings.Split(conn.RemoteAddr().String(), ":")[0] 55 | node, err := CreateNode(ctx, repo, ip, msg.Data) 56 | if err != nil { 57 | fmt.Println("Error creating node", err) 58 | } 59 | fmt.Println("Node connected", node) 60 | err = encoder.Encode(Msg{ 61 | Msg: "sys_stat", 62 | NodeId: node.ID, 63 | }) 64 | if err != nil { 65 | fmt.Println("Error encoding message:", err) 66 | } 67 | } 68 | if msg.Msg == "sys_info" { 69 | 70 | fmt.Println("Sys info received", string(msg.Data)) 71 | } 72 | if msg.Msg == "sys_stat" { 73 | statChan <- msg 74 | monitorChan <- msg 75 | } 76 | } 77 | 78 | } 79 | -------------------------------------------------------------------------------- /server/internal/tcpserver/types.go: -------------------------------------------------------------------------------- 1 | package tcpserver 2 | 3 | import "encoding/json" 4 | 5 | type SystemInfo struct { 6 | OS string `json:"os"` // e.g. linux, windows 7 | Platform string `json:"platform"` // e.g. ubuntu, centos 8 | PlatformVersion string `json:"platform_version"` // e.g. 20.04, 8 9 | KernelVersion string `json:"kernel_version"` // e.g. 5.4.0-42-generic 10 | CPUs int `json:"cpus"` // number of CPUs 11 | TotalMemory uint64 `json:"total_memory"` // total memory in bytes 12 | } 13 | 14 | func (s *SystemInfo) FromBytes(data []byte) error { 15 | return json.Unmarshal(data, s) 16 | } 17 | 18 | type Disk struct { 19 | Device string `json:"device"` // e.g. /dev/sda1 20 | Mountpoint string `json:"mountpoint"` // e.g. / 21 | Fstype string `json:"fstype"` // e.g. ext4 22 | Opts string `json:"opts"` // e.g. rw 23 | Total uint64 `json:"total"` // total disk space in bytes 24 | Used uint64 `json:"used"` // used disk space in bytes 25 | } 26 | 27 | type SystemStat struct { 28 | CPUUsage []float64 `json:"cpu_usage"` 29 | MemUsage float64 `json:"mem_usage"` 30 | DiskUsage float64 `json:"disk_usage"` 31 | NetSentPS int64 `json:"net_sent_ps"` 32 | NetRecvPS int64 `json:"net_recv_ps"` 33 | } 34 | 35 | func (s *SystemStat) FromBytes(data []byte) error { 36 | return json.Unmarshal(data, s) 37 | } 38 | 39 | type Msg struct { 40 | Msg string 41 | NodeId int32 42 | Token string 43 | Data []byte 44 | } 45 | -------------------------------------------------------------------------------- /server/internal/utils/bytecovertor.go: -------------------------------------------------------------------------------- 1 | package utils 2 | 3 | func BytesToKB(bytes float64) float64 { 4 | return float64(bytes) / 1024 5 | } 6 | 7 | func BytesToMB(bytes float64) float64 { 8 | return float64(bytes) / (1024 * 1024) 9 | } 10 | 11 | func BytesToGB(bytes float64) float64 { 12 | return float64(bytes) / (1024 * 1024 * 1024) 13 | } 14 | -------------------------------------------------------------------------------- /server/internal/utils/password.go: -------------------------------------------------------------------------------- 1 | package utils 2 | 3 | import "golang.org/x/crypto/bcrypt" 4 | 5 | func HashString(str string) ([]byte, error) { 6 | return bcrypt.GenerateFromPassword([]byte(str), bcrypt.DefaultCost) 7 | } 8 | 9 | func VerifyPassword(password, hashedPassword string) error { 10 | return bcrypt.CompareHashAndPassword([]byte(hashedPassword), []byte(password)) 11 | } 12 | -------------------------------------------------------------------------------- /server/internal/utils/token.go: -------------------------------------------------------------------------------- 1 | package utils 2 | 3 | import ( 4 | "fmt" 5 | "log" 6 | "net/http" 7 | "os" 8 | "strconv" 9 | "time" 10 | 11 | "github.com/dgrijalva/jwt-go" 12 | "github.com/gin-gonic/gin" 13 | ) 14 | 15 | func ExtractTokenFromHeader(c *gin.Context) string { 16 | // get token from cookie 17 | cookie, err := c.Request.Cookie("__tkn__") 18 | 19 | if err == nil { 20 | return cookie.Value 21 | } 22 | 23 | return "" 24 | 25 | } 26 | 27 | // func ExtractTokenFromHeader(c *gin.Context) string { 28 | // bearerToken := c.Request.Header.Get("Authorization") 29 | // if len(bearerToken) > 7 && bearerToken[:7] == "Bearer " { 30 | // return bearerToken[7:] 31 | // } 32 | // return "" 33 | // } 34 | 35 | func GenerateToken(user_id int32) (string, error) { 36 | token_lifespan, err := strconv.Atoi(os.Getenv("TOKEN_LIFESPAN")) // in minutes 37 | if err != nil { 38 | log.Println("token life span error: ", err) 39 | return "", err 40 | } 41 | 42 | claims := jwt.MapClaims{} 43 | claims["authorized"] = true 44 | claims["user_id"] = user_id 45 | claims["exp"] = time.Now().Add(time.Minute * time.Duration(token_lifespan)).Unix() 46 | token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims) 47 | 48 | return token.SignedString([]byte(os.Getenv("TOKEN_SECRET"))) 49 | 50 | } 51 | 52 | func TokenValid(c *gin.Context) error { 53 | tokenString := ExtractTokenFromHeader(c) 54 | _, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) { 55 | if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok { 56 | return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"]) 57 | } 58 | return []byte(os.Getenv("TOKEN_SECRET")), nil 59 | }) 60 | if err != nil { 61 | 62 | return err 63 | } 64 | return nil 65 | } 66 | 67 | func ExtractTokenID(c *gin.Context) (int32, error) { 68 | 69 | tokenString := ExtractTokenFromHeader(c) 70 | token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) { 71 | if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok { 72 | return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"]) 73 | } 74 | return []byte(os.Getenv("TOKEN_SECRET")), nil 75 | }) 76 | if err != nil { 77 | return 0, err 78 | } 79 | claims, ok := token.Claims.(jwt.MapClaims) 80 | if ok && token.Valid { 81 | user_id := fmt.Sprintf("%v", claims["user_id"]) 82 | userID, err := strconv.ParseUint(user_id, 10, 64) 83 | if err != nil { 84 | return 0, err 85 | } 86 | return int32(userID), nil 87 | } 88 | return 0, nil 89 | } 90 | 91 | func WriteTokenToCookie(c *gin.Context, token string) { 92 | //set http only cookie 93 | cookie := &http.Cookie{ 94 | Name: "__tkn__", 95 | Value: token, 96 | HttpOnly: true, 97 | Secure: false, //TODO: change to true in production 98 | Path: "/", 99 | SameSite: http.SameSiteLaxMode, 100 | } 101 | http.SetCookie(c.Writer, cookie) 102 | 103 | } 104 | -------------------------------------------------------------------------------- /server/main.go: -------------------------------------------------------------------------------- 1 | package main 2 | 3 | import ( 4 | "context" 5 | "database/sql" 6 | "flag" 7 | "fmt" 8 | "log" 9 | "os" 10 | 11 | "github.com/joho/godotenv" 12 | _ "github.com/lib/pq" 13 | "github.com/sanda0/vps_pilot/cmd/app" 14 | "github.com/sanda0/vps_pilot/cmd/cli" 15 | "github.com/sanda0/vps_pilot/internal/db" 16 | "github.com/sanda0/vps_pilot/internal/tcpserver" 17 | ) 18 | 19 | func main() { 20 | 21 | err := godotenv.Load(".env") 22 | if err != nil { 23 | log.Fatal("Error loading .env file") 24 | } 25 | 26 | //init db 27 | con, err := sql.Open("postgres", fmt.Sprintf("dbname=%s password=%s user=%s host=%s sslmode=require", 28 | os.Getenv("DB_NAME"), 29 | os.Getenv("DB_PASSWORD"), 30 | os.Getenv("DB_USER"), 31 | os.Getenv("DB_HOST"), 32 | )) 33 | 34 | if err != nil { 35 | panic(err) 36 | } 37 | 38 | //init ctx 39 | ctx := context.Background() 40 | repo := db.NewRepo(con) 41 | 42 | port := flag.String("port", "8080", "port to listen on") 43 | createSuperuser := flag.Bool("create-superuser", false, "create superuser") 44 | createMakefile := flag.Bool("create-makefile", false, "create makefile") 45 | flag.Parse() 46 | 47 | if *createSuperuser { 48 | cli.CreateSuperuser(ctx, repo) 49 | return 50 | } 51 | 52 | if *createMakefile { 53 | err := cli.CreateMakeFile() 54 | if err != nil { 55 | log.Fatal("Error creating Makefile") 56 | } 57 | return 58 | } 59 | 60 | //init tcp server 61 | go tcpserver.StartTcpServer(ctx, repo, "55001") 62 | 63 | app.Run(ctx, repo, *port) 64 | 65 | } 66 | -------------------------------------------------------------------------------- /server/makefile: -------------------------------------------------------------------------------- 1 | 2 | migration: 3 | @read -p "Enter migration name: " name; \ 4 | migrate create -ext sql -dir internal/db/sql/migrations $$name 5 | 6 | migrate: 7 | migrate -source file://internal/db/sql/migrations \ 8 | -database postgres://postgres:1234@127.0.0.1:5432/vps_pilot?sslmode=disable up 9 | 10 | rollback: 11 | migrate -source file://internal/db/sql/migrations \ 12 | -database postgres://postgres:1234@127.0.0.1:5432/vps_pilot?sslmode=disable down 13 | 14 | drop: 15 | migrate -source file://internal/db/sql/migrations \ 16 | -database postgres://postgres:1234@127.0.0.1:5432/vps_pilot?sslmode=disable drop 17 | 18 | sqlc: 19 | sqlc generate 20 | 21 | 22 | migratef: 23 | @read -p "Enter migration number: " num; \ 24 | migrate -source file://internal/db/sql/migrations \ 25 | -database postgres://postgres:1234@127.0.0.1:5432/vps_pilot?sslmode=disable force $$num 26 | 27 | -------------------------------------------------------------------------------- /server/sqlc.yml: -------------------------------------------------------------------------------- 1 | version: "2" 2 | sql: 3 | - engine: "postgresql" 4 | queries: "internal/db/sql/query/" 5 | schema: "internal/db/sql/migrations/" 6 | gen: 7 | go: 8 | package: "db" 9 | out: "internal/db" 10 | emit_json_tags: true 11 | emit_prepared_queries: true -------------------------------------------------------------------------------- /server/test/custom_query_test.go: -------------------------------------------------------------------------------- 1 | package test 2 | 3 | import ( 4 | "context" 5 | "testing" 6 | 7 | "github.com/sanda0/vps_pilot/internal/db" 8 | ) 9 | 10 | func TestGetCPUStats(t *testing.T) { 11 | q := db.Queries{} 12 | 13 | // Test case 1 14 | _, err := q.GetCPUStats(context.Background(), db.GetCPUStatsParams{ 15 | NodeID: 1, 16 | TimeRange: "1 day", 17 | CpuCount: 8, 18 | }) 19 | if err != nil { 20 | panic(err) 21 | } 22 | 23 | t.Log("GetCPUStats test case 1") 24 | 25 | } 26 | -------------------------------------------------------------------------------- /tmp/build-errors.log: -------------------------------------------------------------------------------- 1 | exit status 1 --------------------------------------------------------------------------------