• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (C) 2015, The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include <memory>
20 #include <ostream>
21 #include <string>
22 
23 #include <stdio.h>
24 
25 #include <android-base/macros.h>
26 
27 namespace android {
28 namespace aidl {
29 
30 class CodeWriter;
31 using CodeWriterPtr = std::unique_ptr<CodeWriter>;
32 
33 class CodeWriter {
34  public:
35   // Get a CodeWriter that writes to a file. When filename is "-",
36   // it is written to stdout.
37   static CodeWriterPtr ForFile(const std::string& filename);
38   // Get a CodeWriter that writes to a string buffer.
39   // The buffer gets updated only after Close() is called or the CodeWriter
40   // is deleted -- much like a real file.
41   static CodeWriterPtr ForString(std::string* buf);
42   // Write a formatted string to this writer in the usual printf sense.
43   // Returns false on error.
44   virtual bool Write(const char* format, ...) __attribute__((format(printf, 2, 3)));
45   void Indent();
46   void Dedent();
47   virtual bool Close();
48   virtual ~CodeWriter() = default;
49   CodeWriter() = default;
50 
51   CodeWriter& operator<<(const char* s);
52   CodeWriter& operator<<(const std::string& str);
53 
54  private:
55   CodeWriter(std::unique_ptr<std::ostream> ostream);
56   std::string ApplyIndent(const std::string& str);
57   const std::unique_ptr<std::ostream> ostream_;
58   int indent_level_ {0};
59   bool start_of_line_ {true};
60 };
61 
62 }  // namespace aidl
63 }  // namespace android
64