128 lines
3.3 KiB
TypeScript
128 lines
3.3 KiB
TypeScript
import { createTRPCRouter, publicProcedure } from "../utils";
|
|
import { z } from "zod";
|
|
import {
|
|
S3Client,
|
|
GetObjectCommand,
|
|
ListObjectsV2Command
|
|
} from "@aws-sdk/client-s3";
|
|
import { getSignedUrl } from "@aws-sdk/s3-request-presigner";
|
|
import { env } from "~/env/server";
|
|
import { TRPCError } from "@trpc/server";
|
|
|
|
const assets: Record<string, string> = {
|
|
lineage: "Life and Lineage.apk",
|
|
cork: "Cork.zip",
|
|
"shapes-with-abigail": "shapes-with-abigail.apk"
|
|
};
|
|
|
|
/**
|
|
* Find the most recent file in S3 matching a prefix and extension.
|
|
*/
|
|
async function getLatestFile(
|
|
client: S3Client,
|
|
bucket: string,
|
|
prefix: string,
|
|
ext: string
|
|
): Promise<string> {
|
|
try {
|
|
const listCommand = new ListObjectsV2Command({
|
|
Bucket: bucket,
|
|
Prefix: prefix,
|
|
MaxKeys: 100
|
|
});
|
|
|
|
const response = await client.send(listCommand);
|
|
|
|
if (!response.Contents || response.Contents.length === 0) {
|
|
throw new Error(`No files found in S3 with prefix ${prefix}`);
|
|
}
|
|
|
|
const files = response.Contents.filter((obj) =>
|
|
obj.Key?.endsWith(ext)
|
|
).sort((a, b) => {
|
|
const dateA = a.LastModified?.getTime() || 0;
|
|
const dateB = b.LastModified?.getTime() || 0;
|
|
return dateB - dateA;
|
|
});
|
|
|
|
if (files.length === 0) {
|
|
throw new Error(`No ${ext} files found in ${prefix} prefix`);
|
|
}
|
|
|
|
const latestFile = files[0].Key!;
|
|
console.log(`Latest file: ${latestFile}`);
|
|
return latestFile;
|
|
} catch (error) {
|
|
console.error(`Error finding latest file for ${prefix}:`, error);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Per-asset S3 lookup config for macOS apps that auto-resolve latest version.
|
|
*/
|
|
const latestAssets: Record<
|
|
string,
|
|
{ prefix: string; ext: string }
|
|
> = {
|
|
gaze: { prefix: "downloads/Gaze-", ext: ".dmg" },
|
|
inputhalo: { prefix: "downloads/InputHalo-", ext: ".dmg" },
|
|
thenook: { prefix: "downloads/TheNook-", ext: ".dmg" }
|
|
};
|
|
|
|
export const downloadsRouter = createTRPCRouter({
|
|
getDownloadUrl: publicProcedure
|
|
.input(z.object({ asset_name: z.string() }))
|
|
.query(async ({ input }) => {
|
|
const bucket = env.VITE_DOWNLOAD_BUCKET_STRING;
|
|
|
|
const credentials = {
|
|
accessKeyId: env.MY_AWS_ACCESS_KEY,
|
|
secretAccessKey: env.MY_AWS_SECRET_KEY
|
|
};
|
|
|
|
const client = new S3Client({
|
|
region: env.AWS_REGION,
|
|
credentials: credentials
|
|
});
|
|
|
|
try {
|
|
let fileKey: string;
|
|
|
|
const latest = latestAssets[input.asset_name];
|
|
if (latest) {
|
|
fileKey = await getLatestFile(client, bucket, latest.prefix, latest.ext);
|
|
} else {
|
|
fileKey = assets[input.asset_name];
|
|
|
|
if (!fileKey) {
|
|
throw new TRPCError({
|
|
code: "NOT_FOUND",
|
|
message: "Asset not found"
|
|
});
|
|
}
|
|
}
|
|
|
|
const command = new GetObjectCommand({
|
|
Bucket: bucket,
|
|
Key: fileKey
|
|
});
|
|
|
|
const signedUrl = await getSignedUrl(client, command, {
|
|
expiresIn: 120
|
|
});
|
|
|
|
return { downloadURL: signedUrl };
|
|
} catch (error) {
|
|
console.error(error);
|
|
throw new TRPCError({
|
|
code: "INTERNAL_SERVER_ERROR",
|
|
message:
|
|
error instanceof Error
|
|
? error.message
|
|
: "Failed to generate download URL"
|
|
});
|
|
}
|
|
})
|
|
});
|