chrome/common/auto_start_linux.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/common/auto_start_linux.h"
   6
   7 #include "base/environment.h"
   8 #include "base/file_util.h"
   9 #include "base/files/file_path.h"
  10 #include "base/logging.h"
  11 #include "base/nix/xdg_util.h"
  12 #include "base/strings/string_tokenizer.h"
  13
  14 namespace {
  15
  16 const base::FilePath::CharType kAutostart[] = "autostart";
  17
  18 base::FilePath GetAutostartDirectory(base::Environment* environment) {
  19   base::FilePath result = base::nix::GetXDGDirectory(
  20       environment,
  21       base::nix::kXdgConfigHomeEnvVar,
  22       base::nix::kDotConfigDir);
  23   result = result.Append(kAutostart);
  24   return result;
  25 }
  26
  27 }  // namespace
  28
  29 bool AutoStart::AddApplication(const std::string& autostart_filename,
  30                                const std::string& application_name,
  31                                const std::string& command_line,
  32                                bool is_terminal_app) {
  33   scoped_ptr<base::Environment> environment(base::Environment::Create());
  34   base::FilePath autostart_directory = GetAutostartDirectory(environment.get());
  35   if (!base::DirectoryExists(autostart_directory) &&
  36       !base::CreateDirectory(autostart_directory)) {
  37     return false;
  38   }
  39
  40   base::FilePath autostart_file =
  41       autostart_directory.Append(autostart_filename);
  42   std::string terminal = is_terminal_app ? "true" : "false";
  43   std::string autostart_file_contents =
  44       "[Desktop Entry]\n"
  45       "Type=Application\n"
  46       "Terminal=" + terminal + "\n"
  47       "Exec=" + command_line + "\n"
  48       "Name=" + application_name + "\n";
  49   std::string::size_type content_length = autostart_file_contents.length();
  50   if (base::WriteFile(autostart_file, autostart_file_contents.c_str(),
  51                       content_length) !=
  52       static_cast<int>(content_length)) {
  53     base::DeleteFile(autostart_file, false);
  54     return false;
  55   }
  56   return true;
  57 }
  58
  59 bool AutoStart::Remove(const std::string& autostart_filename) {
  60   scoped_ptr<base::Environment> environment(base::Environment::Create());
  61   base::FilePath autostart_directory = GetAutostartDirectory(environment.get());
  62   base::FilePath autostart_file =
  63       autostart_directory.Append(autostart_filename);
  64   return base::DeleteFile(autostart_file, false);
  65 }
  66
  67 bool AutoStart::GetAutostartFileContents(
  68     const std::string& autostart_filename, std::string* contents) {
  69   scoped_ptr<base::Environment> environment(base::Environment::Create());
  70   base::FilePath autostart_directory = GetAutostartDirectory(environment.get());
  71   base::FilePath autostart_file =
  72       autostart_directory.Append(autostart_filename);
  73   return base::ReadFileToString(autostart_file, contents);
  74 }
  75
  76 bool AutoStart::GetAutostartFileValue(const std::string& autostart_filename,
  77                                       const std::string& value_name,
  78                                       std::string* value) {
  79   std::string contents;
  80   if (!GetAutostartFileContents(autostart_filename, &contents))
  81     return false;
  82   base::StringTokenizer tokenizer(contents, "\n");
  83   std::string token = value_name + "=";
  84   while (tokenizer.GetNext()) {
  85     if (tokenizer.token().substr(0, token.length()) == token) {
  86       *value = tokenizer.token().substr(token.length());
  87       return true;
  88     }
  89   }
  90   return false;
  91 }