mt_tool 0.1.3 → 0.1.5

data/lib/mt_tool/module/module.rb

@@ -0,0 +1,275 @@
+
+require 'colored'
+require 'fileutils'
+require 'psych'
+require 'yaml'
+require 'thor/actions'
+require 'colored2'
+require 'fileutils'
+require 'psych'
+
+
+module MtTool
+
+  CONFIG_FILE = '.MTModuleFilesConfig.yml'.freeze
+
+  class Module < Thor
+    include Thor::Actions
+    no_commands do
+
+      def initialize(args = [], options = {}, config = {})
+        super
+
+      end
+
+      def generate(path = nil, name, lang, class_prefix, author)
+
+        @name = name
+        @module = @name
+        @lang = lang
+        @class_prefix = class_prefix
+        @final_path = "#{path}/#{@name}"
+        @author = author
+        @prefixed_module = @class_prefix + @module
+
+        say "generating file in path:#{@final_path}", :green
+
+        if File.exist?(@final_path.to_s)
+          say "#{@final_path} 已存在:", :red
+        else
+          prepare_folder
+          if File.exist?("#{@final_path}/configure")
+            system("#{@final_path}/configure", @name, @lang, @class_prefix, *@additional_args)
+          else
+            say 'Template does not have a configure file', :red
+          end
+          yk_module_folders
+          yk_template_files
+        end
+      end
+
+      def create(path = nil)
+
+        path = Dir.pwd if path.nil?
+        say '模块名:', :green
+        config_file_path = "#{path}/#{CONFIG_FILE}"
+        config = File.exist?(config_file_path) ? YAML.load_file(config_file_path) : {}
+        input_name = ask("Project name [#{config[:project]}] ?")
+
+        if input_name != ''
+          @name = input_name
+          config[:project] = input_name if input_name != config[:project]
+        else
+          @name = config[:project]
+        end
+
+        File.open(config_file_path, 'w') do |f|
+          f.write config.to_yaml
+        end
+
+        @final_path = "#{path}/#{@name}"
+
+        if File.exist?(@final_path.to_s)
+          say "#{@final_path} 已存在:", :red
+        else
+          prepare_folder
+          read_config(path)
+
+          if File.exist?("#{@final_path}/configure")
+            system("#{@final_path}/configure", @name, @lang, @class_prefix, *@additional_args)
+          else
+            say 'Template does not have a configure file', :red
+          end
+
+          yk_module_folders
+          yk_template_files
+
+        end
+      end
+
+
+      def read_config(path)
+        config_file_path = "#{path}/#{CONFIG_FILE}"
+        config = File.exist?(config_file_path) ? YAML.load_file(config_file_path) : {}
+
+        project = @name
+        say '语言:', :green
+        language = ask("Project language [#{config[:language]}] ?", limited_to: ['objc', 'swift', ''])
+        say '类名前缀:', :green
+        class_prefix = ask("Class prefix [#{config[:class_prefix]}] ?")
+        say '文件作者:', :green
+        author = ask("Author [#{config[:author]}] ?")
+
+        config[:project] = project.empty? ? config[:project] || '' : project
+        config[:language] = language.empty? ? config[:language] || 'objc' : language
+        config[:class_prefix] = class_prefix.empty? ? config[:class_prefix] || '' : class_prefix
+        config[:author] = author.empty? ? config[:author] || '' : author
+
+        File.open(config_file_path, 'w') do |f|
+          f.write config.to_yaml
+          # f.write YAML.to_yaml(config)
+        end
+
+        @module = @name
+        @class_prefix = config[:class_prefix]
+        @prefixed_module = config[:class_prefix] + @module
+        @project = config[:project]
+        @author = config[:author]
+        @date = Time.now.strftime('%d/%m/%y')
+        @lang = config[:language]
+      end
+      def prepare_folder
+        host_a = 'yeah'
+        host_b = 'ka'
+        template_repo_url = "http://gitlab.#{host_a}#{host_b}.com/App/iOS/YKComponents/YKProjectTemplate.git"
+        system("git clone #{template_repo_url} #{@final_path}")
+
+        # FileUtils.remove_dir(@final_path, true)
+        # FileUtils.cp_r('/Users/imacn24/Documents/dev/YKProjectTemplate', @final_path)
+        # FileUtils.remove_dir("#{@final_path}/.git", true)
+      end
+
+      def create_viper_module(path = nil, name, lang, class_prefix, author)
+        @name = name
+        @module = @name
+        @lang = lang
+        @class_prefix = class_prefix
+        @final_path = "#{path}/#{@name}"
+        @author = author
+        @prefixed_module = @class_prefix + @module
+
+        say "generating file in path:#{@final_path}", :green
+
+        if File.exist?(@final_path.to_s)
+          say "#{@final_path} 已存在:", :red
+        else
+          mt_viper_module
+        end
+      end
+
+
+      def mt_viper_module
+        class_folder_path = "#{@final_path}"
+        class_folder_files= {
+          'Entity.swift' => 'Entity',
+          'Interactor.swift' => 'Entity',
+          'Presenter.swift' => 'Entity',
+          'Router.swift' => 'Entity',
+          'ViewController.swift' => 'Entity',
+        }
+
+        class_folder_files.each do |file_name, _folder|
+          final_file = "#{class_folder_path}/#{@prefixed_module}#{file_name}"
+          template "#{__dir__}/template/swift/#{file_name}", final_file
+        end
+
+      end
+
+
+      def yk_module_folders
+        class_folder_path = "#{@final_path}/#{@name}/Classes"
+
+        first_level_folders = %w[Public Private]
+
+        # public_level_folders = ['Register']
+        # public_level_folders.each do |folder|
+        #   path = "#{class_folder_path}/Public/#{folder}"
+        #   empty_directory path
+        # end
+
+        private_level_folders = %w[Business Category Vendor Tools]
+
+        first_level_folders.each do |folder|
+          path = "#{class_folder_path}/#{folder}"
+          empty_directory path
+        end
+
+        private_level_folders.each do |folder|
+          path = "#{class_folder_path}/Private/#{folder}"
+          empty_directory path
+        end
+      end
+
+
+      MtTool::Module.source_root(File.dirname(__FILE__))
+      def yk_template_files
+
+        register_path = "#{@final_path}/#{@name}/Classes/Private/Register"
+        registger = {
+          'RouterRegister.h' => 'RouterRegister',
+          'RouterRegister.m' => 'RouterRegister',
+          'ServiceRegister.h' => 'ServiceRegister',
+          'ServiceRegister.m' => 'ServiceRegister'
+        }
+
+        registger.each do |file_name, _folder|
+          final_file = "#{register_path}/#{@prefixed_module}#{file_name}"
+          template "#{__dir__}/template/objc/#{file_name}", final_file
+        end
+
+        public_folder_path = "#{@final_path}/#{@name}/Classes/Public"
+
+        template_code_filename = ['ServiceProtocol.h', 'RouterDefine.h']
+        template_code_filename.each do |file_name|
+          final_file = "#{public_folder_path}/#{@prefixed_module}#{file_name}"
+          source = "#{__dir__}/template/objc/#{file_name}"
+          template source, final_file
+        end
+
+        swift_template_code_filename = ['RouterDefine.swift']
+        swift_template_code_filename.each do |file_name|
+          final_file = "#{public_folder_path}/#{@prefixed_module}_Swift_#{file_name}"
+          source = "#{__dir__}/template/swift/#{file_name}"
+          template source, final_file
+        end
+
+        private_folder_path = "#{@final_path}/#{@name}/Classes/Private"
+        #pch file
+        # pch_file_name = "PrefixHeader.pch"
+        # final_file = "#{private_folder_path}/#{@prefixed_module}#{pch_file_name}"
+        # source = "#{__dir__}/template/objc/#{pch_file_name}"
+        # template source, final_file
+
+        private_level_folder_files = {
+          'PrefixHeader.pch' => 'Business',
+          'CategoryHeader.h' => 'Category',
+          'ToolsHeader.h' => 'Tools',
+          'vendorHeader.h' => 'Vendor'
+        }
+
+        private_level_folder_files.each do |file_name, folder|
+          final_prefix = @prefixed_module
+          if file_name == 'PrefixHeader.pch'
+            final_prefix = @module
+          end
+
+          final_file = "#{private_folder_path}/#{folder}/#{final_prefix}#{file_name}"
+          source = "#{__dir__}/template/objc/#{file_name}"
+          template source, final_file
+        end
+
+        tools_files_path = "#{@final_path}/#{@name}/Classes/Private/Tools"
+
+        tool_files = ['Bundle.h','Bundle.m']
+        tool_files.each do |file_name|
+          final_file = "#{tools_files_path}/#{@prefixed_module}#{file_name}"
+          source = "#{__dir__}/template/objc/#{file_name}"
+          template source, final_file
+        end
+
+        business_demo_path = "#{@final_path}/#{@name}/Classes/Private/Business"
+        demo_replace_file = ['DemoViewController.h', 'DemoViewController.m', 'DemoViewModel.h', 'DemoViewModel.m']
+        demo_replace_file.each do |file_name|
+          final_file = "#{business_demo_path}/Demo/#{@prefixed_module}#{file_name}"
+          source = "#{__dir__}/template/objc/demo/#{file_name}"
+          template source, final_file
+        end
+
+        Dir.chdir("#{@final_path}/Example") do
+          system 'pod install'
+          system "open './#{@name}.xcworkspace'"
+        end
+      end
+    end
+  end
+end

