If you are looking to recreate what was original it's easy. You need 2 bearings which are 1 3/8x5/8, you need both sprockets and jack shaft material. All can be found at OldMiniBikes. If you need help with cutting the jack shaft to length, you can give me the dimension you want and I can do that for you and ship it out. I think one of the bearing companies around here sells jack shaft material. I can drill and thread it on the ends for you also should you desire.
Take your picture and count as best you can the number of teeth on the sprockets. I did a couple of Doodle bugs (same frame as the Trail Bug) with jack shafts and I just kept trying different combinations of sprockets until it would pull my fat carcass through the grass lol. What you have there is real easy to recreate and I personally think well worth it.
Doc