platform/lib/src/commands/service.dart

import 'dart:io';
import 'package:args/command_runner.dart';
import 'package:console/console.dart';

class ServiceCommand extends Command {
  final String CUSTOM = 'Custom';
  final String MEMORY = 'In-Memory';
  final String MONGO = 'MongoDB';
  final String MONGO_TYPED = 'MongoDB (typed)';
  final String TRESTLE = 'Trestle';
  final TextPen _pen = new TextPen();

  @override
  String get name => 'service';

  @override
  String get description => 'Creates a new service within the given project.';

  @override
  run() async {
    var name = await readInput('Name of Service (not plural): ');
    var chooser = new Chooser([TRESTLE, MONGO, MONGO_TYPED, MEMORY, CUSTOM],
        message: 'What type of service would you like to create? ');
    var type = await chooser.choose();

    fail() {
      _pen.red();
      _pen('Could not successfully create service $name.');
      _pen();
    }

    String serviceSource = '';

    if (type == MONGO) {
      serviceSource = _generateMongoService(name);
      await _generateMemoryModel(name);
    } else if (type == MONGO_TYPED) {
      serviceSource = _generateMongoTypedService(name);
      await _generateMongoModel(name);
    } else if (type == MEMORY) {
      serviceSource = _generateMemoryService(name);
    } else if (type == CUSTOM) {
      serviceSource = _generateCustomService(name);
    } else if (type == TRESTLE) {
      _pen.blue();
      _pen('${Icon.STAR} Trestle services are not yet implemented. :(');
      _pen();
    } else {
      print('Code to generate a $type service is not yet written.');
    }

    if (serviceSource.isEmpty) {
      fail();
      throw new Exception('Empty generated service code.');
    }

    var servicesDir = new Directory('lib/src/services');
    var serviceFile =
        new File.fromUri(servicesDir.uri.resolve('${name.toLowerCase()}.dart'));
    var testDir = new Directory('test/services');
    var testFile = new File.fromUri(
        testDir.uri.resolve('${name.toLowerCase()}_test.dart'));

    if (!await servicesDir.exists()) await servicesDir.create(recursive: true);

    if (!await testDir.exists()) await testDir.create(recursive: true);

    await serviceFile.writeAsString(serviceSource);

    if (type == MONGO_TYPED || type == MEMORY) {
      var serviceLibrary = new File('lib/src/models/models.dart');
      await serviceLibrary.writeAsString(
          '\nexport "${name.toLowerCase()}.dart";',
          mode: FileMode.APPEND);
    }

    await testFile.writeAsString(_generateTests(name, type));

    final runConfig = new File('./.idea/runConfigurations/${name}_tests.xml');

    if (!await runConfig.exists()) {
      await runConfig.create(recursive: true);
      await runConfig.writeAsString(_generateRunConfiguration(name));
    }

    _pen.green();
    _pen('${Icon.CHECKMARK} Successfully generated service $name.');
    _pen();
  }

  _generateCustomService(String name) {
    return '''
import 'package:angel_framework/angel_framework.dart';

class ${name}Service extends Service {
  ${name}Service():super() {
    // Your logic here!
  }
}
    '''
        .trim();
  }

  _generateMemoryModel(String name) async {
    final lower = name.toLowerCase();
    final file = new File('lib/src/models/$lower.dart');

    if (!await file.exists()) await file.createSync(recursive: true);

    await file.writeAsString('''
library angel.models.$lower;

import 'dart:convert';
import 'package:angel_framework/angel_framework.dart';

class $name extends MemoryModel {
  String name, desc;

  $name({String id, this.name, this.desc}) {
    this.id = id;
  }

  factory $name.fromJson(String json) => new $name.fromMap(JSON.decode(json));

  factory $name.fromMap(Map data) => new $name(
      id: data['id']
      name: data['name'],
      desc: data['desc']);

  Map toJson() {
    return {
      'id': id,
      'name': name,
      'desc': desc
    };
  }
}
    '''
        .trim());
  }

