Compare commits
13 Commits
| Author | SHA1 | Date |
|---|---|---|
|
|
001fef8bbc | |
|
|
a9004c38af | |
|
|
b34c980ea5 | |
|
|
78fd4fd28b | |
|
|
b89d8e3d3f | |
|
|
c4a21861cd | |
|
|
d5c3a52215 | |
|
|
bc1f7779d7 | |
|
|
49880c837d | |
|
|
4dd5a40dfe | |
|
|
13b583a77e | |
|
|
c9c4800d68 | |
|
|
d793e13361 |
|
|
@ -1,4 +1,13 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"tts-room/src/server"
|
||||||
|
)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
s := server.NewServer()
|
||||||
|
if err := http.ListenAndServe(":10000", s); err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
9
go.mod
9
go.mod
|
|
@ -1,3 +1,10 @@
|
||||||
module tts-room
|
module tts-room
|
||||||
|
|
||||||
go 1.22.2
|
go 1.24.0
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/gorilla/websocket v1.5.3
|
||||||
|
golang.org/x/time v0.14.0
|
||||||
|
)
|
||||||
|
|
||||||
|
require github.com/google/uuid v1.6.0 // indirect
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,6 @@
|
||||||
|
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||||
|
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
|
github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
|
||||||
|
github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||||
|
golang.org/x/time v0.14.0 h1:MRx4UaLrDotUKUdCIqzPC48t1Y9hANFKIRpNx+Te8PI=
|
||||||
|
golang.org/x/time v0.14.0/go.mod h1:eL/Oa2bBBK0TkX57Fyni+NgnyQQN4LitPmob2Hjnqw4=
|
||||||
|
|
@ -0,0 +1,158 @@
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<meta http-equiv="X-UA-Compatible" content="IE=edge" />
|
||||||
|
<meta name="viewport" content="width=device-width" />
|
||||||
|
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/water.css@2/out/water.css">
|
||||||
|
<script>
|
||||||
|
window.addEventListener("load", function(evt) {
|
||||||
|
var output = document.getElementById("output");
|
||||||
|
var input = document.getElementById("input");
|
||||||
|
var ws;
|
||||||
|
|
||||||
|
var print = function(message) {
|
||||||
|
var before = output.innerHTML.split("<br>");
|
||||||
|
before.unshift(`<span>${new Date().toLocaleTimeString()} | ${message}</span>`);
|
||||||
|
if (before.length > 20) {
|
||||||
|
before = before.slice(0, 20);
|
||||||
|
}
|
||||||
|
var after = before.join("<br>");
|
||||||
|
output.innerHTML = after;
|
||||||
|
output.scroll(0, 0);
|
||||||
|
};
|
||||||
|
|
||||||
|
const synth = window.speechSynthesis;
|
||||||
|
const voices = synth.getVoices().sort(function (a, b) {
|
||||||
|
const aname = a.name.toUpperCase();
|
||||||
|
const bname = b.name.toUpperCase();
|
||||||
|
|
||||||
|
if (aname < bname) {
|
||||||
|
return -1;
|
||||||
|
} else if (aname == bname) {
|
||||||
|
return 0;
|
||||||
|
} else {
|
||||||
|
return +1;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
var voices_idx = 0;
|
||||||
|
|
||||||
|
{
|
||||||
|
var ele = document.getElementById("voices");
|
||||||
|
for(var i in voices) {
|
||||||
|
var voice = voices[i];
|
||||||
|
const opt = document.createElement("input");
|
||||||
|
opt.type = "radio";
|
||||||
|
opt.name = "voice";
|
||||||
|
opt.id = `voices-${i}-${voice.name}`
|
||||||
|
opt.value = i;
|
||||||
|
|
||||||
|
const label = document.createElement("label");
|
||||||
|
label.for = opt.id;
|
||||||
|
label.textContent = voice.name;
|
||||||
|
|
||||||
|
ele.appendChild(opt);
|
||||||
|
ele.appendChild(label);
|
||||||
|
ele.appendChild(document.createElement("br"));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ws = new WebSocket("ws://"+window.location.host+"/ws"+window.location.search);
|
||||||
|
ws.onopen = function(evt) {
|
||||||
|
print("READY");
|
||||||
|
}
|
||||||
|
ws.onclose = function(evt) {
|
||||||
|
print("CLOSE");
|
||||||
|
ws = null;
|
||||||
|
}
|
||||||
|
ws.onmessage = function(evt) {
|
||||||
|
console.log("evt.data:", evt.data)
|
||||||
|
const data = JSON.parse(evt.data);
|
||||||
|
|
||||||
|
const utterThis = new SpeechSynthesisUtterance(data.Text);
|
||||||
|
|
||||||
|
const idx = data.VoiceIdx || 0;
|
||||||
|
utterThis.voice = voices[idx];
|
||||||
|
|
||||||
|
if (data.Pitch) {
|
||||||
|
utterThis.pitch = data.Pitch
|
||||||
|
}
|
||||||
|
|
||||||
|
if (data.Rate) {
|
||||||
|
utterThis.rate = data.Rate
|
||||||
|
}
|
||||||
|
|
||||||
|
window.speechSynthesis.speak(utterThis);
|
||||||
|
print(data.Text);
|
||||||
|
}
|
||||||
|
ws.onerror = function(evt) {
|
||||||
|
print("ERROR: " + evt.data);
|
||||||
|
}
|
||||||
|
|
||||||
|
document.getElementById("send").onclick = function(evt) {
|
||||||
|
if (!ws || !input.value) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
var voiceIdx = "0";
|
||||||
|
var voiceEles = document.getElementsByName("voice");
|
||||||
|
for (var i = 0; i < voiceEles.length; i++)
|
||||||
|
if (voiceEles[i].checked)
|
||||||
|
voiceIdx = voiceEles[i].value;
|
||||||
|
|
||||||
|
const data = JSON.stringify({
|
||||||
|
"Text": input.value,
|
||||||
|
"Pitch": Number.parseInt(document.getElementById("pitch").value, 10),
|
||||||
|
"Rate": Number.parseFloat(document.getElementById("rate").value, 10),
|
||||||
|
"VoiceIdx": Number.parseInt(voiceIdx, 10),
|
||||||
|
});
|
||||||
|
|
||||||
|
ws.send(data);
|
||||||
|
print("SENT: " + input.value);
|
||||||
|
input.value = "";
|
||||||
|
if (document.getElementById("listen").checked) {
|
||||||
|
ws.onmessage({data: data});
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
};
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
<style>
|
||||||
|
#output > span {
|
||||||
|
opacity: 0.33;
|
||||||
|
}
|
||||||
|
#output > span:nth-child(1) {
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
#output > span:nth-child(2)
|
||||||
|
, #output > span:nth-child(3)
|
||||||
|
{
|
||||||
|
opacity: 0.66;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div style="width: 80%; height: 80%; margin: auto; display: flex; flex-direction: row;">
|
||||||
|
<div style="display: flex; flex-direction: column;">
|
||||||
|
<form style="flex-grow: 1;">
|
||||||
|
<p><input id="input" type="textarea" value="" autofocus style="width: 80%">
|
||||||
|
<p>
|
||||||
|
<button id="send">Send</button>
|
||||||
|
</form>
|
||||||
|
<div>
|
||||||
|
<form id="voices"></form>
|
||||||
|
|
||||||
|
<input id="pitch" type="number" value="0"/>
|
||||||
|
<label for="pitch">pitch</label></br>
|
||||||
|
|
||||||
|
<input id="rate" type="number" value="0"/>
|
||||||
|
<label for="rate">rate</label></br>
|
||||||
|
|
||||||
|
<input id="listen" type="checkbox" />
|
||||||
|
<label for="listen">listen</label></br>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div id="output" style="flex-grow: 1; overflow-y: scroll;"></div>
|
||||||
|
</div>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
|
@ -0,0 +1,9 @@
|
||||||
|
package server
|
||||||
|
|
||||||
|
type message struct {
|
||||||
|
Text string
|
||||||
|
Pitch int
|
||||||
|
Rate float64
|
||||||
|
VoiceIdx int
|
||||||
|
room string
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,60 @@
|
||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Server struct {
|
||||||
|
sessions []*session
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewServer() *Server {
|
||||||
|
return &Server{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||||
|
switch r.URL.Path {
|
||||||
|
case "/ws":
|
||||||
|
if err := s.WS(w, r); err != nil {
|
||||||
|
log.Println("[ws]", err)
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
http.FileServer(http.Dir("./src/public")).ServeHTTP(w, r)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Server) WS(w http.ResponseWriter, r *http.Request) error {
|
||||||
|
sess, err := newSession(w, r, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer sess.Close()
|
||||||
|
|
||||||
|
log.Println("someone has joined", sess.room)
|
||||||
|
defer log.Println("someone has left", sess.room)
|
||||||
|
|
||||||
|
sess.cb = func(m message) error {
|
||||||
|
for i := range s.sessions {
|
||||||
|
if s.sessions[i].id != sess.id && s.sessions[i].room == sess.room {
|
||||||
|
select {
|
||||||
|
case s.sessions[i].scatterc <- m:
|
||||||
|
case <-s.sessions[i].ctx.Done():
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
s.sessions = append(s.sessions, sess)
|
||||||
|
defer func() {
|
||||||
|
for i := range s.sessions {
|
||||||
|
if s.sessions[i].id == sess.id {
|
||||||
|
s.sessions = append(s.sessions[:i], s.sessions[i+1:]...)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
return sess.Run()
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,106 @@
|
||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"github.com/gorilla/websocket"
|
||||||
|
"golang.org/x/time/rate"
|
||||||
|
)
|
||||||
|
|
||||||
|
type session struct {
|
||||||
|
ctx context.Context
|
||||||
|
can context.CancelFunc
|
||||||
|
ws *websocket.Conn
|
||||||
|
wg sync.WaitGroup
|
||||||
|
cb func(message) error
|
||||||
|
id string
|
||||||
|
scatterc chan (message)
|
||||||
|
room string
|
||||||
|
}
|
||||||
|
|
||||||
|
var upgrader = websocket.Upgrader{}
|
||||||
|
|
||||||
|
func newSession(w http.ResponseWriter, r *http.Request, cb func(message) error) (*session, error) {
|
||||||
|
c, err := upgrader.Upgrade(w, r, nil)
|
||||||
|
ctx, can := context.WithCancel(r.Context())
|
||||||
|
return &session{
|
||||||
|
ctx: ctx,
|
||||||
|
can: can,
|
||||||
|
ws: c,
|
||||||
|
cb: cb,
|
||||||
|
id: uuid.New().String(),
|
||||||
|
scatterc: make(chan message, 20),
|
||||||
|
room: r.URL.Query().Get("room"),
|
||||||
|
}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *session) Close() {
|
||||||
|
if s.ws != nil {
|
||||||
|
s.ws.Close()
|
||||||
|
}
|
||||||
|
s.ws = nil
|
||||||
|
s.can()
|
||||||
|
s.wg.Wait()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *session) Run() error {
|
||||||
|
defer s.Close()
|
||||||
|
|
||||||
|
go s.gather()
|
||||||
|
go s.scatter()
|
||||||
|
|
||||||
|
<-s.ctx.Done()
|
||||||
|
return s.ctx.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *session) gather() {
|
||||||
|
s.while(func() error {
|
||||||
|
mt, msg, err := s.ws.ReadMessage()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if mt != 1 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var m message
|
||||||
|
if err := json.Unmarshal(msg, &m); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
m.room = s.room
|
||||||
|
|
||||||
|
return s.cb(m)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *session) scatter() {
|
||||||
|
s.while(func() error {
|
||||||
|
select {
|
||||||
|
case m := <-s.scatterc:
|
||||||
|
b, _ := json.Marshal(m)
|
||||||
|
return s.ws.WriteMessage(1, b)
|
||||||
|
case <-s.ctx.Done():
|
||||||
|
return s.ctx.Err()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *session) while(foo func() error) {
|
||||||
|
defer s.can()
|
||||||
|
|
||||||
|
s.wg.Add(1)
|
||||||
|
defer s.wg.Done()
|
||||||
|
|
||||||
|
l := rate.NewLimiter(20, 1)
|
||||||
|
for l.Wait(s.ctx) == nil {
|
||||||
|
if err := foo(); err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue