command.go

package main

import (
	"strconv"
	"strings"
)

func handleCommand(c *Client, store *Storage, parts []string) {
	cmd := strings.ToUpper(parts[0])
	switch cmd {
	case "SET":
		if !canWrite(c.User) {
			c.Conn.Write([]byte("ERR no permission\n"))
			return
		}
		if len(parts) < 3 {
			c.Conn.Write([]byte("ERR usage: SET key value [TTL_seconds]\n"))
			return
		}
		ttl := 0
		if len(parts) == 4 {
			t, err := strconv.Atoi(parts[3])
			if err != nil {
				c.Conn.Write([]byte("ERR TTL must be integer\n"))
				return
			}
			ttl = t
		}
		store.Set(parts[1], parts[2], ttl)
		c.Conn.Write([]byte("OK\n"))
	case "GET":
		if !canRead(c.User) {
			c.Conn.Write([]byte("ERR no permission\n"))
			return
		}
		if len(parts) != 2 {
			c.Conn.Write([]byte("ERR usage: GET key\n"))
			return
		}
		val, ok := store.Get(parts[1])
		if !ok {
			c.Conn.Write([]byte("(nil)\n"))
			return
		}
		c.Conn.Write([]byte(val + "\n"))
	default:
		c.Conn.Write([]byte("ERR unknown command\n"))
	}
}