data/lib/mt_tool/module/template/swift/Entity.swift

@@ -0,0 +1,15 @@
+//
+//  <%= @prefixed_module %>Entity.swift
+//  <%= @project %>
+//
+//  Created by <%= @author %> on <%= @date %>.
+//
+//
+
+
+
+import Foundation
+
+struct <%= @prefixed_module %>Entity {
+
+}

data/lib/mt_tool/module/template/swift/Interactor.swift

@@ -0,0 +1,16 @@
+//
+//  <%= @prefixed_module %>Interactor.swift
+//  <%= @project %>
+//
+//  Created by <%= @author %> on <%= @date %>.
+//
+//
+
+import Foundation
+
+protocol <%= @prefixed_module %>InteractorProtocol {
+}
+
+class <%= @prefixed_module %>Interactor: <%= @prefixed_module %>InteractorProtocol {
+    weak var presenter: <%= @prefixed_module %>PresenterProtocol?
+}

data/lib/mt_tool/module/template/swift/Presenter.swift

@@ -0,0 +1,28 @@
+//
+//  <%= @prefixed_module %>Presenter.swift
+//  <%= @project %>
+//
+//  Created by <%= @author %> on <%= @date %>.
+//
+//
+
+import Foundation
+
+protocol <%= @prefixed_module %>PresenterProtocol: AnyObject {
+    func notifySignInTapped()
+    func notifySignUpTapped()
+}
+
+class <%= @prefixed_module %>Presenter: <%= @prefixed_module %>PresenterProtocol {
+    weak var view: <%= @prefixed_module %>ViewProtocol?
+    var router: <%= @prefixed_module %>RouterProtocol?
+    var interactor: <%= @prefixed_module %>InteractorProtocol?
+    
+    func notifySignInTapped() {
+        router?.routeToSignIn(view as! <%= @prefixed_module %>ViewController)
+    }
+    
+    func notifySignUpTapped() {
+        router?.routeToSignUp(view as! <%= @prefixed_module %>ViewController)
+    }
+}

