disk-wrap

local driveSide
for _, side in ipairs(rs.getSides()) do
  if peripheral.isPresent(side) and peripheral.getType(side) == "drive" then
    driveSide = side
    break
  end
end

if driveSide then
  -- do stuff with the disk drive
else
  error("No disk drive found.")
end