Add support for opening place files using Studio command

CHANGELOG.md

@@ -6,6 +6,10 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 
 ## [Unreleased]
 
+### Added
+
+- It is now possible to open `.rbxl` pr `.rbxlx` file using `Studio` command
+
 ## [2.0.3] - 2024-05-05
 
 ### Changed

src/argon.ts

@@ -101,12 +101,8 @@ export function init(project: string, template: string, options: string[]) {
  spawn(['init', project, '--template', template, ...options])
 }
 
-export function stop(id: number | number[]) {
- if (typeof id === 'number') {
- spawn(['stop', id.toString()])
- } else {
- spawn(['stop', ...id.map((id) => id.toString())])
- }
+export function stop(ids: number[]) {
+ spawn(['stop', ...ids.map((id) => id.toString())])
 }
 
 export function debug(mode: string) {
@@ -118,8 +114,17 @@ export function exec(code: string, focus?: boolean) {
  spawn(['exec', code, ...args])
 }
 
-export function studio(check?: boolean) {
- const args = check ? ['--check'] : []
+export function studio(check?: boolean, place?: string) {
+ const args = []
+
+ if (place) {
+ args.push(place)
+ }
+
+ if (check) {
+ args.push('--check')
+ }
+
  spawn(['studio', ...args])
 }
 

src/menu/index.ts

@@ -104,7 +104,7 @@ export async function onDidAccept(action: string, state: State) {
  exec.handler()
  break
  case 'studio':
- studio.handler()
+ await studio.handler()
  break
  case 'plugin':
  await plugin.handler()

src/menu/stop.ts

@@ -47,10 +47,7 @@ export async function handler(state: State): Promise<void> {
 
  if (ids.length !== 0) {
  argon.stop(ids)
-
- ids.forEach((id) => {
- state.removeSession(id)
- })
+ state.removeSessions(ids)
  }
 
  resolve()

src/menu/studio.ts

@@ -1,4 +1,6 @@
+import * as vscode from 'vscode'
 import * as argon from '../argon'
+import { findPlaces } from '../util'
 import { Item } from '.'
 
 export const item: Item = {
@@ -7,6 +9,31 @@ export const item: Item = {
  action: 'studio',
 }
 
-export function handler() {
- argon.studio()
+function getPlace(): Promise<string | undefined> {
+ return new Promise((resolve, reject) => {
+ const places = findPlaces()
+ places.unshift('$(window) Launch empty')
+
+ vscode.window
+ .showQuickPick(places, {
+ title: 'Select a place',
+ })
+ .then(async (place) => {
+ if (!place) {
+ return reject()
+ }
+
+ place.endsWith('.rbxl') || place.endsWith('.rbxlx')
+ ? resolve(place)
+ : resolve(undefined)
+ })
+ })
+}
+
+export async function handler() {
+ const place = await getPlace()
+
+ console.log(place)
+
+ argon.studio(false, place)
 }

src/state.ts

@@ -40,11 +40,11 @@ export class State {
  this.context.workspaceState.update('lastProject', session.project)
  }
 
- public removeSession(id: number) {
+ public removeSessions(ids: number[]) {
  const lastProject = this.context.workspaceState.get('lastProject')
 
  this.sessions = this.sessions.filter((session) => {
- const matches = session.id === id
+ const matches = ids.includes(session.id)
 
  if (matches && session.project === lastProject) {
  this.context.workspaceState.update('lastProject', undefined)
@@ -61,10 +61,10 @@ export class State {
  }
 
  public cleanup() {
- this.sessions.forEach((session) => {
- console.log(`Stopping session ${session.id}...`)
-  argon.stop(session.id)
- })
+ const ids = this.sessions.map((session) => session.id)
+
+ console.log(`Stopping ${ids.length} sessions...`)
+ argon.stop(ids)
  }
 
  private updateItem() {

src/util.ts

@@ -44,6 +44,24 @@ export function findProjects(placesOnly?: boolean): string[] {
  return projects
 }
 
+export function findPlaces(): string[] {
+ const dir = getCurrentDir()
+
+ if (!dir) {
+ throw new Error('Cannot find places without a workspace folder open!')
+ }
+
+ let places = fs
+ .readdirSync(dir)
+ .filter(
+ (entry) =>
+ (entry.endsWith('.rbxl') || entry.endsWith('.rbxlx')) &&
+ fs.statSync(path.join(dir, entry)).isFile(),
+ )
+
+ return places
+}
+
 export function getProjectName(project: string): string {
  if (!path.isAbsolute(project)) {
  const dir = getCurrentDir()