package main

import (
    "encoding/json"
    "log"
    "net/http"

    "github.com/sairash/chitosocket"
)

func main() {
    // Create socket server (pass number of CPU cores)
    socket, err := chitosocket.StartUp(1)
    if err != nil {
        log.Fatal(err)
    }
    defer socket.Close()

    // Handle connection
    socket.On["connected"] = func(sub *chitosocket.Subscriber, data []byte) {
        welcome := map[string]string{"id": sub.ID}
        welcomeData, _ := json.Marshal(welcome)
        socket.EmitDirect(sub, "welcome", welcomeData)
    }

    // Handle messages
    socket.On["chat"] = func(sub *chitosocket.Subscriber, data []byte) {
        socket.Emit("chat", data, "general")
    }

    // Handle disconnect
    socket.On["disconnect"] = func(sub *chitosocket.Subscriber, data []byte) {
        log.Printf("Client %s disconnected", sub.ID)
    }

    // HTTP handler for WebSocket upgrades
    http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
        _, _, sub, err := socket.UpgradeConnection(r, w)
        if err != nil {
            return
        }
        if handler, ok := socket.On["connected"]; ok {
            handler(sub, nil)
        }
    })

    log.Println("Server starting on :8080")
    http.ListenAndServe(":8080", nil)
}

// With default config based on CPU cores
socket, err := chitosocket.StartUp(numCPU)

// With custom config
config := chitosocket.Config{
    NumWorkers:    8,
    NumShards:     16,
    EventChanSize: 8192,
}
socket, err := chitosocket.StartUpWithConfig(config)

// Listen for events
socket.On["event_name"] = func(sub *chitosocket.Subscriber, data []byte) {
    // Handle event
}

// Special events
socket.On["connected"]   // Called when client connects
socket.On["disconnect"]  // Called when client disconnects

// Send to specific subscriber
socket.EmitDirect(sub, "event", data)

// Send to all subscribers in room(s)
// if excludeSub is not nil, the message will not be sent to the subscriber
socket.Emit("event", nil, data, "room1", "room2")


// Broadcast to all connected clients
socket.Broadcast("event", data)

// Add subscriber to room
socket.AddToRoom(sub, "room_name")

// Remove subscriber from room
socket.RemoveFromRoom(sub, "room_name")

// Get all members in a room
members := socket.GetRoomMembers("room_name")

// Get count of members in a room
count := socket.GetRoomCount("room_name")

// Send message (non-blocking, returns false if channel full)
sub.Send(payload)

// Check if subscriber is closed
sub.IsClosed()

// Store metadata
sub.Metadata.Store("key", value)
val, ok := sub.Metadata.Load("key")