#include "h5dsm_example.h"

int od_test_g;

herr_t iter_cb(hid_t loc_id, const char *link_name, const H5L_info_t *linfo,
        void *op_data) {
    /* Print name, type, and address or value size */
    printf("%s: ", link_name);
    if(linfo->type == H5L_TYPE_HARD)
        printf("hard, address = 0x%016llx\n", (unsigned long long)linfo->u.address);
    else if(linfo->type == H5L_TYPE_SOFT)
        printf("soft, val_size = %llu\n", (unsigned long long)linfo->u.val_size);
    else
        PRINTF_ERROR("invalid link type");

    /* Check op_data */
    if(op_data != &od_test_g)
        PRINTF_ERROR("op_data incorrect");

    return 0;

error:
    return -1;
}

int main(int argc, char *argv[]) {
    uuid_t pool_uuid;
    char *pool_grp = NULL;
    hid_t file = -1, fapl = -1;
    hsize_t num_link = 0;
    herr_t ret;
    H5VL_daosm_snap_id_t snap_id;

    (void)MPI_Init(&argc, &argv);

    if(argc < 4 || argc > 5)
        PRINTF_ERROR("argc must be 4 or 5\n");

    /* Parse UUID */
    if(0 != uuid_parse(argv[1], pool_uuid))
        ERROR;

    /* Initialize VOL */
    if(H5VLdaosm_init(MPI_COMM_WORLD, pool_uuid, pool_grp) < 0)
        ERROR;

    /* Set up FAPL */
    if((fapl = H5Pcreate(H5P_FILE_ACCESS)) < 0)
        ERROR;
    if(H5Pset_fapl_daosm(fapl, MPI_COMM_WORLD, MPI_INFO_NULL) < 0)
        ERROR;
    if(H5Pset_all_coll_metadata_ops(fapl, true) < 0)
        ERROR;

    /* Open snapshot if specified */
    if(argc == 5) {
        snap_id = (H5VL_daosm_snap_id_t)atoi(argv[4]);
        printf("Opening snapshot %llu\n", (long long unsigned)snap_id);
        if(H5Pset_daosm_snap_open(fapl, snap_id) < 0)
            ERROR;
    } /* end if */

    /* Open file */
    if((file = H5Fopen(argv[2], H5F_ACC_RDONLY, fapl)) < 0)
        ERROR;

    printf("Iterating over links\n");

    /* Iterate */
    if((ret = H5Literate_by_name(file, argv[3], H5_INDEX_NAME, H5_ITER_NATIVE, &num_link, iter_cb, &od_test_g, H5P_DEFAULT)) < 0)
        ERROR;

    printf("Complete.  Number of links: %d\n", (int)num_link);

    /* Close */
    if(H5Fclose(file) < 0)
        ERROR;
    if(H5Pclose(fapl) < 0)
        ERROR;

    printf("Success\n");

    (void)MPI_Finalize();
    return 0;

error:
    H5E_BEGIN_TRY {
        H5Fclose(file);
        H5Pclose(fapl);
    } H5E_END_TRY;

    (void)MPI_Finalize();
    return 1;
}