import fs from "node:fs";
import beamcoder from "beamcoder";

//let outFile = fs.createWriteStream("a.h264");
let endcode = Buffer.from([0, 0, 1, 0xb7]);

async function imageToVideo(imagePath, duration, frameRate = 30) {
  // Create a demuxer for the JPEG image
  let demuxer = await beamcoder.demuxer(imagePath);

  // Read the image packet
  let packet = await demuxer.read();

  // Create a decoder for the image
  let decoder = beamcoder.decoder({ demuxer: demuxer, name: "mjpeg" });

  // Decode the image to get the frame
  let frames = await decoder.decode(packet);
  // decoder.flush();
  //console.log("frames", frames);
  let frame = frames.frames[0];
  //console.log("frame", frame.width, frame.height);

  // Create an H.264 encoder
  let encoder = beamcoder.encoder({
    name: "libx264",
    width: frame.width,
    height: frame.height,
    bit_rate: 500000,
    time_base: [1, frameRate],
    framerate: [frameRate, 1],
    pix_fmt: "yuv420p",
    preset: "faster",
    gop_size: 10,
    max_b_frames: 1,
  });

  let stream = beamcoder.muxerStream({});
  stream.pipe(fs.createWriteStream("test.mp4"));
  //  Create a muxer for the output video
  let muxer = stream.muxer({ format_name: "mpegts" }); //"mp4" });

  //console.log(demuxer.streams[0].codecpar.extradata); // null
  let vstr = muxer.newStream({
    name: "ts", //"h264",
    time_base: [1, 90000], //frameRate],
    interleaved: true,
    codecpar: {
      width: encoder.width,
      height: encoder.height,
      format: encoder.pix_fmt,
      //extradata: demuxer.streams[0].codecpar.extradata,
    },
  });
  //await muxer.openIO();
  //await muxer.initOutput({ movflags: "frag_keyframe+empty_moov+default_base_moof" });
  console.log("inited");
  // Add a video stream to the muxer
  await muxer.writeHeader();
  console.log("header written");
  // Number of frames to encode
  let totalFrames = duration * frameRate;

  console.log("making frames", totalFrames);
  for (let i = 0; i < totalFrames; i++) {
    // Encode the frame
    frame.pts = i;
    let encodedPackets = await encoder.encode(frame);
    // console.log(encodedPackets);

    // Write the encoded packets to the output file
    for (let packet of encodedPackets.packets) {
      packet.duration = 1;
      packet.stream_index = vstr.index;
      packet.pts = (packet.pts * 90000) / frameRate;
      packet.dts = (packet.dts * 90000) / frameRate;
      //packet.pts = i;
      await muxer.writeFrame(packet);
      // outFile.write(packet.data);
    }
  }

  // Finalize the encoder and muxer
  let encodedPackets = await encoder.flush();
  // after flushing the encoder, we may hve some more packets

  // Write the encoded packets to the output file
  for (let packet of encodedPackets.packets) {
    packet.duration = 1;
    packet.stream_index = vstr.index;
    packet.pts = (packet.pts * 90000) / frameRate;
    packet.dts = (packet.dts * 90000) / frameRate;
    await muxer.writeFrame(packet);
  }
  //outFile.end(endcode);
  await muxer.writeTrailer();
}

// Usage example
imageToVideo("captures/0010.jpg", 20)
  .then(() => {
    console.log("Video created successfully");
  })
  .catch(console.error);