data/lib/mt_tool/module/template/swift/Router.swift

@@ -0,0 +1,50 @@
+//
+//  <%= @prefixed_module %>Router.swift
+//  <%= @project %>
+//
+//  Created by <%= @author %> on <%= @date %>.
+//
+//
+
+import UIKit
+
+typealias <%= @prefixed_module %>Entry = <%= @prefixed_module %>ViewProtocol & UIViewController
+
+protocol <%= @prefixed_module %>RouterProtocol {
+    var entry: <%= @prefixed_module %>Entry? { get set }
+    func routeToSignIn(_ view: <%= @prefixed_module %>ViewProtocol)
+    func routeToSignUp(_ view: <%= @prefixed_module %>ViewProtocol)
+}
+
+class <%= @prefixed_module %>Router: <%= @prefixed_module %>RouterProtocol {
+    
+    var entry: <%= @prefixed_module %>Entry?
+    
+    static func createModule() -> <%= @prefixed_module %>RouterProtocol {
+        let view = <%= @prefixed_module %>ViewController()
+        let interactor = <%= @prefixed_module %>Interactor()
+        let presenter = <%= @prefixed_module %>Presenter()
+        let router = <%= @prefixed_module %>Router()
+        
+        view.presenter = presenter
+        interactor.presenter = presenter
+        presenter.view = view
+        presenter.router = router
+        presenter.interactor = interactor
+        router.entry = view
+        
+        return router
+    }
+    
+    func routeToSignIn(_ view: <%= @prefixed_module %>ViewProtocol) {
+        let signInVC = SignInRouter.createModule()
+        guard let view = view as? UIViewController else { return }
+        view.navigationController?.pushViewController(signInVC, animated: true)
+    }
+    
+    func routeToSignUp(_ view: <%= @prefixed_module %>ViewProtocol) {
+        let signUpVC = SignUpRouter.createModule()
+        guard let view = view as? UIViewController else { return }
+        view.navigationController?.pushViewController(signUpVC, animated: true)
+    }
+}

