Iterate Over Typescript Enums

iterate_enums.ts

// Credit due: https://github.com/microsoft/TypeScript/issues/4753#issuecomment-694557208

function enumValues<T extends string>(enumObj: {
  [key: string]: T;
}): IterableIterator<T>;
function enumValues<T extends string | number>(enumObj: {
  [key: string]: T;
}): IterableIterator<Exclude<T, string>>;
function* enumValues<T>(enumObj: { [key: string]: T }): IterableIterator<T> {
  let isStringEnum = true;
  for (const property in enumObj) {
    if (typeof enumObj[property] === "number") {
      isStringEnum = false;
      break;
    }
  }
  for (const property in enumObj) {
    if (isStringEnum || typeof enumObj[property] === "number") {
      yield enumObj[property];
    }
  }
}

Usage: for (const tileColor of enumValues(TileColor))