sol-mem-gen/src/utils/uploadToPinata.ts

import { PinataSDK } from 'pinata-web3';
import 'dotenv/config';
import assert from 'assert';

import { FluxGenerationResult } from '../services/fluxService';

assert(process.env.PINATA_JWT, "PINATA_JWT is required");
assert(process.env.PINATA_GATEWAY, "PINATA_GATEWAY is required");

const pinata = new PinataSDK({
  pinataJwt: process.env.PINATA_JWT,
  pinataGateway: process.env.PINATA_GATEWAY,
});

export async function uploadToPinata(imageUrl: string): Promise<FluxGenerationResult> {
  try {
    const upload = await pinata.upload.url(imageUrl);

    const publicURL = await pinata.gateways.convert(upload.IpfsHash);

    return { imageUrl: publicURL };
  } catch (error) {
    console.error('Error uploading to Pinata:', error)
    return {
      error: error instanceof Error ? error.message : 'Upload failed'
    };
  }
}