120 lines
2.9 KiB
Go
120 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/github/go-liveshare"
|
|
)
|
|
|
|
var workspaceIdFlag = flag.String("w", "", "workspace session id")
|
|
|
|
func init() {
|
|
flag.Parse()
|
|
}
|
|
|
|
func main() {
|
|
liveShare, err := liveshare.New(
|
|
liveshare.WithWorkspaceID(*workspaceIdFlag),
|
|
liveshare.WithToken(os.Getenv("CODESPACE_TOKEN")),
|
|
)
|
|
if err != nil {
|
|
log.Fatal(fmt.Errorf("error creating liveshare: %v", err))
|
|
}
|
|
|
|
ctx := context.Background()
|
|
liveShareClient := liveShare.NewClient()
|
|
if err := liveShareClient.Join(ctx); err != nil {
|
|
log.Fatal(fmt.Errorf("error joining liveshare with client: %v", err))
|
|
}
|
|
|
|
terminal, err := liveShareClient.NewTerminal()
|
|
if err != nil {
|
|
log.Fatal(fmt.Errorf("error creating liveshare terminal"))
|
|
}
|
|
|
|
containerID, err := getContainerID(ctx, terminal)
|
|
if err != nil {
|
|
log.Fatal(fmt.Errorf("error getting container id: %v", err))
|
|
}
|
|
|
|
if err := setupSSH(ctx, terminal, containerID); err != nil {
|
|
log.Fatal(fmt.Errorf("error setting up ssh: %v", err))
|
|
}
|
|
|
|
fmt.Println("Starting server...")
|
|
|
|
server, err := liveShareClient.NewServer()
|
|
if err != nil {
|
|
log.Fatal(fmt.Errorf("error creating server: %v", err))
|
|
}
|
|
|
|
fmt.Println("Starting sharing...")
|
|
if err := server.StartSharing(ctx, "sshd", 2222); err != nil {
|
|
log.Fatal(fmt.Errorf("error server sharing: %v", err))
|
|
}
|
|
|
|
portForwarder := liveshare.NewLocalPortForwarder(liveShareClient, server, 2222)
|
|
|
|
fmt.Println("Listening on port 2222")
|
|
if err := portForwarder.Start(ctx); err != nil {
|
|
log.Fatal(fmt.Errorf("error forwarding port: %v", err))
|
|
}
|
|
}
|
|
|
|
func setupSSH(ctx context.Context, terminal *liveshare.Terminal, containerID string) error {
|
|
cmd := terminal.NewCommand(
|
|
"/",
|
|
fmt.Sprintf("/usr/bin/docker exec -t %s /bin/bash -c \"echo -e \\\"testpwd1\\ntestpwd1\\n\\\" | sudo passwd codespace;/usr/local/share/ssh-init.sh\"", containerID),
|
|
)
|
|
stream, err := cmd.Run(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("error running command: %v", err)
|
|
}
|
|
|
|
scanner := bufio.NewScanner(stream)
|
|
scanner.Scan()
|
|
|
|
fmt.Println("> Debug:", scanner.Text())
|
|
if err := scanner.Err(); err != nil {
|
|
return fmt.Errorf("error scanning stream: %v", err)
|
|
}
|
|
|
|
if err := stream.Close(); err != nil {
|
|
return fmt.Errorf("error closing stream: %v", err)
|
|
}
|
|
|
|
time.Sleep(2 * time.Second)
|
|
|
|
return nil
|
|
}
|
|
|
|
func getContainerID(ctx context.Context, terminal *liveshare.Terminal) (string, error) {
|
|
cmd := terminal.NewCommand(
|
|
"/",
|
|
"/usr/bin/docker ps -aq --filter label=Type=codespaces --filter status=running",
|
|
)
|
|
stream, err := cmd.Run(ctx)
|
|
if err != nil {
|
|
return "", fmt.Errorf("error running command: %v", err)
|
|
}
|
|
|
|
scanner := bufio.NewScanner(stream)
|
|
scanner.Scan()
|
|
|
|
containerID := scanner.Text()
|
|
if err := scanner.Err(); err != nil {
|
|
return "", fmt.Errorf("error scanning stream: %v", err)
|
|
}
|
|
|
|
if err := stream.Close(); err != nil {
|
|
return "", fmt.Errorf("error closing stream: %v", err)
|
|
}
|
|
|
|
return containerID, nil
|
|
}
|