  _generateMemoryService(String name) {
    return '''
import 'package:angel_framework/defs.dart';
import 'package:angel_framework/angel_framework.dart';

/// Store in-memory instances of this class.
class $name extends MemoryModel {
  String name, desc;

  $name({String id, this.name, this.desc}) {
    this.id = id;
  }

  factory $name.fromJson(String json) => new $name.fromMap(JSON.decode(json));

  factory $name.fromMap(Map data) => new $name(
      id: data['id']
      name: data['name'],
      desc: data['desc']);

  Map toJson() {
    return {
      'id': id,
      'name': name,
      'desc': desc
    };
  }
}

/// Manages [$name] in-memory.
class ${name}Service extends MemoryService<$name> {
  ${name}Service():super() {
    // Your logic here!
  }
}
    '''
        .trim();
  }

  _generateMongoModel(String name) async {
    final lower = name.toLowerCase();
    final file = new File('lib/src/models/$lower.dart');

    if (!await file.exists()) await file.createSync(recursive: true);

    await file.writeAsString('''
library angel.models.$lower;

import 'dart:convert';
import 'package:angel_mongo/model.dart';

class $name extends Model {
  String name, desc;

  $name({String id, this.name, this.desc}) {
    this.id = id;
  }

  factory $name.fromJson(String json) => new $name.fromMap(JSON.decode(json));

  factory $name.fromMap(Map data) => new $name(
      id: data['id']
      name: data['name'],
      desc: data['desc']);

  Map toJson() {
    return {
      'id': id,
      'name': name,
      'desc': desc
    };
  }
}
    '''
        .trim());
  }

  _generateMongoService(String name) {
    final lower = name.toLowerCase();

    return '''
import 'package:angel_framework/angel_framework.dart';
import 'package:angel_mongo/angel_mongo.dart';
import 'package:mongo_dart/mongo_dart.dart';

configureServer(Db db) {
  return (Angel app) async {
    app.use('/api/${lower}s', new ${name}Service(db.collection('${lower}s')));

    HookedService service = app.service('api/${lower}s');
    app.container.singleton(service.inner);
  };
}

/// Manages [$name] in the database.
class ${name}Service extends MongoService {
  ${name}Service(collection):super(collection) {
    // Your logic here!
  }
}
    '''
        .trim();
  }

  _generateMongoTypedService(String name) {
    final lower = name.toLowerCase();

    return '''
import 'package:angel_framework/angel_framework.dart';
import 'package:angel_mongo/angel_mongo.dart';
import 'package:mongo_dart/mongo_dart.dart';
import '../models/$lower.dart';
export '../models/$lower.dart';

configureServer(Db db) {
  return (Angel app) async {
    app.use('/api/${lower}s', new ${name}Service(db.collection('${lower}s')));

    HookedService service = app.service('api/${lower}s');
    app.container.singleton(service.inner);
  };
}

/// Manages [$name] in the database.
class ${name}Service extends MongoTypedService<$name> {
  ${name}Service(collection):super(collection) {
    // Your logic here!
  }
}
    '''
        .trim();
  }

  _generateRunConfiguration(String name) {
    final lower = name.toLowerCase();

    return '''
    <component name="ProjectRunConfigurationManager">
      <configuration default="false" name="$name Tests" type="DartTestRunConfigurationType" factoryName="Dart Test" singleton="true">
        <option name="filePath" value="\$PROJECT_DIR\$/test/services/${lower}_test.dart" />
        <method />
      </configuration>
    </component>
'''
        .trim();
  }

  _generateTests(String name, String type) {
    return '''
import 'dart:io';
import 'package:angel/angel.dart';
import 'package:angel_framework/angel_framework.dart';
import 'package:angel_test/angel_test.dart';
import 'package:test/test.dart';

main() async {
  Angel app;
  TestClient client;

  setUp(() async {
    app = await createServer();
    client = await connectTo(app, saveSession: false);
  });

  tearDown(() async {
    await client.close();
    app = null;
  });

  test('index via REST', () async {
    final response = await client.get('/api/${name.toLowerCase()}');
    expect(response, hasStatus(HttpStatus.OK));
  });

  test('Index ${name.toLowerCase()}s', () async {
    final ${name.toLowerCase()}s = await client.service('api/${name.toLowerCase()}').index();
    print(${name.toLowerCase()}s);
  });
}

    '''
        .trim();
  }
}