import * as sst from "@serverless-stack/resources";

export default class MyStack extends sst.Stack {

constructor(scope, id, props) {

super(scope, id, props);

// Create the table

const table = new sst.Table(this, "Connections", {

fields: {

id: sst.TableFieldType.STRING,

},

primaryIndex: { partitionKey: "id" },

});

// Create the WebSocket API

const api = new sst.WebSocketApi(this, "Api", {

defaultFunctionProps: {

environment: {

tableName: table.dynamodbTable.tableName,

},

},

routes: {

$connect: "src/connect.main",

$disconnect: "src/disconnect.main",

sendmessage: "src/sendMessage.main",

},

});

// Allow the API to access the table

api.attachPermissions([table]);

// Show the API endpoint in the output

this.addOutputs({

ApiEndpoint: api.url,

});

}

}