feat: update README, signal hub, and add project infrastructure

Author: shijiashuai

CONTRIBUTING.md

@@ -0,0 +1,35 @@
+# Contributing
+
+感谢你对本项目的关注！欢迎通过 Issue 和 Pull Request 参与贡献。
+
+## 开发流程
+
+1. Fork 本仓库
+2. 创建特性分支：`git checkout -b feature/your-feature`
+3. 提交更改：`git commit -m "feat: add your feature"`
+4. 推送分支：`git push origin feature/your-feature`
+5. 创建 Pull Request
+
+## 开发与测试
+
+```bash
+go mod tidy
+go run ./cmd/server
+```
+
+## 代码规范
+
+- Go 代码遵循 `gofmt` 和 `go vet` 规范
+- 使用 `.editorconfig` 中定义的缩进和格式规则
+- 前端代码保持简洁，避免引入重型框架
+- 确保构建无错误
+
+## 提交信息格式
+
+推荐使用 [Conventional Commits](https://www.conventionalcommits.org/)：
+
+- `feat:` 新功能
+- `fix:` 修复 Bug
+- `docs:` 文档更新
+- `refactor:` 重构
+- `chore:` 构建/工具链

README.md

@@ -1,5 +1,10 @@
 # WebRTC
 
+[![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](LICENSE)
+![Go](https://img.shields.io/badge/Go-1.22+-00ADD8?logo=go&logoColor=white)
+![WebRTC](https://img.shields.io/badge/WebRTC-Enabled-333333?logo=webrtc&logoColor=white)
+![Docker](https://img.shields.io/badge/Docker-Ready-2496ED?logo=docker&logoColor=white)
+
 WebRTC 是一个基于 Go 的最小可用 WebRTC 示例项目，提供 WebSocket 信令服务器与浏览器端 Demo，适合作为实时音视频应用的入门模板。
 
 ## 特性

changelog/2025-02-13_project-infrastructure.md

@@ -0,0 +1,4 @@
+# 2025-02-13 项目基础设施优化
+
+## 新增
+- README 添加标准化 badges（License、Go、WebRTC、Docker）

cmd/server/main.go

@@ -13,28 +13,30 @@ import (
 	sig "lessup/webrtc/internal/signal"
 )
 
+func parseOrigins(raw string) (origins []string, allowAll bool) {
+	if raw == "" {
+		return nil, false
+	}
+	if raw == "*" {
+		return nil, true
+	}
+	parts := strings.Split(raw, ",")
+	origins = make([]string, 0, len(parts))
+	for _, p := range parts {
+		if p = strings.TrimSpace(p); p != "" {
+			origins = append(origins, p)
+		}
+	}
+	return origins, false
+}
+
 func main() {
 	addr := ":8080"
 	if v := os.Getenv("ADDR"); v != "" {
 		addr = v
 	}
 
-	var wsAllowAll bool
-	var wsAllowed []string
-	if v := os.Getenv("WS_ALLOWED_ORIGINS"); v != "" {
-		if v == "*" {
-			wsAllowAll = true
-		} else {
-			parts := strings.Split(v, ",")
-			wsAllowed = make([]string, 0, len(parts))
-			for _, p := range parts {
-				p = strings.TrimSpace(p)
-				if p != "" {
-					wsAllowed = append(wsAllowed, p)
-				}
-			}
-		}
-	}
+	wsAllowed, wsAllowAll := parseOrigins(os.Getenv("WS_ALLOWED_ORIGINS"))
 
 	hub := sig.NewHubWithOptions(sig.Options{
 		AllowedOrigins:  wsAllowed,

internal/signal/hub.go

@@ -56,10 +56,7 @@ func (h *Hub) isOriginAllowed(r *http.Request) bool {
 	origin := r.Header.Get("Origin")
 	if origin == "" {
 		host := r.Host
-		if host == "localhost:8080" || strings.HasPrefix(host, "127.0.0.1:") || strings.HasPrefix(host, "localhost:") {
-			return true
-		}
-		return false
+		return strings.HasPrefix(host, "localhost:") || strings.HasPrefix(host, "127.0.0.1:")
 	}
 	if len(h.allowedOrigins) == 0 {
 		u, err := url.Parse(origin)
@@ -133,7 +130,7 @@ func (h *Hub) addClient(c *Client) {
 	}
 	m[c.id] = c
 	log.Printf("signal: join room=%s id=%s", c.room, c.id)
-	h.broadcastMembers(c.room, m)
+	broadcastMembers(c.room, m)
 }
 
 func (h *Hub) removeClient(c *Client) {
@@ -149,19 +146,20 @@ func (h *Hub) removeClient(c *Client) {
 			return
 		}
 		delete(m, c.id)
+		log.Printf("signal: leave room=%s id=%s", room, c.id)
 		c.room = ""
 		if len(m) == 0 {
 			delete(h.rooms, room)
 			log.Printf("signal: room %s closed", room)
 			return
 		}
-		h.broadcastMembers(room, m)
+		broadcastMembers(room, m)
 	}
 }
 
 // broadcastMembers sends the current member list to all clients in a room.
 // Must be called with h.mu held.
-func (h *Hub) broadcastMembers(room string, m map[string]*Client) {
+func broadcastMembers(room string, m map[string]*Client) {
 	members := make([]string, 0, len(m))
 	for id := range m {
 		members = append(members, id)

internal/signal/hub_test.go

@@ -1,11 +1,44 @@
 package signal
 
 import (
+	"net/http"
 	"testing"
 
 	"github.com/gorilla/websocket"
 )
 
+func TestIsOriginAllowed(t *testing.T) {
+	tests := []struct {
+		name    string
+		opts    Options
+		host    string
+		origin  string
+		allowed bool
+	}{
+		{"no origin localhost", Options{}, "localhost:8080", "", true},
+		{"no origin 127.0.0.1", Options{}, "127.0.0.1:8080", "", true},
+		{"no origin external", Options{}, "example.com:8080", "", false},
+		{"origin localhost", Options{}, "localhost:8080", "http://localhost:8080", true},
+		{"origin 127.0.0.1", Options{}, "127.0.0.1:8080", "http://127.0.0.1:8080", true},
+		{"origin external blocked", Options{}, "example.com", "https://example.com", false},
+		{"allow all", Options{AllowAllOrigins: true}, "example.com", "https://evil.com", true},
+		{"whitelist match", Options{AllowedOrigins: []string{"https://example.com"}}, "", "https://example.com", true},
+		{"whitelist miss", Options{AllowedOrigins: []string{"https://example.com"}}, "", "https://evil.com", false},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			h := NewHubWithOptions(tt.opts)
+			r := &http.Request{Host: tt.host, Header: http.Header{}}
+			if tt.origin != "" {
+				r.Header.Set("Origin", tt.origin)
+			}
+			if got := h.isOriginAllowed(r); got != tt.allowed {
+				t.Errorf("isOriginAllowed() = %v, want %v", got, tt.allowed)
+			}
+		})
+	}
+}
+
 func drainMessages(ch <-chan Message) {
 	for {
 		select {