data/lib/mt_tool/module/template/swift/ViewController.swift

@@ -0,0 +1,97 @@
+//
+//  <%= @prefixed_module %>ViewController.swift
+//  <%= @project %>
+//
+//  Created by <%= @author %> on <%= @date %>.
+//
+//
+
+import UIKit
+protocol <%= @prefixed_module %>ViewProtocol: AnyObject {
+    
+}
+
+class <%= @prefixed_module %>ViewController: UIViewController, <%= @prefixed_module %>ViewProtocol {
+
+    var presenter: <%= @prefixed_module %>PresenterProtocol?
+
+    lazy var button1: UIButton = {
+            let button = UIButton(type: .system)
+            button.setTitle("button 1", for: .normal)
+            button.addTarget(self, action: #selector(sampleButton1Tapped), for: .touchUpInside)
+            return button
+        }()
+
+        lazy var button2: UIButton = {
+            let button = UIButton(type: .system)
+            button.setTitle("button 2", for: .normal)
+            button.addTarget(self, action: #selector(sampleButton2Tapped), for: .touchUpInside)
+            return button
+        }()
+
+        //MARK: - View LifeCycle
+        override func viewDidLoad() {
+            super.viewDidLoad()
+            configureUI()
+            navigationItem.title = "Sample ViewController"
+        }
+
+        override func viewWillAppear(_ animated: Bool) {
+            super.viewWillAppear(animated)
+
+        }
+
+        override func viewWillDisappear(_ animated: Bool) {
+            super.viewWillDisappear(animated)
+
+        }
+
+        override func viewDidLayoutSubviews() {
+            super.viewDidLayoutSubviews()
+
+        }
+
+        //MARK: - View Configurations
+        private func configureUI() {
+            view.backgroundColor = .white
+            embedViewController(containerView: view, controller: PeaceCometsViewController(), previous: nil)
+
+            view.addSubview(button1)
+            view.addSubview(button2)
+
+            button1.backgroundColor = UIColor.purple.withAlphaComponent(0.5)
+            button2.backgroundColor = UIColor.red.withAlphaComponent(0.5)
+
+            setupConstraints()
+        }
+
+
+        func setupConstraints() {
+            button1.snp.makeConstraints { make in
+                make.centerX.equalToSuperview()
+                make.top.equalTo(view.safeAreaLayoutGuide.snp.top).offset(100)
+                make.width.equalTo(200)
+                make.height.equalTo(50)
+            }
+
+            button2.snp.makeConstraints { make in
+                make.centerX.equalToSuperview()
+                make.top.equalTo(button1.snp.bottom).offset(20)
+                make.width.equalTo(200)
+                make.height.equalTo(50)
+            }
+        }
+
+
+        @objc func sampleButton1Tapped() {
+            print("sample button 1 tapped")
+
+        }
+
+
+        @objc func sampleButton2Tapped() {
+            print("sample button 2 tapped")
+
+        }
+}
+

data/lib/mt_tool/oc_model/main.rb

@@ -0,0 +1,112 @@
+require 'json'
+
+$class_full_name = ARGV[0]
+input_file_path = ARGV[1]
+
+input_file = File.read(input_file_path)
+input_json = JSON.parse(input_file)
+
+input_dir = File.dirname(input_file_path)
+
+# constants
+OBJC_TYPE_STRING = "NSString *"
+OBJC_TYPE_NUMBER = "NSNumber *"
+OBJC_TYPE_NSDIC = "NSDictionary *"
+OBJC_TYPE_ID = "id"
+OBJC_TYPE_ARRAY = "NSArray *"
+OBJC_TYPE_BOOL = "BOOL"
+OBJC_TYPE_NULL = "NSNull *"
+
+# attributes
+ATTRIBUTE_NONATOMIC = "nonatomic"
+ATTRIBUTE_COPY = "copy"
+$class_full_name
+
+$type_map = {
+  "String" => OBJC_TYPE_STRING,
+  "Integer" => OBJC_TYPE_NUMBER,
+  "Float" => OBJC_TYPE_NUMBER,
+  "Hash" => OBJC_TYPE_NSDIC,
+  #"Hash" => OBJC_TYPE_ID,
+  "Array" => OBJC_TYPE_ARRAY,
+  "TrueClass" => OBJC_TYPE_BOOL,
+  "FalseClass" => OBJC_TYPE_BOOL,
+  "NilClass" => OBJC_TYPE_NULL,
+}
+
+$attr_map = {
+  OBJC_TYPE_STRING => [ATTRIBUTE_NONATOMIC, ATTRIBUTE_COPY],
+  OBJC_TYPE_NUMBER => [ATTRIBUTE_NONATOMIC],
+  OBJC_TYPE_ID => [ATTRIBUTE_NONATOMIC],
+  OBJC_TYPE_NSDIC => [ATTRIBUTE_NONATOMIC],
+  OBJC_TYPE_ARRAY => [ATTRIBUTE_NONATOMIC, ATTRIBUTE_COPY],
+  OBJC_TYPE_BOOL => [ATTRIBUTE_NONATOMIC],
+  OBJC_TYPE_NULL => [ATTRIBUTE_NONATOMIC]
+}
+
+# generate info
+$prop_declare = "@property"
+
+def attr_name_for_attr_array(attr_array)
+  attr_name = ""
+
+  attr_array.each_with_index do |attribute, index|
+    if index != 0
+      attr_name << ", #{attribute}"
+    else
+      attr_name << "#{attribute}"
+    end
+
+  end
+
+  return attr_name
+end
+
+def prop_for_key_value(key, value)
+  type = $type_map["#{value.class}"]
+  attribute = attr_name_for_attr_array($attr_map[type])
+  var_name = key
+  prop_ret = "#{$prop_declare} (#{attribute}) #{type}#{var_name};"
+
+  return prop_ret
+end
+
+# puts type_map
+header_comment = <<comment
+// #{$class_full_name}.h
+// 
+// Created by #{ENV['USER']} on #{Time.now.strftime("%Y/%m/%d")}
+// Model file Generated using objc_export https://github.com/gogozs/objc_export
+comment
+
+header_result = ""
+header_result << header_comment
+header_result << "\n#import <Foundation/Foundation.h>\n\n"
+header_result << "@interface #{$class_full_name} : NSObject\n\n"
+header_file_name = "#{$class_full_name}.h"
+
+imp_result = ""
+imp_result << "#import \"#{header_file_name}\"\n\n"
+imp_result << "@interface #{$class_full_name} ()\n@end\n\n"
+
+imp_result << "@implementation #{$class_full_name}\n"
+imp_result << "@end"
+
+imp_file_name = "#{$class_full_name}.m"
+
+input_json.each do |key, value|
+  header_result << "#{prop_for_key_value(key, value)}\n"
+end
+header_result << "\n@end"
+
+header_file_path = "#{input_dir}/#{header_file_name}"
+imp_file_path = "#{input_dir}/#{imp_file_name}"
+
+File.open(header_file_path, 'w') do |io|
+  io << header_result
+end
+
+File.open(imp_file_path, 'w') do |io|
+  io << imp_result
+end
+