'use strict';
const path = require('path');
const assert = require('assert');
const fs = require('fs');

module.exports = {
  name: require('./package').name,
  included() {
    const app = this._findHost();
    const addonConfig = this.app.project.config(app.env)['orbit'] || {};
    const collections = addonConfig.collections || {};
    let modelsPath;

    if (collections.models) {
      modelsPath = path.join('app', collections.models);
    } else if (
      app.project.pkg['ember-addon'] &&
      app.project.pkg['ember-addon'].configPath
    ) {
      modelsPath = path.join('tests', 'dummy', 'app', 'data-models');
    } else {
      modelsPath = path.join('app', 'data-models');
    }

    const modelsDirectory = path.join(app.project.root, modelsPath);

    assert(
      fs.existsSync(modelsDirectory),
      `You need to create models directory: "${modelsDirectory}"`
    );

    this._super.included.apply(this, arguments);
  }
};

@@ -337,18 +337,20 @@
     
-le
+cons
 t models
 Path
@@ -349,23 +349,14 @@
 dels
-Path;%0A%0A    if (
+Dir = 
 coll
@@ -373,79 +373,51 @@
 dels
-) %7B%0A      modelsPath = path.join('app', collections.
+ %7C%7C 'data-models';%0A    let 
 models
-);%0A 
+Path;%0A%0A
    
-%7D else
  if 
@@ -564,37 +564,33 @@
 my', 'app', 
-'data-
 models
-'
+Dir
 );%0A    %7D els
@@ -629,29 +629,25 @@
 ('app', 
-'data-
 models
-'
+Dir
 );%0A    %7D
@@ -779,25 +779,24 @@
    %60
-You need to creat
+%5Bember-orbit%5D Th
 e mo
@@ -809,16 +809,27 @@
 irectory
+ is missing
 : %22$%7Bmod
@@ -842,16 +842,104 @@
 ectory%7D%22
+. You can run 'ember g ember-orbit' to initialize ember-orbit and create this directory.
 %60%0A    );

Improve initial check for `data-models` dir