import deprecated from "discourse-common/lib/deprecated";
import { getOwner as emberGetOwner, setOwner } from "@ember/application";

let _default = {};

export function getOwner(obj) {
  if (emberGetOwner) {
    return emberGetOwner(obj) || emberGetOwner(_default);
  }

  return obj.container;
}

export function setDefaultOwner(container) {
  setOwner(_default, container);
}

// `this.container` is deprecated, but we can still build a container-like
// object for components to use
export function getRegister(obj) {
  const owner = getOwner(obj);
  const register = {
    lookup: (...args) => owner.lookup(...args),
    lookupFactory: (...args) => {
      if (owner.factoryFor) {
        return owner.factoryFor(...args);
      } else if (owner._lookupFactory) {
        return owner._lookupFactory(...args);
      }
    },

    deprecateContainer(target) {
      Object.defineProperty(target, "container", {
        get() {
          deprecated(
            "Use `this.register` or `getOwner` instead of `this.container`"
          );
          return register;
        }
      });
    }
  };

  return register;
}