debian-mirror-gitlab/app/services/snippets/create_service.rb

# frozen_string_literal: true

module Snippets
  class CreateService < Snippets::BaseService
    include SpamCheckMethods

    CreateRepositoryError = Class.new(StandardError)

    def execute
      filter_spam_check_params

      snippet = if project
                  project.snippets.build(params)
                else
                  PersonalSnippet.new(params)
                end

      unless Gitlab::VisibilityLevel.allowed_for?(current_user, snippet.visibility_level)
        deny_visibility_level(snippet)

        return snippet_error_response(snippet, 403)
      end

      snippet.author = current_user

      spam_check(snippet, current_user)

      if save_and_commit(snippet)
        UserAgentDetailService.new(snippet, @request).create
        Gitlab::UsageDataCounters::SnippetCounter.count(:create)

        ServiceResponse.success(payload: { snippet: snippet } )
      else
        snippet_error_response(snippet, 400)
      end
    end

    private

    def save_and_commit(snippet)
      result = snippet.with_transaction_returning_status do
        (snippet.save && snippet.store_mentions!).tap do |saved|
          break false unless saved

          if Feature.enabled?(:version_snippets, current_user)
            create_repository_for(snippet)
          end
        end
      end

      create_commit(snippet) if result && snippet.repository_exists?

      result
    rescue => e # Rescuing all because we can receive Creation exceptions, GRPC exceptions, Git exceptions, ...
      snippet.errors.add(:base, e.message)

      # If the commit action failed we need to remove the repository if exists
      snippet.repository.remove if snippet.repository_exists?

      # If the snippet was created, we need to remove it as we
      # would do like if it had had any validation error
      snippet.delete if snippet.persisted?

      false
    end

    def create_repository_for(snippet)
      snippet.create_repository

      raise CreateRepositoryError, 'Repository could not be created' unless snippet.repository_exists?
    end

    def create_commit(snippet)
      commit_attrs = {
        branch_name: 'master',
        message: 'Initial commit'
      }

      snippet.snippet_repository.multi_files_action(current_user, snippet_files, commit_attrs)
    end

    def snippet_files
      [{ file_path: params[:file_name], content: params[:content] }]
    end
  end
end