+19

src/client.ts

··· 100 100 101 101 return response.json(); 102 102 } 103 103 + 104 104 + async search(query: string, { size }) { 105 105 + const response = await fetch( 106 106 + `${ROCKSKY_API_URL}/search?q=${query}&size=${size}`, 107 107 + { 108 108 + method: "GET", 109 109 + headers: { 110 110 + Authorization: this.token ? `Bearer ${this.token}` : undefined, 111 111 + "Content-Type": "application/json", 112 112 + }, 113 113 + } 114 114 + ); 115 115 + 116 116 + if (!response.ok) { 117 117 + throw new Error(`Failed to fetch search data: ${response.statusText}`); 118 118 + } 119 119 + 120 120 + return response.json(); 121 121 + } 103 122 }

+71

src/cmd/search.ts

··· 1 1 + import chalk from "chalk"; 2 2 + import { RockskyClient } from "client"; 3 3 + 4 4 + export async function search( 5 5 + query: string, 6 6 + { limit = 20, albums = false, artists = false, tracks = false, users = false } 7 7 + ) { 8 8 + const client = new RockskyClient(); 9 9 + const results = await client.search(query, { size: limit }); 10 10 + if (results.records.length === 0) { 11 11 + console.log(`No results found for ${chalk.magenta(query)}.`); 12 12 + return; 13 13 + } 14 14 + 15 15 + // merge all results into one array with type and sort by xata_scrore 16 16 + let mergedResults = results.records.map((record) => ({ 17 17 + ...record, 18 18 + type: record.table, 19 19 + })); 20 20 + mergedResults.sort((a, b) => b.xata_score - a.xata_score); 21 21 + 22 22 + if (albums) { 23 23 + mergedResults = mergedResults.filter((record) => record.table === "albums"); 24 24 + } 25 25 + 26 26 + if (artists) { 27 27 + mergedResults = mergedResults.filter( 28 28 + (record) => record.table === "artists" 29 29 + ); 30 30 + } 31 31 + 32 32 + if (tracks) { 33 33 + mergedResults = mergedResults.filter(({ table }) => table === "tracks"); 34 34 + } 35 35 + 36 36 + if (users) { 37 37 + mergedResults = mergedResults.filter(({ table }) => table === "users"); 38 38 + } 39 39 + 40 40 + for (const { table, record } of mergedResults) { 41 41 + if (table === "users") { 42 42 + console.log( 43 43 + `${chalk.bold.magenta(record.handle)} ${ 44 44 + record.display_name 45 45 + } ${chalk.yellow(`https://rocksky.app/profile/${record.did}`)}` 46 46 + ); 47 47 + } 48 48 + 49 49 + if (table === "albums") { 50 50 + const link = record.uri 51 51 + ? `https://rocksky.app/${record.uri?.split("at://")[1]}` 52 52 + : ""; 53 53 + console.log( 54 54 + `${chalk.bold.magenta(record.title)} ${record.artist} ${chalk.yellow( 55 55 + link 56 56 + )}` 57 57 + ); 58 58 + } 59 59 + 60 60 + if (table === "tracks") { 61 61 + const link = record.uri 62 62 + ? `https://rocksky.app/${record.uri?.split("at://")[1]}` 63 63 + : ""; 64 64 + console.log( 65 65 + `${chalk.bold.magenta(record.title)} ${record.artist} ${chalk.yellow( 66 66 + link 67 67 + )}` 68 68 + ); 69 69 + } 70 70 + } 71 71 + }

+14

src/index.ts

··· 2 2 3 3 import { nowplaying } from "cmd/nowplaying"; 4 4 import { scrobbles } from "cmd/scrobbles"; 5 5 + import { search } from "cmd/search"; 5 6 import { whoami } from "cmd/whoami"; 6 7 import { Command } from "commander"; 7 8 import version from "../package.json" assert { type: "json" }; ··· 43 44 .argument("[did]", "The DID or handle of the user to get the scrobbles for.") 44 45 .description("Display recently played tracks.") 45 46 .action(scrobbles); 47 47 + 48 48 + program 49 49 + .command("search") 50 50 + .option("-a, --albums", "Search for albums") 51 51 + .option("-t, --tracks", "Search for tracks") 52 52 + .option("-u, --users", "Search for users") 53 53 + .option("-l, --limit <number>", "Number of results to limit") 54 54 + .argument( 55 55 + "<query>", 56 56 + "The search query, e.g., artist, album, title or account" 57 57 + ) 58 58 + .description("Search for tracks, albums, or accounts.") 59 59 + .action(search); 46 60 47 61 program.parse(process.argv);