#ifndef shell_helper_functions
#define shell_helper_functions

#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

void *malloce(size_t size){
    void* mal=malloc(size);
    if(!mal){
        fprintf(stderr, "lsh: allocation error\n");
        exit(EXIT_FAILURE);
    }
    return mal;
}

char *pop(char *token, int ch_i){
    if(token[ch_i]=='\0'){
        return token;
    }

    char *right=(char*)malloce(sizeof(char)*strlen(token)+1);

    strcpy(right,&token[ch_i+1]);
    token[ch_i]='\0';
    strcat(token,right);

    free(right);
    return token;
}


// You must free the result if result is non-NULL.
char *str_replace(char *orig, char *rep, char *with) {
    char *result; // the return string
    char *ins;    // the next insert point
    char *tmp;    // varies
    int len_rep;  // length of rep (the string to remove)
    int len_with; // length of with (the string to replace rep with)
    int len_front; // distance between rep and end of last rep
    int count;    // number of replacements

    // sanity checks and initialization
    if (!orig || !rep)
        return NULL;
    len_rep = strlen(rep);
    if (len_rep == 0)
        return NULL; // empty rep causes infinite loop during count
    if (!with)
        with = "";
    len_with = strlen(with);

    // count the number of replacements needed
    ins = orig;
    for (count = 0; (tmp = strstr(ins, rep)); ++count) {
        ins = tmp + len_rep;
    }

    tmp = result = malloc(strlen(orig) + (len_with - len_rep) * count + 1);

    if (!result)
        return NULL;

    // first time through the loop, all the variable are set correctly
    // from here on,
    //    tmp points to the end of the result string
    //    ins points to the next occurrence of rep in orig
    //    orig points to the remainder of orig after "end of rep"
    while (count--) {
        ins = strstr(orig, rep);
        len_front = ins - orig;
        tmp = strncpy(tmp, orig, len_front) + len_front;
        tmp = strcpy(tmp, with) + len_with;
        orig += len_front + len_rep; // move to next "end of rep"
    }
    strcpy(tmp, orig);
    return result;
}

char *basename(char *path_in) 
{
    char *token = NULL;
    char *lastToken = NULL;
    char *basePath;

    basePath = (char *)malloce(sizeof(char) * (strlen(path_in) + 1));
    strcpy(basePath, path_in);

    token = strtok(basePath, "/");
    while (token != NULL) {
        lastToken = token;
        token = strtok(NULL, "/");
    }

    if (lastToken == NULL){
        free(basePath);
        return NULL;
    }

    char *ret_str = (char*)malloce(sizeof(char)*strlen(lastToken)+1);
    strcpy(ret_str,lastToken);
    free(basePath);
    return ret_str;
}

#endif