BigW Consortium Gitlab

project_wiki.rb 4.07 KB
Newer Older
1 2 3 4
class ProjectWiki
  include Gitlab::ShellAdapter

  MARKUPS = {
5
    'Markdown' => :markdown,
6 7
    'RDoc'     => :rdoc,
    'AsciiDoc' => :asciidoc
8
  } unless defined?(MARKUPS)
9 10 11 12 13 14

  class CouldNotCreateWikiError < StandardError; end

  # Returns a string describing what went wrong after
  # an operation fails.
  attr_reader :error_message
15
  attr_reader :project
16 17 18 19 20 21 22 23 24 25 26 27 28 29

  def initialize(project, user = nil)
    @project = project
    @user = user
  end

  def path
    @project.path + '.wiki'
  end

  def path_with_namespace
    @project.path_with_namespace + ".wiki"
  end

30 31 32 33
  def web_url
    Gitlab::Routing.url_helpers.namespace_project_wiki_url(@project.namespace, @project, :home)
  end

34 35 36 37 38 39 40 41 42 43 44 45
  def url_to_repo
    gitlab_shell.url_to_repo(path_with_namespace)
  end

  def ssh_url_to_repo
    url_to_repo
  end

  def http_url_to_repo
    [Gitlab.config.gitlab.url, "/", path_with_namespace, ".git"].join('')
  end

46
  def wiki_base_path
47
    [Gitlab.config.gitlab.relative_url_root, "/", @project.path_with_namespace, "/wikis"].join('')
48 49
  end

50 51 52 53
  # Returns the Gollum::Wiki object.
  def wiki
    @wiki ||= begin
      Gollum::Wiki.new(path_to_repo)
54
    rescue Rugged::OSError
55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
      create_repo!
    end
  end

  def empty?
    pages.empty?
  end

  # Returns an Array of Gitlab WikiPage instances or an
  # empty Array if this Wiki has no pages.
  def pages
    wiki.pages.map { |page| WikiPage.new(self, page, true) }
  end

  # Finds a page within the repository based on a tile
  # or slug.
  #
  # title - The human readable or parameterized title of
  #         the page.
  #
  # Returns an initialized WikiPage instance or nil
  def find_page(title, version = nil)
77 78
    page_title, page_dir = page_title_and_dir(title)
    if page = wiki.page(page_title, version, page_dir)
79 80 81 82 83 84
      WikiPage.new(self, page, true)
    else
      nil
    end
  end

85 86 87 88 89 90 91 92 93
  def find_file(name, version = nil, try_on_disk = true)
    version = wiki.ref if version.nil? # Gollum::Wiki#file ?
    if wiki_file = wiki.file(name, version, try_on_disk)
      wiki_file
    else
      nil
    end
  end

94 95 96
  def create_page(title, content, format = :markdown, message = nil)
    commit = commit_details(:created, message, title)

97
    wiki.write_page(title, format.to_sym, content, commit)
98 99

    update_project_activity
100 101 102 103 104 105 106 107
  rescue Gollum::DuplicatePageError => e
    @error_message = "Duplicate page: #{e.message}"
    return false
  end

  def update_page(page, content, format = :markdown, message = nil)
    commit = commit_details(:updated, message, page.title)

108
    wiki.update_page(page, page.name, format.to_sym, content, commit)
109 110

    update_project_activity
111 112 113 114
  end

  def delete_page(page, message = nil)
    wiki.delete_page(page, commit_details(:deleted, message, page.title))
115 116

    update_project_activity
117 118
  end

119
  def page_title_and_dir(title)
120
    title_array = title.split("/")
121
    title = title_array.pop
122
    [title, title_array.join("/")]
123 124
  end

Dmitriy Zaporozhets committed
125 126 127 128 129
  def search_files(query)
    repository.search_files(query, default_branch)
  end

  def repository
130
    @repository ||= Repository.new(path_with_namespace, @project)
Dmitriy Zaporozhets committed
131 132 133 134 135 136
  end

  def default_branch
    wiki.class.default_ref
  end

137 138
  def create_repo!
    if init_repo(path_with_namespace)
139
      wiki = Gollum::Wiki.new(path_to_repo)
140 141 142
    else
      raise CouldNotCreateWikiError
    end
143 144 145 146

    repository.after_create

    wiki
147
  end
Gabriel Mazetto committed
148

149 150 151 152 153 154
  def hook_attrs
    {
      web_url: web_url,
      git_ssh_url: ssh_url_to_repo,
      git_http_url: http_url_to_repo,
      path_with_namespace: path_with_namespace,
Gabriel Mazetto committed
155
      default_branch: default_branch
156 157
    }
  end
158

159 160
  private

161 162 163 164 165 166 167
  def init_repo(path_with_namespace)
    gitlab_shell.add_repository(path_with_namespace)
  end

  def commit_details(action, message = nil, title = nil)
    commit_message = message || default_message(action, title)

168
    { email: @user.email, name: @user.name, message: commit_message }
169 170 171 172 173 174 175 176 177
  end

  def default_message(action, title)
    "#{@user.username} #{action} page: #{title}"
  end

  def path_to_repo
    @path_to_repo ||= File.join(Gitlab.config.gitlab_shell.repos_path, "#{path_with_namespace}.git")
  end
178 179 180 181

  def update_project_activity
    @project.touch(:last_activity_at)
  end
182
end